IC-291ViT-B/16 and MAE-B/16 exhibit nearly inverted distributions of knowledge neurons across layers despite identical architecture and training data

Yifan Wang, Yifei Liu, Yingdong Shi, Changming Li, Anqi Pang, Sibei Yang, Jingyi Yu, Kan Ren

SourceDiscovering Influential Neuron Path in Vision Transformers

Using knowledge attribution (integrated gradients on individual FFN neurons), the paper measures which neurons are most important for classification in each of the 12 layers. Despite both models having 12 transformer layers, the same hidden size (768), the same patch size (16), and being pretrained and fine-tuned on ImageNet, the layer-wise distribution of top-attribution neurons is almost inverted between the supervised ViT-B/16 and the self-supervised MAE-B/16. This suggests that the pretraining paradigm fundamentally changes how knowledge is organized across depth.

Evidence
observational
Caveat
The comparison is limited to two models of the same size; the paper does not test whether the inversion holds across other sizes or architectures.
Model
ViT ViT-B/16, MAE-B/16
Concepts
Depth-dependent structure
Datasets
ImageNet-1k / ImageNet / ImageNet-1k-val / ImageNet-Val [eval]
Methods
Knowledge Attribution [primary]
Related work
Knowledge Neurons in Pretrained Transformers [builds-on]
Related findings
IC-290, IC-292, IC-293
Extraction
automatic-extraction