mlsanityData to 7 Oct 2026

Copy the Same, Distill the Difference: Initializing Linear Vision Transformers

arXiv 2609.35745 ↗First posted 28 Sep 2026cs.CV cs.AI cs.LGcited by 0

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

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  4. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  5. 2017Mask R-CNNcited by 7,259
  6. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
  7. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  8. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  9. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
  10. 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
  11. 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
  12. 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
32 more
  1. 2019Natural Adversarial Examplescited by 1,401
  2. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
  3. 2018Unified Perceptual Parsing for Scene Understandingcited by 1,265
  4. 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,010
  5. 2024VMamba: Visual State Space Modelcited by 955
  6. 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
  7. 2022Vision Transformer Adapter for Dense Predictionscited by 564
  8. 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
  9. 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
  10. 2023FLatten Transformer: Vision Transformer using Focused Linear Attentioncited by 170
  11. 2024The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicrycited by 119
  12. 2023Mimetic Initialization of Self-Attention Layerscited by 52
  13. 2024Bridging the Divide: Reconsidering Softmax and Linear Attentioncited by 48
  14. 2023Initializing Models with Larger Onescited by 39
  15. 2024CLEAR: Conv-Like Linearization Revs Pre-Trained Diffusion Transformers Upcited by 26
  16. 2025RADLADS: Rapid Attention Distillation to Linear Attention Decoders at Scalecited by 24
  17. 2022Attention Distillation: self-supervised vision transformer students need more guidancecited by 16
  18. 2024On the Surprising Effectiveness of Attention Transfer for Vision Transformerscited by 16
  19. 2024Mimetic Initialization Helps State Space Models Learn to Recallcited by 15
  20. 2025ViT3: Unlocking Test-Time Training in Visioncited by 15
  21. 2026Test-Time Training with KV Binding Is Secretly Linear Attentioncited by 10
  22. 2025LiT: Delving into a Simple Linear Diffusion Transformer for Image Generationcited by 9
  23. 2025Structured Initialization for Vision Transformerscited by 6
  24. 2025ViT-Linearizer: Distilling Quadratic Knowledge into Linear-Time Vision Modelscited by 5
  25. 2025Data Efficient Any Transformer-to-Mamba Distillation via Attention Bridgecited by 4
  26. 2025Vision Transformers are Circulant Attention Learnerscited by 4
  27. 2026MHLA: Restoring Expressivity of Linear Attention via Token-Level Multi-Headcited by 4
  28. 2026ViT-AdaLA: Adapting Vision Transformers with Linear Attentioncited by 4
  29. 2026Attention Transfer Is Not Universally Effective for Vision Transformerscited by 3
  30. 2026Mimetic Initialization of MLPscited by 2
  31. 2026Linearizing Vision Transformer with Test-Time Trainingcited by 2
  32. 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.