arXiv AI
Jun 12

MiniMax Sparse Attention

arXiv:2606. 13392v1 Announce Type: new Abstract: Ultra-long-context capability is becoming indispensable for frontier LLMs: agentic workflows, repository-scale code reasoning, and persistent memory all require the model to jointly attend over hundreds of thousands to millions of tokens, yet the quadratic cost of softmax attention makes this untenable at deployment scale.

By Xunhao Lai, Weiqi Xu, Yufeng Yang, Qiaorui Chen, Yang Xu, Lunbin Zeng, Xiaolong Li, Haohai Sun, Haichao Zhu, Vito Zhang, Pengyu Zhao
arXiv Machine Learning
2d ago

Muir: Linear Attention with Sensitivity-Allocated Gating and Matched-Filter Readout

Muir is a new linear‑attention architecture that consolidates gating mechanisms into a low‑rank projection and uses a matched‑filter readout to re‑weight state rows. It improves over GDN‑2 and Transformers across a wide range of model sizes, lowering perplexity, boosting commonsense accuracy, and enhancing extractive recall while maintaining comparable throughput. The design also includes a numerically stable parallel update for efficient training on large chunk sizes.

By Tushaar Gangavarapu, Gil Tabak, Clemens JS Schaefer
arXiv AI
Jun 19

StreamKL: Fast and Memory-Efficient KL Divergence for Boosting Attention Distillation

arXiv:2606. 20005v1 Announce Type: cross Abstract: Attention distillation, which trains one attention distribution to match another by minimizing their Kullback-Leibler (KL) divergence, is widely used in knowledge distillation, model compression, continual learning, and sparse-attention LLM training.

By Guangda Liu, Yiquan Wang, Chengwei Li, Wenhao Chen, Jing Lin, Yiwu Yao, Danning Ke, Wenchao Ding, Jieru Zhao