arXiv Computation and Language

Multi-State Linear Attention with Generalized Second-Order Routing

Multi-State Linear Attention (MSLA) improves long‑sequence modeling by compressing context into separate memory states, but its aggregation only uses mean alignment, ignoring distribution within segments. MERLIN (MomEnt Routing for multi‑state LINear attention) enhances this by tracking first and second moments of each segment’s keys, enabling queries to recover both mean and variance of alignment for more content‑aware aggregation. Experiments on Gated DeltaNet‑1.3B and Mamba‑2‑780M demonstrate that MERLIN consistently outperforms existing aggregation methods on long‑context tasks, achieving significant accuracy gains while using less memory.

Hugging Face Trending Papers
Oct 27

Multi-State Linear Attention with Generalized Second-Order Routing

Multi-State Linear Attention (MSLA) improves long‑sequence modeling by compressing context into separate memory states, but its aggregation only uses mean alignment, ignoring distribution. MERLIN (MomEnt Routing for multi‑state LINear attention) enhances this by tracking first and second moments of each segment’s keys, enabling queries to recover both mean and variance of alignment for more content‑aware aggregation. Experiments on Gated DeltaNet‑1.3B and Mamba‑2‑780M show MERLIN consistently outperforms existing aggregation methods on long‑context tasks, achieving significant gains on RULER and LongBench while using less memory.

arXiv AI
Oct 12

MS-GLA: Multi-Scale Gated Linear Attention for Addressing Representational Bottlenecks via Multi-Temporal Resolution

MS-GLA introduces Multi-Scale Gated Linear Attention, distributing attention heads across multiple temporal resolutions to address representational bottlenecks in GLA Transformers. Coarser resolutions pool longer token spans for long‑range dependencies, while finer heads capture local syntax, and a learnable fusion layer recombines outputs dynamically. Experiments on language modeling, recall‑intensive tasks, and long‑context generalization show MS‑GLA outperforms GLA with up to 18.9% improvement on recall tasks and 9.5% lower perplexity on language modeling benchmarks.

By Prasoon Dev, Anirudh Sankar, Vasudeva Varma
arXiv Machine Learning
Aug 31

Sliding-window beats linear attention

arXiv:2608.28444v1 Announce Type: cross Abstract: Due to the nature of quadratic attention, Large Language Models (LLMs) consume a lot of memory and energy. Every new token costs more than the previo...

By Alexia Jolicoeur-Martineau, Rhea Sanjay Sukthanker, Pashmina Cameron, Emy Gervais