Research Demo
ViT Attribution Methods Comparison
Architecture: Google ViT
Dataset: ImageNet-1K
F2HF excels at pinpointing small objects and objects harder to distinguish from background
When target objects occupy only a small area (e.g., 50×50 pixels in a 224×224 image), or the object is very similar to the background (e.g., a turtle in mat with turtle's body color), F2HF can filter out noisy attention heads better than other methods and therefore generates more accurate heatmap.
Our key idea was taking into account the layer-norm's function in each block. Layer-norm takes the global mean of all inputs to it, and in ViT, the layer-norm of each block get a concatenated vector of both the attention output and the previous block's output. That means, in training, during the backward pass, the attention output of each block is influenced through the previous block's FFD. That means, each block (l-1)'s FFD learn to filter attention output of block l.
Sample Examples
Click any sample to load image and default class
Click or drop image here
Supports JPEG, PNG, WebP (224×224)
Change Image
▼
▼
Note: please check footer to find all available class name or class index and their mapping for ImageNet dataset.
•
Running... (Please wait)
Heatmaps
Model Predicted (Top-1):
—
Target Explanation Class:
—
Active Model:
—
F2HF
Proposed Method
Awaiting simulation
BeyondIntuition
Baseline
Awaiting simulation
Integrated Gradients
Baseline
Awaiting simulation
Grad-CAM
Baseline
Awaiting simulation
Attention Rollout
Baseline
Awaiting simulation