Copy the Same, Distill the Difference: Initializing Linear Vision Transformers
Not on the map. Why?
Abstract (arXiv)
Linear Vision Transformers (ViTs) are designed to replace the attention in Softmax ViTs with the linear-complexity attention operator for more efficient token routing, but they require from-scratch pre-training and typically underperform the original Softmax version. How to initialize linear ViTs both efficiently and effectively still remains unclear. In this work, we explicitly ask: given that most foundation ViTs are built on the mainstream Softmax attention, can linear ViTs benefit from their pre-trained weights? Recent works on Attention Transfer show that attention is the effective transferable component between Softmax ViTs, suggesting attention alone suffices for such reuse. However, we find the opposite for Softmax-to-linear transfer. The attention weights are operator-specific: copying them barely helps, and is sometimes even worse than random initialization. Instead, the attention's token routing behavior can be recovered through distillation with a proper loss design, letting linear ViTs reduce the gap and even match Softmax ones. In contrast, the MLP weights, which carry the learned representation, are operator-agnostic: they can be transferred by simple direct copying, which already carries most of the benefit of the pre-trained weights. Thus, copying MLPs can serve as an effective foundation for Softmax-to-linear transfer: paired with the distilled attention, linear ViTs eventually close the remaining gap and even surpass Softmax ones. These findings hold consistently across various linear ViT variants, different model sizes, and diverse datasets. We hope this study deepens the understanding of reusing pre-trained weights across attention operators: copy what stays the same and distill what differs, to recover the benefit across the Softmax-to-linear boundary.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 44 of its 52 references that are in the map. First posted on arXiv 28 Sep 2026.
Builds on
44 of 52 references.
References · 44
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2017Mask R-CNNcited by 7,259
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
32 more
- 2019Natural Adversarial Examplescited by 1,401
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,265
- 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,010
- 2024VMamba: Visual State Space Modelcited by 955
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2022Vision Transformer Adapter for Dense Predictionscited by 564
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2023FLatten Transformer: Vision Transformer using Focused Linear Attentioncited by 170
- 2024The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicrycited by 119
- 2023Mimetic Initialization of Self-Attention Layerscited by 52
- 2024Bridging the Divide: Reconsidering Softmax and Linear Attentioncited by 48
- 2023Initializing Models with Larger Onescited by 39
- 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Upcited by 26
- 2025RADLADS: Rapid Attention Distillation to Linear Attention Decoders at Scalecited by 24
- 2022Attention Distillation: self-supervised vision transformer students need more guidancecited by 16
- 2024On the Surprising Effectiveness of Attention Transfer for Vision Transformerscited by 16
- 2024Mimetic Initialization Helps State Space Models Learn to Recallcited by 15
- 2025ViT3: Unlocking Test-Time Training in Visioncited by 15
- 2026Test-Time Training with KV Binding Is Secretly Linear Attentioncited by 10
- 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generationcited by 9
- 2025Structured Initialization for Vision Transformerscited by 6
- 2025ViT-Linearizer: Distilling Quadratic Knowledge into Linear-Time Vision Modelscited by 5
- 2025Data Efficient Any Transformer-to-Mamba Distillation via Attention Bridgecited by 4
- 2025Vision Transformers are Circulant Attention Learnerscited by 4
- 2026MHLA: Restoring Expressivity of Linear Attention via Token-Level Multi-Headcited by 4
- 2026ViT-AdaLA: Adapting Vision Transformers with Linear Attentioncited by 4
- 2026Attention Transfer Is Not Universally Effective for Vision Transformerscited by 3
- 2026Mimetic Initialization of MLPscited by 2
- 2026Linearizing Vision Transformer with Test-Time Trainingcited by 2
- 2026DiD It in 87 Minutes: A Label-Free Softmax-to-Linear Adaptation of Vision Transformers for Object Detectioncited by 1
Built on by
Not cited yet.