Zoology: Measuring and Improving Recall in Efficient Language Models
Not on the map. Why?
Cited by 30 in State-space sequence models (S4, Mamba), but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Attention-free language models that combine gating and convolutions are growing in popularity due to their efficiency and increasingly competitive performance. To better understand these architectures, we pretrain a suite of 17 attention and "gated-convolution" language models, finding that SoTA gated-convolution architectures still underperform attention by up to 2.1 perplexity points on the Pile. In fine-grained analysis, we find 82% of the gap is explained by each model's ability to recall information that is previously mentioned in-context, e.g. "Hakuna Matata means no worries Hakuna Matata it means no" rightarrow "??". On this task, termed "associative recall", we find that attention outperforms gated-convolutions by a large margin: a 70M parameter attention model outperforms a 1.4 billion parameter gated-convolution model on associative recall. This is surprising because prior work shows gated convolutions can perfectly solve synthetic tests for AR capability. To close the gap between synthetics and real language, we develop a new formalization of the task called multi-query associative recall (MQAR) that better reflects actual language. We perform an empirical and theoretical study of MQAR that elucidates differences in the parameter-efficiency of attention and gated-convolution recall. Informed by our analysis, we evaluate simple convolution-attention hybrids and show that hybrids with input-dependent sparse attention patterns can close 97.4% of the gap to attention, while maintaining sub-quadratic scaling. Our code is accessible at: https://github.com/HazyResearch/zoology.
Key facts. Cited by 204 papers in the map (data through 14 Oct 2026). First posted on arXiv 8 Dec 2023.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2023: 1; 2024: 51; 2025: 55; 2026: 97.
204 papers.
Sort:
By area:
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inferencecited by 235
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 196
- 2024An Empirical Study of Mamba-based Language Modelscited by 186
- 2024Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modelingcited by 174
- 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 144
- 2024HGRN2: Gated Linear RNNs with State Expansioncited by 127
- 2025NoLiMa: Long-Context Evaluation Beyond Literal Matchingcited by 126
- 2024The Mamba in the Llama: Distilling and Accelerating Hybrid Modelscited by 122
38 more
- 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Taskscited by 105
- 2024In-Context Language Learning: Architectures and Algorithmscited by 89
- 2025Cartridges: Lightweight and general-purpose long context representations via self-studycited by 86
- 2024Theoretical Foundations of Deep Selective State-Space Modelscited by 74
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 69
- 2025ATLAS: Learning to Optimally Memorize the Context at Test Timecited by 66
- 2024RNNs are not Transformers (Yet): The Key Bottleneck on In-context Retrievalcited by 64
- 2025The Illusion of Diminishing Returns: Measuring Long Horizon Execution in LLMscited by 64
- 2024MambaByte: Token-free Selective State Space Modelcited by 59
- 2024Mechanistic Design and Scaling of Hybrid Architecturescited by 59
- 2024Longhorn: State Space Models are Amortized Online Learnerscited by 59
- 2024Jamba-1.5: Hybrid Transformer-Mamba Models at Scalecited by 50
- 2024Hydra: Bidirectional State Space Models Through Generalized Matrix Mixerscited by 44
- 2025Log-Linear Attentioncited by 40
- 2024BlackMamba: Mixture of Experts for State-Space Modelscited by 37
- 2025How do language models learn facts? Dynamics, curricula and hallucinationscited by 37
- 2025PaTH Attention: Position Encoding via Accumulating Householder Transformationscited by 34
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attentioncited by 33
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolationcited by 31
- 2024Separations in the Representational Capabilities of Transformers and Recurrent Architecturescited by 29
- 2024Understanding the differences in Foundation Models: Attention, State Space Models, and Recurrent Neural Networkscited by 26
- 2024Understanding Transformers via N-gram Statisticscited by 26
- 2025The emergence of sparse attention: impact of data distribution and benefits of repetitioncited by 25
- 2025RADLADS: Rapid Attention Distillation to Linear Attention Decoders at Scalecited by 24
- 2024MetaLA: Unified Optimal Linear Approximation to Softmax Attention Mapcited by 21
- 2025Long-Context Generalization with Sparse Attentioncited by 21
- 2025Distilling to Hybrid Attention Models via KL-Guided Layer Selectioncited by 20
- 2024LoCoCo: Dropping In Convolutions for Long Context Compressioncited by 19
- 2025Lattice: Learning to Efficiently Compress the Memorycited by 19
- 2024Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothingcited by 18
- 2024An Efficient Recipe for Long Context Extension via Middle-Focused Positional Encodingcited by 16
- 2025Mamba Modulation: On the Length Generalization of Mambacited by 16
- 2025Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanismcited by 15
- 2025Comba: Improving Bilinear RNNs with Closed-loop Controlcited by 15
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Modelscited by 15
- 2026Memory Caching: RNNs with Growing Memorycited by 15
- 2025Mixture of Sparse Attention: Content-Based Learnable Sparse Attention via Expert-Choice Routingcited by 14
50 most cited of 204 All 204 (JSON)
- 2026Beyond State Capacity: Why Retrieval Breaks in Mamba14 Oct 2026
- 2026Copy-Paste Language Models13 Oct 2026
- 2026AMemOps: Evaluating Memory Updates in Long Audio Streams13 Oct 2026
- 2026History Replays Itself: Unleashing Long-Context Capability in Linear Attention11 Oct 2026
- 2026DeltaNet-PC: Test-Time Memory Without Backpropagation Through Time10 Oct 2026
- 2026Orthros: Phase-Aware Heterogeneous Attention for Efficient Transformers9 Oct 2026
- 2026Double-Take: Improving Capacity, Binding, and Selectivity In Long-Context Within Bidirectional Memory9 Oct 2026
- 2026An all-transistor physical-learning transformer can be trained to perform in-context learning8 Oct 2026
- 2026Voltic: Distinguishing Volatility from Stochasticity in Recurrent Memory5 Oct 2026
- 2026Length Generalization Needs Proper Regularization3 Oct 2026
- 2026Learning What to Investigate Next: Meta-Reasoning for Long-Horizon Research Agents1 Oct 2026
- 2026Why Do Conventional World Models Fail to Learn Cellular Automata?30 Sep 2026
38 more
- 2026Switching Linear Attention30 Sep 2026
- 2026Triadic Linear Attention: Three-Dimensional Recurrent States for Long-Context Sequence Modeling29 Sep 2026
- 2026Periodic Weak Spots: Phase Sensitivity from Chunked KV-Cache Compression28 Sep 2026
- 2026SMat-Attention: Structured Long-Context Sequence Modeling28 Sep 2026
- 2026Quasi Linear Kernel Attention with Infinite Capacity28 Sep 2026
- 2026ReCAT: Remember, Count, and Time: Structured Recurrent Memory for Robot Manipulation28 Sep 2026
- 2026When Can Attention Heads Be Statically Defined?28 Sep 2026
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models28 Sep 2026
- 2026An Input-Dependent Readout Phase for Resonant Sequence Models26 Sep 2026
- 2026MassAlloc Attention: Let Attention Allocate Its Own Compute26 Sep 2026
- 2026CoWindow Attention: Full Causal Coverage Is a Collective Property26 Sep 2026
- 2026A Comparative Analysis of Attention versus State-Space Models for In-Context Learning26 Sep 2026
- 2026On-Policy Attention Linearization25 Sep 2026
- 2026Constant-Memory Recall: Learned Associations in a Fixed Matrix State23 Sep 2026
- 2026Latest Exact Match Attention22 Sep 2026
- 2026On the Importance of Gating: Memorization vs. In-Context Learning in State Space Models15 Sep 2026
- 2026Anatomy of Associative Recall in Fixed-State Recurrences: A Matched-State Decomposition, an Interference Wall, and a Curriculum That Breaks It14 Sep 2026
- 2026The Token Before the Value Is the Key: How Hybrid Architectures Organize Induction Circuits14 Sep 2026
- 2026RunningTensor: Generalizing Linear Attention to Higher-Order Recurrent States11 Sep 2026
- 2026What Attention Recalls and Recurrence Controls in Hybrid Language Models3 Sep 2026
- 2026Coupled Scaling: A Representational Accessibility Framework for Neural Scaling Laws3 Sep 2026
- 2026Good Memory Has ECC: Evaluating the Memory of Vision-Language Models Beyond Accuracy31 Aug 2026
- 2026Safin-1: Safety from Within through Memory-Native State Evolution31 Aug 2026
- 2026Reading Is Not Using: Retrieval, Judgment, and the Design of AI Financial Research Workflows25 Aug 2026
- 2026MARCH: Scaling Recurrent Memory with Content-Routed State Anchors12 Aug 2026
- 2026DART: Decoded Attention over Recurrent States for Efficient Long-Context Sequence Modeling3 Aug 2026
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routing28 Jul 2026
- 2026Learning What Matters: Supervising Global Context Pruning with Causal Evidence Sets23 Jul 2026
- 2026Naju: A Native Discrete State-Space Model with Independent Retention and Writing for Long-Sequence Memory23 Jul 2026
- 2026Convolution for Large Language Models20 Jul 2026
- 2026Kernelized Linear Attention: Breaking the Capacity Wall with Symmetric Cones19 Jul 2026
- 2026The Capability Convergence Hypothesis: Capability from Access Structure, Not Scale14 Jul 2026
- 2026Context by Distinct Information: An Auditable Dirichlet-Process Working Memory for Long, Redundant Context Streams11 Jul 2026
- 2026DSSMs: State Space Models with Explicit Memory via Delay Differential Equations11 Jul 2026
- 2026Remembering Distinct Items, Not Tokens: A Learnable Dirichlet-Process Cache Between State-Space Models and Attention10 Jul 2026
- 2026SHiPPO: Recurrent Memory with Transported Polynomial Projections3 Jul 2026
- 2026The Orthogonalized Read Is a Removable Training Scaffold for Recurrent Memory2 Jul 2026
- 2026A Hippocampus for Linear Attention: An Exact Memory for What the Recurrent State Forgets2 Jul 2026
50 newest of 204 All 204 (JSON)
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2024An Empirical Study of Mamba-based Language Modelscited by 186
- 2024Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modelingcited by 174
- 2024The Mamba in the Llama: Distilling and Accelerating Hybrid Modelscited by 122
- 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Taskscited by 105
- 2024Theoretical Foundations of Deep Selective State-Space Modelscited by 74
- 2024MambaByte: Token-free Selective State Space Modelcited by 59
- 2024Longhorn: State Space Models are Amortized Online Learnerscited by 59
- 2024Jamba-1.5: Hybrid Transformer-Mamba Models at Scalecited by 50
- 2024Hydra: Bidirectional State Space Models Through Generalized Matrix Mixerscited by 44
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attentioncited by 33
- 2024Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothingcited by 18
12 most cited of 28 All 28 (JSON)
- 2026Beyond State Capacity: Why Retrieval Breaks in Mamba14 Oct 2026
- 2026Anatomy of Associative Recall in Fixed-State Recurrences: A Matched-State Decomposition, an Interference Wall, and a Curriculum That Breaks It14 Sep 2026
- 2026DART: Decoded Attention over Recurrent States for Efficient Long-Context Sequence Modeling3 Aug 2026
- 2026Naju: A Native Discrete State-Space Model with Independent Retention and Writing for Long-Sequence Memory23 Jul 2026
- 2026Gated DeltaNet-2: Decoupling Erase and Write in Linear Attention21 May 2026
- 2026Sessa: Selective State Space Attention20 Apr 2026
- 2026Optimal Decay Spectra for Linear Recurrences8 Apr 2026
- 2026When Does Content-Based Routing Work? Representation Requirements for Selective Attention in Hybrid Sequence Models22 Mar 2026
- 2025Mamba Modulation: On the Length Generalization of Mamba23 Sep 2025
- 2025Understanding Input Selectivity in Mamba: Impact on Approximation Power, Memorization, and Associative Recall Capacity13 Jun 2025
- 2025Sequential-Parallel Duality in Prefix Scannable Models12 Jun 2025
- 2025Mechanistic evaluation of Transformers and state space models21 May 2025
12 newest of 28 All 28 (JSON)
- 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inferencecited by 235
- 2025Cartridges: Lightweight and general-purpose long context representations via self-studycited by 86
- 2024LoCoCo: Dropping In Convolutions for Long Context Compressioncited by 19
- 2025Alleviating Forgetfulness of Linear Attention by Hybrid Sparse Attention and Contextualized Learnable Token Evictioncited by 3
- 2026What Attention Recalls and Recurrence Controls in Hybrid Language Modelscited by 2
- 2026SR-TTT Does Not Learn Retrieval: A Correction and Mechanistic Post-Mortem of Surprisal-Aware Residual Test-Time Trainingcited by 1
- 2026Blurry Window Attentioncited by 1
- 2026Periodic Weak Spots: Phase Sensitivity from Chunked KV-Cache Compressioncited by 0
- 2026Periodic Weak Spots: Phase Sensitivity from Chunked KV-Cache Compression28 Sep 2026
- 2026What Attention Recalls and Recurrence Controls in Hybrid Language Models3 Sep 2026
- 2026Blurry Window Attention31 May 2026
- 2026SR-TTT Does Not Learn Retrieval: A Correction and Mechanistic Post-Mortem of Surprisal-Aware Residual Test-Time Training26 Feb 2026
- 2025Alleviating Forgetfulness of Linear Attention by Hybrid Sparse Attention and Contextualized Learnable Token Eviction23 Oct 2025
- 2025Cartridges: Lightweight and general-purpose long context representations via self-study6 Jun 2025
- 2024Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference16 Jul 2024
- 2024LoCoCo: Dropping In Convolutions for Long Context Compression8 Jun 2024
- 2025Benefits and Limitations of Communication in Multi-Agent Reasoningcited by 7
- 2024Exploring the Limitations of Mamba in COPY and CoT Reasoningcited by 6
- 2026Chain-of-Thought and Compressed Looped Transformers: A Memory-Budget Separationcited by 2
- 2026On the Expressive Power and Limitations of Multi-Layer SSMscited by 1
- 2026Echo: KV-Cache-Free Associative Recall with Spectral Koopman Operatorscited by 1
- 2026On-Policy Attention Linearizationcited by 0
- 2026On-Policy Attention Linearization25 Sep 2026
- 2026Chain-of-Thought and Compressed Looped Transformers: A Memory-Budget Separation29 May 2026
- 2026Echo: KV-Cache-Free Associative Recall with Spectral Koopman Operators7 May 2026
- 2026On the Expressive Power and Limitations of Multi-Layer SSMs16 Apr 2026
- 2025Benefits and Limitations of Communication in Multi-Agent Reasoning14 Oct 2025
- 2024Exploring the Limitations of Mamba in COPY and CoT Reasoning4 Oct 2024
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 196
- 2025Kimi Linear: An Expressive, Efficient Attention Architecture30 Oct 2025
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 144
- 2024HGRN2: Gated Linear RNNs with State Expansioncited by 127
- 2025NoLiMa: Long-Context Evaluation Beyond Literal Matchingcited by 126
- 2024In-Context Language Learning: Architectures and Algorithmscited by 89
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 69
- 2025ATLAS: Learning to Optimally Memorize the Context at Test Timecited by 66
- 2024RNNs are not Transformers (Yet): The Key Bottleneck on In-context Retrievalcited by 64
- 2025The Illusion of Diminishing Returns: Measuring Long Horizon Execution in LLMscited by 64
12 most cited of 151 All 151 (JSON)
- 2026Copy-Paste Language Models13 Oct 2026
- 2026AMemOps: Evaluating Memory Updates in Long Audio Streams13 Oct 2026
- 2026History Replays Itself: Unleashing Long-Context Capability in Linear Attention11 Oct 2026
- 2026DeltaNet-PC: Test-Time Memory Without Backpropagation Through Time10 Oct 2026
- 2026Orthros: Phase-Aware Heterogeneous Attention for Efficient Transformers9 Oct 2026
- 2026Double-Take: Improving Capacity, Binding, and Selectivity In Long-Context Within Bidirectional Memory9 Oct 2026
- 2026An all-transistor physical-learning transformer can be trained to perform in-context learning8 Oct 2026
- 2026Voltic: Distinguishing Volatility from Stochasticity in Recurrent Memory5 Oct 2026
- 2026Length Generalization Needs Proper Regularization3 Oct 2026
- 2026Switching Linear Attention30 Sep 2026
- 2026Triadic Linear Attention: Three-Dimensional Recurrent States for Long-Context Sequence Modeling29 Sep 2026
- 2026SMat-Attention: Structured Long-Context Sequence Modeling28 Sep 2026
12 newest of 151 All 151 (JSON)