Parameter Reduction Improves Vision Transformers: A Comparative Study of Sharing and Width Reduction
By: Anantha Padmanaban Krishna Kumar
Potential Business Impact:
Makes computer vision models work better, faster.
Although scaling laws and many empirical results suggest that increasing the size of Vision Transformers often improves performance, model accuracy and training behavior are not always monotonically increasing with scale. Focusing on ViT-B/16 trained on ImageNet-1K, we study two simple parameter-reduction strategies applied to the MLP blocks, each removing 32.7\% of the baseline parameters. Our \emph{GroupedMLP} variant shares MLP weights between adjacent transformer blocks and achieves 81.47\% top-1 accuracy while maintaining the baseline computational cost. Our \emph{ShallowMLP} variant halves the MLP hidden dimension and reaches 81.25\% top-1 accuracy with a 38\% increase in inference throughput. Both models outperform the 86.6M-parameter baseline (81.05\%) and exhibit substantially improved training stability, reducing peak-to-final accuracy degradation from 0.47\% to the range 0.03\% to 0.06\%. These results suggest that, for ViT-B/16 on ImageNet-1K with a standard training recipe, the model operates in an overparameterized regime in which MLP capacity can be reduced without harming performance and can even slightly improve it. More broadly, our findings suggest that architectural constraints such as parameter sharing and reduced width may act as useful inductive biases, and highlight the importance of how parameters are allocated when designing Vision Transformers. All code is available at: https://github.com/AnanthaPadmanaban-KrishnaKumar/parameter-efficient-vit-mlps.
Similar Papers
Mechanisms of Non-Monotonic Scaling in Vision Transformers
Machine Learning (CS)
Makes computer "eyes" learn better by changing how they see.
Diminishing Returns in Self-Supervised Learning
CV and Pattern Recognition
Makes small AI models learn better with less data.
Diversity-Guided MLP Reduction for Efficient Large Vision Transformers
CV and Pattern Recognition
Makes big AI models much smaller and faster.