arXiv:2603.15569cs.LG2026-03被引 74

Mamba-3通过状态空间模型改进序列建模,兼顾高效推理与模型性能。

Mamba-3: Improved Sequence Modeling using State Space Principles

  • 基于状态空间模型设计更丰富的递归机制,提升序列建模能力。
  • 1.5B规模下比最优模型高出0.6个百分点,MIMO变体再提升1.2点,总增1.8点。
  • 仅用一半状态大小即达Mamba-2水平,适合追求高效推理的部署场景。

提升推理时计算量已成为推动大语言模型性能的关键因素,使推理效率成为模型设计的核心关注点。尽管当前基于Transformer的模型具备强大性能,但其二次计算复杂度和线性内存开销导致推理成本高昂。这促使人们开发出具有次二次计算复杂度和恒定内存需求的模型。然而,许多近期线性模型为追求算法效率牺牲了模型质量和能力,在状态追踪等任务上表现不佳,且实际硬件效率仍不理想。受‘推理优先’理念驱动,我们提出三项受状态空间模型(SSM)启发的核心方法改进:(1) 基于SSM离散化推导出更具表达力的递归结构;(2) 采用复数状态更新规则以增强状态追踪能力;(3) 引入多输入多输出(MIMO)架构,在不增加解码延迟的前提下提升性能。结合架构优化,Mamba-3在检索、状态追踪及下游语言建模任务中均取得显著提升。在1.5B参数规模下,平均下游准确率比最优模型Gated DeltaNet高0.6个百分点,其MIMO变体进一步提升1.2个百分点,总计提升1.8个百分点。在不同状态尺寸实验中,Mamba-3在仅使用一半状态规模的情况下,达到与Mamba-2相当的困惑度。评估结果表明,Mamba-3有效推进了性能-效率的帕累托前沿。

原文摘要 · Abstract (English)

Scaling inference-time compute has emerged as an important driver of LLM performance, making inference efficiency a central focus of model design alongside model quality. While the current Transformer-based models deliver strong model quality, their quadratic compute and linear memory make inference expensive. This has spurred the development of sub-quadratic models with reduced linear compute and constant memory requirements. However, many recent linear models trade off model quality and capability for algorithmic efficiency, failing on tasks such as state tracking. Moreover, their theoretically linear inference remains hardware-inefficient in practice. Guided by an inference-first perspective, we introduce three core methodological improvements inspired by the state space model (SSM) viewpoint of linear models. We combine: (1) a more expressive recurrence derived from SSM discretization, (2) a complex-valued state update rule that enables richer state tracking, and (3) a multi-input, multi-output (MIMO) formulation for better model performance without increasing decode latency. Together with architectural refinements, our Mamba-3 model achieves significant gains across retrieval, state-tracking, and downstream language modeling tasks. At the 1.5B scale, Mamba-3 improves average downstream accuracy by 0.6 percentage points compared to the next best model (Gated DeltaNet), with Mamba-3's MIMO variant further improving accuracy by another 1.2 points for a total 1.8 point gain. Across state-size experiments, Mamba-3 achieves comparable perplexity to Mamba-2 despite using half of its predecessor's state size. Our evaluations demonstrate Mamba-3's ability to advance the performance-efficiency Pareto frontier.

序列建模状态空间推理效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。