mlsanityData to 14 Oct 2026

When to Think Fast and Slow? AMOR: Adaptive Entropy Gate for Hybrid Models

arXiv 2602.13215 ↗First posted 22 Jan 2026cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

Recurrent-attention hybrids aim to combine the efficiency of recurrence with the contextual recall of attention, but existing approaches typically apply attention uniformly across all positions, even when the recurrent state alone is sufficient for accurate prediction. We introduce AMOR (Adaptive Metacognitive Output Router), a post-hoc hybrid architecture that selectively invokes attention based on predictive uncertainty. A recurrent backbone is augmented with entropy-gated attention blocks that activate only when the model's output entropy exceeds a dynamic threshold derived from a running batch median and scaled standard deviation. The resulting binary gate requires no learned routing parameters. Pretrained from scratch on FineWeb-Edu and with attention invoked on only ~40% of positions, one of the AMOR variants (Mamba2 or Gated DeltaNet backbones) achieves the highest eight-task common-sense reasoning average at each scale among pure recurrent, pure attention, and fixed-schedule hybrid models. AMOR also improves retrieval performance over pure recurrent models while remaining competitive against fixed-schedule hybrids. Additionally, AMOR retains the long-context robustness of its recurrent backbones, where the Transformer and other hybrid architectures degrade under distribution shift. These results suggest that when attention is applied matters as much as how much: selectively allocating attention based on predictive uncertainty improves accuracy, robustness, and efficiency, offering a simple alternative to uniform or fixed routing strategies.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 62 of its 77 references that are in the map. First posted on arXiv 22 Jan 2026.

Builds on

62 of 77 references.

References · 62

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  5. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  6. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  7. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  8. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  9. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
  10. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  11. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
  12. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
38 more
  1. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  2. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  3. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  4. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  5. 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,079
  6. 2020GLU Variants Improve Transformercited by 1,804
  7. 2019Root Mean Square Layer Normalizationcited by 1,480
  8. 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,417
  9. 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
  10. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
  11. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
  12. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  13. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  14. 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
  15. 2019DROP: A Reading Comprehension Benchmark Requiring Discrete Reasoning Over Paragraphscited by 903
  16. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  17. 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
  18. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
  19. 2023YaRN: Efficient Context Window Extension of Large Language Modelscited by 601
  20. 2017The NarrativeQA Reading Comprehension Challengecited by 597
  21. 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 535
  22. 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
  23. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
  24. 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 486
  25. 2019Multi-News: a Large-Scale Multi-Document Summarization Dataset and Abstractive Hierarchical Modelcited by 431
  26. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 407
  27. 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 401
  28. 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
  29. 2021QMSum: A New Benchmark for Query-based Multi-domain Meeting Summarizationcited by 365
  30. 2023RepoBench: Benchmarking Repository-Level Code Auto-Completion Systemscited by 353
  31. 2021Efficient Attentions for Long Document Summarizationcited by 345
  32. 2022Confident Adaptive Language Modelingcited by 310
  33. 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
  34. 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 242
  35. 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
  36. 2024Simple linear attention language models balance the recall-throughput tradeoffcited by 208
  37. 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
  38. 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 196

50 most cited of 62 All 62 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.