PRISM: Parallel Residual Iterative Sequence Model
Not on the map. Why?
Abstract (arXiv)
Generative sequence modeling faces a fundamental tension between the expressivity of Transformers and the efficiency of linear sequence models. Existing efficient architectures are theoretically bounded by shallow, single-step linear updates, while powerful iterative methods like Test-Time Training (TTT) break hardware parallelism due to two dimensions of serial dependency: token-level state reliance and step-level iteration loops. We propose PRISM (Parallel Residual Iterative Sequence Model) to resolve this tension. PRISM explicitly approximates the expressive gate-residual-direction iteration pattern of TTT in a parallelizable form. We employ a Write-Forget Decoupling strategy that isolates non-linearity within the injection operator. To bypass the serial dependency of explicit solvers, PRISM utilizes a two-stage proxy architecture: a short-convolution anchors the initial residual using local history energy, while a learned predictor estimates the refinement updates directly from the input. This design distills structural patterns associated with iterative correction into a parallelizable feedforward operator. Theoretically, we prove that this formulation achieves Rank-L accumulation, structurally expanding the update scheme beyond the single-step Rank-1 bottleneck. Empirically, it achieves comparable performance to explicit optimization methods while achieving 174x higher throughput. Codes are available in https://github.com/gpr-prism/prism/.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 25 of its 32 references that are in the map. First posted on arXiv 11 Feb 2026.
Builds on
25 of 32 references.
References · 25
- 2017Attention Is All You Needcited by 59,932
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
- 2018Self-Attentive Sequential Recommendationcited by 1,169
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 390
13 more
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2024Titans: Learning to Memorize at Test Timecited by 303
- 2024Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendationscited by 184
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 169
- 2025OneRec: Unifying Retrieve and Rank with Generative Recommender and Iterative Preference Alignmentcited by 167
- 2025RWKV-7 "Goose" with Expressive Dynamic State Evolutioncited by 134
- 2025Test-Time Training Done Rightcited by 121
- 2024Gated Slot Attention for Efficient Linear-Time Sequence Modelingcited by 73
- 2025It's All Connected: A Journey Through Test-Time Memorization, Attentional Bias, Retention, and Online Optimizationcited by 60
- 2025ATLAS: Learning to Optimally Memorize the Context at Test Timecited by 57
- 2025MoM: Linear Sequence Modeling with Mixture-of-Memoriescited by 44
- 2025Exact Flow Linear Attention: Exact Solution from Continuous-Time Dynamicscited by 4
Built on by
Not cited yet.