mlsanityData to 14 Oct 2026

Data Efficient Any Transformer-to-Mamba Distillation via Attention Bridge

arXiv 2510.19266 ↗First posted 22 Oct 2025cs.LGcited by 4

On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language

Abstract (arXiv)

State-space models (SSMs) have emerged as promising alternatives to Transformers for sequence modeling. However, training competitive SSMs from scratch remains computationally intensive, and the ecosystem around them is far less mature than that of Transformers. Moreover, the architectural differences between SSMs and Transformers make it challenging to efficiently transfer knowledge from pretrained Transformers. In this work, we propose Cross-architecture distillation via Attention Bridge (CAB), a distillation framework that transfers attention-related representations from Transformer teachers to state-space student models. Unlike conventional knowledge distillation that supervises only final predictions, CAB enables token-level intermediate supervision through a lightweight bridge and flexible layer-wise alignment. By aligning Transformer attention-related representations with Mamba's token-dependent state projections, CAB facilitates efficient cross-architecture knowledge transfer without inference-time overhead. Experiments on image classification and language modeling demonstrate that CAB improves Transformer-to-SSM distillation, particularly under limited-supervision settings. Overall, CAB provides an efficient pathway for transferring Transformer representations to SSM-based models, helping bridge the gap between mature Transformer ecosystems and emerging SSM ecosystems.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 44 of its 51 references that are in the map. First posted on arXiv 22 Oct 2025.

Builds on

44 of 51 references.

Closest ancestors · 8

Same direction, newest first

  1. 2025TransMamba: Fast Universal Architecture Adaption from Transformers to Mamba21 Feb
  2. 2025Llamba: Scaling Distilled Recurrent Models for Efficient Language Processing20 Feb
  3. 2024The Mamba in the Llama: Distilling and Accelerating Hybrid Models27 Aug
  4. 2024Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Models19 Aug
  5. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  6. 2024The Hidden Attention of Mamba Models3 Mar
  7. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  8. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 36

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
  6. 2016Pointer Sentinel Mixture Modelscited by 3,096
  7. 2013Visualizing and Understanding Convolutional Networkscited by 2,290
  8. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  9. 2014FitNets: Hints for Thin Deep Netscited by 1,653
  10. 2020Rethinking Attention with Performerscited by 1,601
  11. 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
  12. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
24 more
  1. 2016Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfercited by 1,077
  2. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
  3. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
  4. 2021Scaling Vision Transformerscited by 999
  5. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  6. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
  7. 2017Sim-to-Real Transfer of Robotic Control with Dynamics Randomizationcited by 672
  8. 2020The Future of Digital Health with Federated Learningcited by 633
  9. 2019Patient Knowledge Distillation for BERT Model Compressioncited by 508
  10. 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 482
  11. 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
  12. 2018Efficient Attention: Attention with Linear Complexitiescited by 368
  13. 2019Edge Intelligence: Paving the Last Mile of Artificial Intelligence with Edge Computingcited by 228
  14. 2022cosFormer: Rethinking Softmax in Attentioncited by 213
  15. 2019Privacy-preserving Federated Brain Tumour Segmentationcited by 169
  16. 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrencecited by 159
  17. 2025RWKV-7 "Goose" with Expressive Dynamic State Evolutioncited by 141
  18. 2020ConvBERT: Improving BERT with Span-based Dynamic Convolutioncited by 88
  19. 2020Transferring Inductive Biases through Knowledge Distillationcited by 41
  20. 2022Cross-Architecture Knowledge Distillationcited by 41
  21. 2025REPA Works Until It Doesn't: Early-Stopped, Holistic Alignment Supercharges Diffusion Trainingcited by 37
  22. 2022Bridging the Gap Between Vision Transformers and Convolutional Neural Networks on Small Datasetscited by 29
  23. 2025MambaIC: State Space Models for High-Performance Learned Image Compressioncited by 20
  24. 2024On the Surprising Effectiveness of Attention Transfer for Vision Transformerscited by 17

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Attention to Mamba: A Recipe for Cross-Architecture Distillation
  2. 2026InfoMamba: An Attention-Free Hybrid Mamba-Transformer Model
  3. 2025TransMamba: A Sequence-Level Hybrid Transformer-Mamba Language Model
  4. 2025A2Mamba: Attention-augmented State Space Models for Visual Recognition
  5. 2023A Comparative Analysis of Task-Agnostic Distillation Methods for Compressing Transformer Language Models
  6. 2025TransXSSM: A Hybrid Transformer State Space Model with Unified Rotary Position Embedding
  7. 2025Optimizing Knowledge Distillation in Transformers: Enabling Multi-Head Attention without Alignment Barriers
  8. 2025SCOUT: Toward Sub-Quadratic Attention via Segment Compression for Optimized Utility in Transformers