arXiv:2507.19427cs.LGcs.AI2025-07被引 58

321B参数模型通过软硬件协同设计,实现低成本高效解码。

Step-3 is Large yet Affordable: Model-system Co-design for Cost-effective Decoding

  • 采用多矩阵分解注意力机制,减少显存占用和计算量。
  • 每令牌激活380亿参数,解码速度达每卡4039词元/秒。
  • 适合追求高吞吐、低延迟的长文本生成场景。

大语言模型在解码阶段面临硬件效率低的问题,尤其在长上下文推理任务中。本文提出3210亿参数的视觉语言模型Step-3,通过面向硬件的软硬件协同设计,显著降低解码成本。该模型在两个关键维度创新:(1) 提出新型多矩阵分解注意力(MFA)机制,大幅降低键值缓存大小与计算开销,同时保持高注意力表达能力;(2) 设计注意力-前馈网络拆分(AFD)分布式推理系统,将注意力层与前馈网络分离至专用子系统。这种协同设计实现前所未有的成本效益:相较于DeepSeek-V3和Qwen3 MoE 235B,Step-3在更长上下文下优势持续扩大。其每令牌激活380亿参数,性能优于同类模型。在Hopper GPU上,4096上下文、FP8精度、50毫秒端到端延迟(TPOT SLA)条件下,解码吞吐达每卡4039词元/秒,超越DeepSeek-V3的2324,刷新大模型解码性能前沿。

原文摘要 · Abstract (English)

Large language models (LLMs) face low hardware efficiency during decoding, especially for long-context reasoning tasks. This paper introduces Step-3, a 321B-parameter VLM with hardware-aware model-system co-design optimized for minimizing decoding costs. Step-3 innovates in two key dimensions: (1) A novel Multi-Matrix Factorization Attention (MFA) mechanism that significantly reduces both KV cache size and computation while maintaining high attention expressiveness, and (2) Attention-FFN Disaggregation (AFD), a distributed inference system that decouples attention and Feed-Forward Network (FFN) layers into specialized subsystems. This co-design achieves unprecedented cost efficiency: Step-3 significantly reduces theoretical decoding costs compared with models like DeepSeek-V3 and Qwen3 MoE 235B, with the gains widening at longer context. Step-3 achieves low cost while activating 38B parameters per token (more than DeepSeek-V3 and Qwen3 MoE 235B), demonstrating that hardware-aligned attention arithmetic intensity, MoE sparsity, and AFD are critical to cost-effectiveness. We perform a head-to-head comparison with DeepSeek-V3 in its favorable scenarios. Our implementation on Hopper GPUs achieves a decoding throughput of up to 4,039 tokens per second per GPU under 50ms TPOT SLA (4K context, FP8, no MTP). It is higher than DeepSeek-V3's 2,324 in the same setup and sets a new Pareto frontier for LLM decoding.

大模型解码优化硬件协同推理加速

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