mlsanityData to 7 Oct 2026

Sessa: Selective State Space Attention

arXiv 2604.18580 ↗First posted 20 Apr 2026cs.LG cs.AI cs.CLcited by 0

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

Abstract (arXiv)

Modern sequence modeling is dominated by two families: Transformers, whose self-attention can access arbitrary elements of the visible sequence, and structured state-space models, which propagate information through an explicit recurrent state. These mechanisms face different limitations on long contexts: when attention is diffuse, the influence of individual tokens is diluted across the effective support, while recurrent state propagation can lose long-range sensitivity unless information is actively preserved. As a result, both mechanisms face challenges in preserving and selectively retrieving information over long contexts. We propose Sessa, a decoder that places attention inside a recurrent feedback path. This creates many attention-based paths through which past tokens can influence future states, rather than relying on a single attention read or a single recurrent chain. We prove that, under explicit assumptions and matched regimes, Sessa admits power-law memory tails O(ℓ-β) for 0 < β < 1, with slower decay than in the corresponding Transformer and Mamba-style baselines. We further give an explicit construction that achieves this power-law rate. Under the same assumptions, Sessa is the only model class among those considered that realizes flexible selective retrieval, including profiles whose influence does not decay with distance. Consistent with this theoretical advantage, across matched experiments, Sessa achieves the strongest performance on long-context benchmarks while remaining competitive with Transformer and Mamba-style baselines on short-context language modeling.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 31 of its 46 references that are in the map. First posted on arXiv 20 Apr 2026.

Builds on

31 of 46 references.

Closest ancestors · 4

Same direction, newest first

  1. 2025Understanding Input Selectivity in Mamba: Impact on Approximation Power, Memorization, and Associative Recall Capacity13 Jun
  2. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  4. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 27

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020Language Models are Few-Shot Learnerscited by 32,199
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2016Layer Normalizationcited by 5,579
  6. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  7. 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
  8. 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,853
  9. 2016Gaussian Error Linear Units (GELUs)cited by 3,431
  10. 2020Longformer: The Long-Document Transformercited by 3,362
  11. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
  12. 2020GLU Variants Improve Transformercited by 1,724
15 more
  1. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  2. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  3. 2024Chronos: Learning the Language of Time Seriescited by 712
  4. 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
  5. 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 418
  6. 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 201
  7. 2022Recurrent Memory Transformercited by 197
  8. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 188
  9. 2021Self-attention Does Not Need O(n2) Memorycited by 152
  10. 2022Block-Recurrent Transformerscited by 107
  11. 2024TransformerFAM: Feedback attention is working memorycited by 21
  12. 2025Parameterized Synthetic Text Generation with SimpleStoriescited by 9
  13. 2020Addressing Some Limitations of Transformers with Feedback Memorycited by 8
  14. 2025Limitations of Normalization in Attention Mechanismcited by 7
  15. 2025On the locality bias and results in the Long Range Arenacited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Mathematical Formalism for Memory Compression in Selective State Space Models
  2. 2026Expressivity-Efficiency Tradeoffs for Hybrid Sequence Models
  3. 2025Advancing Intelligent Sequence Modeling: Evolution, Trade-offs, and Applications of State-Space Architectures from S4 to Mamba
  4. 2026MARCH: Scaling Recurrent Memory with Content-Routed State Anchors
  5. 2026DART: Decoded Attention over Recurrent States for Efficient Long-Context Sequence Modeling
  6. 2025Forgetting Transformer: Softmax Attention with a Forget Gate
  7. 2026SchemaMem: Schema-Indexed Recurrent Memory for Delayed State Retrieval
  8. 2024Theoretical Foundations of Deep Selective State-Space Models