arXiv:2505.15431cs.CL2025-05被引 26

混合Mamba与Transformer,实现高效长文本推理的大型语言模型

Hunyuan-TurboS: Advancing Large Language Models through Mamba-Transformer Synergy and Adaptive Chain-of-Thought

  • 融合Mamba长序列处理与Transformer上下文理解优势,动态切换快速响应与深度思考模式
  • 560亿参数模型支持256K上下文长度,在23项基准测试中平均得分77.9%
  • 首个工业级大规模Mamba模型,推理成本低于同类推理模型,适合高并发场景

随着大语言模型的快速发展,我们提出Hunyuan-TurboS——一种新型混合Transformer-Mamba的专家混合(MoE)大模型。该模型协同利用Mamba在长序列处理中的高效性与Transformer在上下文理解上的优越性。其采用自适应长短链式思维(CoT)机制,根据问题复杂度动态切换快速响应与深度思考模式,优化计算资源消耗。架构上,该560亿总参数、560亿激活参数的模型包含128层(Mamba2、注意力、前馈网络),采用创新的AMF/MF块结构。更快的Mamba2实现线性复杂度,分组查询注意力减少键值缓存,前馈网络使用MoE结构。在16万亿高质量令牌上预训练,支持256K上下文长度,是首个部署于工业界的大型Mamba模型。通过监督微调(300万指令)、自适应长短链融合、多轮反思学习及两阶段大规模强化学习,显著提升能力,尤其在STEM和通用指令遵循任务上表现优异。评估显示,其在LMSYS Chatbot Arena中排名前七,得分为1356,优于Gemini-2.0-Flash-001(1352)和o4-mini-2025-04-16(1345)。TurboS在23项自动化基准测试中平均得分77.9%。该模型在性能与效率间取得良好平衡,以更低推理成本提供强大能力,为高效大规模预训练模型树立新范式。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) rapidly advance, we introduce Hunyuan-TurboS, a novel large hybrid Transformer-Mamba Mixture of Experts (MoE) model. It synergistically combines Mamba's long-sequence processing efficiency with Transformer's superior contextual understanding. Hunyuan-TurboS features an adaptive long-short chain-of-thought (CoT) mechanism, dynamically switching between rapid responses for simple queries and deep "thinking" modes for complex problems, optimizing computational resources. Architecturally, this 56B activated (560B total) parameter model employs 128 layers (Mamba2, Attention, FFN) with an innovative AMF/MF block pattern. Faster Mamba2 ensures linear complexity, Grouped-Query Attention minimizes KV cache, and FFNs use an MoE structure. Pre-trained on 16T high-quality tokens, it supports a 256K context length and is the first industry-deployed large-scale Mamba model. Our comprehensive post-training strategy enhances capabilities via Supervised Fine-Tuning (3M instructions), a novel Adaptive Long-short CoT Fusion method, Multi-round Deliberation Learning for iterative improvement, and a two-stage Large-scale Reinforcement Learning process targeting STEM and general instruction-following. Evaluations show strong performance: overall top 7 rank on LMSYS Chatbot Arena with a score of 1356, outperforming leading models like Gemini-2.0-Flash-001 (1352) and o4-mini-2025-04-16 (1345). TurboS also achieves an average of 77.9% across 23 automated benchmarks. Hunyuan-TurboS balances high performance and efficiency, offering substantial capabilities at lower inference costs than many reasoning models, establishing a new paradigm for efficient large-scale pre-trained models.

大模型Mamba推理优化MoE

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