mlsanityData to 14 Oct 2026

When Does Content-Based Routing Work? Representation Requirements for Selective Attention in Hybrid Sequence Models

arXiv 2603.20997 ↗First posted 22 Mar 2026cs.LGcited by 1

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

Abstract (arXiv)

We identify a routing paradox in hybrid sequence models: content-based routing - deciding which tokens deserve expensive attention - requires pairwise computation, and this requirement is inescapable. Through 20+ controlled experiments across three tasks, multiple scales (200K to 1.4B parameters), and 15+ routing mechanisms, we map the routing landscape exhaustively. Every system that achieves high routing precision does so through pairwise token comparison. Every mechanism that avoids pairwise computation fails: recurrent models (Mamba-1.4B: 29%), memory banks (12%), bandits (0.7-3.6%), contrastive pretraining (1.6%), and 12 other approaches all cluster at 1-29%. Routing needs two ingredients: (1) per-token representations with bidirectional context and (2) pairwise token comparison. Bidirectional Mamba (O(n)) + pairwise comparison achieves 99.5%; replacing the full pairwise router with rank-1 projection improves this to 99.7%. Adding one bidirectional layer to frozen Pythia-1B recovers 99.4% routing. Six different O(n) preprocessing mechanisms (bidirectional Mamba, Perceiver inducing points, causal attention with E2E training, sparse attention, bidirectional attention, rank-1 projection) all succeed; global mean pooling (1.9%) and Fourier mixing (0.9%) fail. The routing signal occupies a ~34-dimensional latent subspace, invisible to cosine similarity. Non-learned indices (Bloom filter: 90.9%; BM25: 82.7%) bypass the bottleneck for exact/keyword matching. Combining O(n) bidirectional Mamba with rank-1 pairwise projection yields 99.7% routing at linear inference cost.

Key facts. Cited by 1 paper 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 34 of its 38 references that are in the map. First posted on arXiv 22 Mar 2026.

Builds on

34 of 38 references.

Closest ancestors · 5

Same direction, newest first

  1. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  2. 2024Jamba: A Hybrid Transformer-Mamba Language Model28 Mar
  3. 2024Simple linear attention language models balance the recall-throughput tradeoff28 Feb
  4. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  5. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 29

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2023GPT-4 Technical Reportcited by 18,396
  4. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  5. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
  6. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
  7. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  8. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  9. 2020Longformer: The Long-Document Transformercited by 3,413
  10. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  11. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  12. 2024Mixtral of Expertscited by 1,823
17 more
  1. 2020Big Bird: Transformers for Longer Sequencescited by 1,684
  2. 2020Reformer: The Efficient Transformercited by 1,668
  3. 2020Rethinking Attention with Performerscited by 1,601
  4. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  5. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  6. 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,178
  7. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  8. 2020Efficient Transformers: A Surveycited by 790
  9. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  10. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  11. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
  12. 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
  13. 2019Are Transformers universal approximators of sequence-to-sequence functions?cited by 406
  14. 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
  15. 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 242
  16. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
  17. 2024Zamba: A Compact 7B SSM Hybrid Modelcited by 124

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2025Rethinking Predictive Modeling for LLM Routing: When Simple kNN Beats Complex Learned Routers
  2. 2026All Routes Lead to Collapse
  3. 2026Scalable Prompt Routing via Fine-Grained Latent Task Discovery
  4. 2026Do Dynamic Routers Need Memory? HeRo: History-Aware Routing for Efficient LLM Inference
  5. 2026Expressivity-Efficiency Tradeoffs for Hybrid Sequence Models
  6. 2026Equifinality in Mixture of Experts: Routing Topology Does Not Determine Language Modeling Quality
  7. 2024Routing in Sparsely-gated Language Models responds to Context
  8. 2026Self-Routing: Parameter-Free Expert Routing from Hidden States