Forget Attention: Importance-Aware Attention Is All You Need
Not on the map. Why?
Abstract (arXiv)
Combining attention's global retrieval with the sequential importance signal of state space models (SSMs) is the open challenge of hybrid language modeling. Transformers see everywhere but cannot prioritize; SSMs know what matters but cannot revisit. Existing hybrids -- Jamba (block level) and Hymba (head level) -- place the two in separate compartments, so neither informs the other during the attention computation itself. We propose SISA (SSM-Informed Softmax Attention), which adds an SSM-derived importance term directly inside the attention score and realizes the full operation as a single SDPA call on augmented query/key vectors -- no recurrent state, no custom kernel. At 152M / 5B tokens, SISA reaches LAMBADA-greedy 17.3% (vs. Transformer 13.9 and Mamba-3 15.5) and attains NIAH 100% from step 1K, 7x faster than Transformer's retrieval convergence; at 369M, Mamba-3 leads LAMBADA while SISA preserves perfect NIAH and stock-SDPA execution. SISA thus defines a third design axis for SSM-attention hybrids -- score-level fusion -- beyond the block-level and head-level paradigms that have dominated the field.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 28 references that are in the map. First posted on arXiv 1 Jun 2026.
Builds on
20 of 28 references.
References · 20
- 2017Attention Is All You Needcited by 59,932
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 232
- 2024An Empirical Study of Mamba-based Language Modelscited by 177
- 2024Zamba: A Compact 7B SSM Hybrid Modelcited by 118
- 2024Hymba: A Hybrid-head Architecture for Small Language Modelscited by 95
8 more
- 2025Nemotron-H: A Family of Accurate and Efficient Hybrid Mamba-Transformer Modelscited by 88
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 76
- 2024Theory, Analysis, and Best Practices for Sigmoid Self-Attentioncited by 62
- 2024RecurrentGemma: Moving Past Transformers for Efficient Open Language Modelscited by 61
- 2025Falcon-H1: A Family of Hybrid-Head Language Models Redefining Efficiency and Performancecited by 52
- 2025Forgetting Transformer: Softmax Attention with a Forget Gatecited by 50
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolationcited by 30
- 2024The Zamba2 Suite: Technical Reportcited by 20
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Understanding and Enhancing Mamba-Transformer Hybrids for Memory Recall and Language Modeling
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids
- 2024Taipan: Efficient and Expressive State Space Language Models with Selective Attention
- 2026Interdomain Attention: Beyond Token-Level Key-Value Memory
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Models
- 2026Rethinking the Role of Efficient Attention in Hybrid Architectures
- 2025Native Hybrid Attention for Efficient Sequence Modeling
- 2026Blurry Window Attention