2606.13392v1 Jun 11, 2026 cs.AI

MiniMax Sparse Attention

Yujia Liu
Yujia Liu
Citations: 0
h-index: 0
Pengyu Zhao
Pengyu Zhao
Citations: 74
h-index: 3
Yangyifan Xu
Yangyifan Xu
Citations: 95
h-index: 5
Qiaorui Chen
Qiaorui Chen
Citations: 19
h-index: 3
Haichao Zhu
Haichao Zhu
Harbin Institute of Technology
Citations: 694
h-index: 12
Xunhao Lai
Xunhao Lai
Citations: 306
h-index: 4
Weiqi Xu
Weiqi Xu
Citations: 1,033
h-index: 10
Yufeng Yang
Yufeng Yang
Citations: 9
h-index: 2
Lunbin Zeng
Lunbin Zeng
Citations: 12
h-index: 2
Haohai Sun
Haohai Sun
Citations: 314
h-index: 2
Vito Zhang
Vito Zhang
Citations: 0
h-index: 0

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. We introduce MiniMax Sparse Attention (MSA), a blockwise sparse attention built upon Grouped Query Attention (GQA). A lightweight Index Branch scores key-value blocks and independently selects a Top-k subset for each GQA group, enabling group-specific sparse retrieval while maintaining efficient block-level execution; the Main Branch then performs exact block-sparse attention over only the selected blocks. Designed around a principle of simplicity and scalability, MSA is deliberately streamlined, making it straightforward to deploy efficiently across a broad range of GPUs. To translate sparsity into practical speedups, we co-design MSA with a GPU execution path that uses exp-free Top-k selection and KV-outer sparse attention to improve tensor-core utilization under block-granular access. On a 109B-parameter model with native multimodal training, MSA performs on par with GQA while reducing per-token attention compute by 28.4x at 1M context. Paired with our co-designed kernel, MSA achieves 14.2x prefill and 7.6x decoding wall-clock speedups on H800. Our inference kernel is available at: https://github.com/MiniMax-AI/MSA. A production-grade natively multimodal model powered by MSA has been publicly released at: https://huggingface.co/MiniMaxAI/MiniMax-M3.

0 Citations
0 Influential
55.085555799816 Altmetric
275.4 Score
Original PDF
335

No Analysis Report Yet

This paper hasn't been analyzed by Gemini yet.

Log in to request an AI analysis.

댓글

댓글을 작성하려면 로그인하세요.

아직 댓글이 없습니다. 첫 번째 댓글을 남겨보세요!