arXiv:2507.06607cs.CLcs.LG2025-07NeurIPS被引 12

提出新型混合架构SambaY,用记忆共享机制提升长文本推理效率。

Decoder-Hybrid-Decoder Architecture for Efficient Reasoning with Long Generation

  • 引入门控记忆单元(GMU),实现跨层记忆状态高效共享
  • 在2K上下文下推理速度最高提升10倍,且无需显式位置编码
  • 适用于大规模算力下的长序列推理任务,尤其适合数学与知识问答

近期语言建模进展表明状态空间模型(SSMs)在高效序列建模中表现优异。尽管如Samba和YOCO等混合架构已在性能上超越Transformer,但此前研究未探索SSM层间表示共享的效率潜力。本文提出门控记忆单元(GMU),一种简单而有效的跨层记忆共享机制,并构建SambaY——一种基于Samba的自解码器与交叉解码器构成的解码器-混合-解码器架构,通过在跨解码器中引入GMU共享记忆读出状态。SambaY显著提升解码效率,保持线性预填充时间复杂度,增强长上下文表现,同时无需显式位置编码。大规模扩展实验表明,本模型相比强基线YOCO展现出更低的不可缩减损失,表明在大算力条件下具备更优可扩展性。最大模型结合微分注意力(Differential Attention)后,在不使用强化学习的情况下,于Math500、AIME24/25和GPQA Diamond等推理任务上优于Phi4-mini-Reasoning,且在vLLM推理框架下,2K长度提示、32K生成长度时达到最高10倍的解码吞吐量提升。训练代码已开源:https://github.com/microsoft/ArchScale。

原文摘要 · Abstract (English)

Recent advances in language modeling have demonstrated the effectiveness of State Space Models (SSMs) for efficient sequence modeling. While hybrid architectures such as Samba and the decoder-decoder architecture, YOCO, have shown promising performance gains over Transformers, prior works have not investigated the efficiency potential of representation sharing between SSM layers. In this paper, we introduce the Gated Memory Unit (GMU), a simple yet effective mechanism for efficient memory sharing across layers. We apply it to create SambaY, a decoder-hybrid-decoder architecture that incorporates GMUs in the cross-decoder to share memory readout states from a Samba-based self-decoder. SambaY significantly enhances decoding efficiency, preserves linear pre-filling time complexity, and boosts long-context performance, all while eliminating the need for explicit positional encoding. Through extensive scaling experiments, we demonstrate that our model exhibits a significantly lower irreducible loss compared to a strong YOCO baseline, indicating superior performance scalability under large-scale compute regimes. Our largest model enhanced with Differential Attention, Phi4-mini-Flash-Reasoning, achieves significantly better performance than Phi4-mini-Reasoning on reasoning tasks such as Math500, AIME24/25, and GPQA Diamond without any reinforcement learning, while delivering up to 10x higher decoding throughput on 2K-length prompts with 32K generation length under the vLLM inference framework. We release our training codebase on open-source data at https://github.com/microsoft/ArchScale.

长序列推理状态空间模型高效解码混合架构

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