arXiv:2510.26912cs.CL2025-10被引 5

解析混合模型记忆机制,提出增强长文本召回的训练方法。

Understanding and Enhancing Mamba-Transformer Hybrids for Memory Recall and Language Modeling

  • 对比串行与并行融合结构,发现不同上下文长度下各有优势。
  • 通过改写数据持续训练,显著提升召回能力且不损失其他性能。
  • 方法通用性强,适用于多种基线模型,优于复杂架构调整。

将状态空间模型(SSMs)与注意力机制结合的混合模型,凭借SSM的高效性与注意力的高召回能力表现出色,但其架构设计原理仍不清晰。本文从记忆利用与整体性能角度分析混合架构,发现串行融合在短上下文下表现更优,而并行融合在长上下文下更具优势。此外,提出一种以数据为中心的持续训练策略,通过在数据中加入改写样本进行训练,进一步提升召回能力,同时保持其他性能不变。该方法在不同基线模型上泛化良好,优于专门针对召回增强的架构修改。研究深化了对SSM-注意力混合模型的理解,为面向不同场景的模型设计提供了实用指导。

原文摘要 · Abstract (English)

Hybrid models that combine state space models (SSMs) with attention mechanisms have shown strong performance by leveraging the efficiency of SSMs and the high recall ability of attention. However, the architectural design choices behind these hybrid models remain insufficiently understood. In this work, we analyze hybrid architectures through the lens of memory utilization and overall performance, and propose a complementary method to further enhance their effectiveness. We first examine the distinction between sequential and parallel integration of SSM and attention layers. Our analysis reveals several interesting findings, including that sequential hybrids perform better on shorter contexts, whereas parallel hybrids are more effective for longer contexts. We also introduce a data-centric approach of continually training on datasets augmented with paraphrases, which further enhances recall while preserving other capabilities. It generalizes well across different base models and outperforms architectural modifications aimed at enhancing recall. Our findings provide a deeper understanding of hybrid SSM-attention models and offer practical guidance for designing architectures tailored to various use cases. Our findings provide a deeper understanding of hybrid SSM-attention models and offer practical guidance for designing architectures tailored to various use cases.

混合模型记忆召回训练策略

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