mlsanityData to 14 Oct 2026

Naju: A Native Discrete State-Space Model with Independent Retention and Writing for Long-Sequence Memory

arXiv 2607.21000 ↗First posted 23 Jul 2026cs.AIcited by 0

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

Abstract (arXiv)

Long-sequence memory tracking places two opposing demands on a recurrent state: near-lossless retention of stored bindings over long horizons, and active overwriting of stale ones. In our diagnostic suite, the strongest efficient baselines tend to solve only one side well. Continuous-time-parameterized state-space models (SSMs) such as Mamba obtain their discrete recurrence by zero-order-hold discretization of a continuous-time system; we argue that this detour is unnecessary for memory tracking and parameterize the discrete transition directly. Naju (Native Adaptive Junction Unit) factorizes the recurrent update, schematically xn = fnodot xn-1 + inodot(Bn un), into an explicit discrete pole (a learned forget gate fn), an independent write gain in, and input-dependent write/read maps. Since the sigmoid pole satisfies 0<fn<1, each frozen local coordinate is Schur-stable by construction, and the full time-varying recurrence satisfies a fading-memory/BIBO bound under uniform boundedness assumptions, with no stability regularizer. We formalize the key structural limitation of coupled designs: any non-expansive complementary single-gate recurrence ties the effective retention r and write gain w through |r|+w≤ 1, so near-complete retention forces weak writing; decoupling fn from in removes this constraint. Empirically, Naju is the only evaluated model that remains strong on both retention and overwriting at 4x the training length. Beyond the diagnostic suite, we evaluate Naju on WikiText-103 language modeling, Long Range Arena, and multi-query associative recall. Across these settings, Naju consistently combines strong long-range memory with competitive or superior performance, outperforming the Mamba baselines in the principal comparisons while remaining competitive with the Transformer and preserving linear-time, linear-memory scaling.

Key facts. No citing papers in the map yet (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 21 of its 25 references that are in the map. First posted on arXiv 23 Jul 2026.

Builds on

21 of 25 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 · 16

  1. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  2. 2016Pointer Sentinel Mixture Modelscited by 3,096
  3. 2020GLU Variants Improve Transformercited by 1,804
  4. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
  5. 2016Language Modeling with Gated Convolutional Networkscited by 702
  6. 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
  7. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  8. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
  9. 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 408
  10. 2024xLSTM: Extended Long Short-Term Memorycited by 327
  11. 2022Transformers Learn Shortcuts to Automatacited by 268
  12. 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 242
4 more
  1. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
  2. 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrencecited by 159
  3. 2023Hierarchically Gated Recurrent Neural Network for Sequence Modelingcited by 102
  4. 2024RecurrentGemma: Moving Past Transformers for Efficient Open Language Modelscited by 64

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Length-Independent State Tracking Under a Parallel Scan
  2. 2026SHiPPO: Recurrent Memory with Transported Polynomial Projections
  3. 2025LISM: Long-range Integrative State space Models via Input-Latent State Interactions
  4. 2023State-space Models with Layer-wise Nonlinearity are Universal Approximators with Exponential Decaying Memory
  5. 2026Multi-Head Recurrent Memory Agents
  6. 2026Rethinking State Tracking in Recurrent Models Through Error Control Dynamics
  7. 2025Stateful KV Cache Management for LLMs: Balancing Space, Time, Accuracy, and Positional Fidelity
  8. 2026Fast Weight Attention for Continual Learning