针对小模型部署延迟优化,提出新型混合架构Nemotron-Flash。
Nemotron-Flash: Towards Latency-Optimal Hybrid Small Language Models
- 通过分析深度宽度比与算子选择,定位延迟关键因素。
- 实现比Qwen3-1.7B/0.6B低1.3~1.9倍延迟,高18.7~45.6倍吞吐。
- 适合对响应速度要求高的边缘计算、实时交互场景。
小语言模型(SLMs)的高效部署对诸多具有严格延迟约束的实际应用至关重要。以往研究多聚焦于减少参数量以实现参数最优,但参数效率并不必然带来实际设备上的速度提升。本文旨在识别影响SLMs真实设备延迟的关键因素,并为以延迟为核心目标的SLM设计与训练提供可泛化的原则与方法。具体而言,我们发现两个核心架构因素:深度-宽度比和算子选择。前者影响小批量延迟,后者同时影响延迟与大批量吞吐。基于此,我们研究了延迟最优的深度-宽度比,发现虽深而薄模型在相同参数预算下通常更准确,却未必位于精度-延迟权衡前沿。接着,我们评估了新兴高效注意力机制作为候选算子的潜力。利用筛选出的优质算子,构建进化搜索框架,自动发现混合SLM中算子组合的延迟最优解,从而推进精度-延迟前沿。此外,通过引入权重归一化技术改进训练过程,提升权重更新效率并改善最终收敛。综合上述方法,我们提出新一代混合小模型Nemotron-Flash,显著超越现有SOTA SLM的精度-效率前沿,例如相比Qwen3-1.7B/0.6B,在平均精度上提升超5.5%,延迟降低1.3倍/1.9倍,吞吐量提升18.7倍/45.6倍。
原文摘要 · Abstract (English)
Efficient deployment of small language models (SLMs) is essential for numerous real-world applications with stringent latency constraints. While previous work on SLM design has primarily focused on reducing the number of parameters to achieve parameter-optimal SLMs, parameter efficiency does not necessarily translate into proportional real-device speed-ups. This work aims to identify the key determinants of SLMs' real-device latency and offer generalizable principles and methodologies for SLM design and training when real-device latency is the primary consideration. Specifically, we identify two central architectural factors: depth-width ratios and operator choices. The former is crucial for small-batch-size latency, while the latter affects both latency and large-batch-size throughput. In light of this, we first study latency-optimal depth-width ratios, with the key finding that although deep-thin models generally achieve better accuracy under the same parameter budget, they may not lie on the accuracy-latency trade-off frontier. Next, we explore emerging efficient attention alternatives to evaluate their potential as candidate building operators. Using the identified promising operators, we construct an evolutionary search framework to automatically discover latency-optimal combinations of these operators within hybrid SLMs, thereby advancing the accuracy-latency frontier. In addition to architectural improvements, we further enhance SLM training using a weight normalization technique that enables more effective weight updates and improves final convergence. Combining these methods, we introduce a new family of hybrid SLMs, called Nemotron-Flash, which significantly advances the accuracy-efficiency frontier of state-of-the-art SLMs, e.g., achieving over +5.5% average accuracy, 1.3x/1.9x lower latency, and 18.7x/45.6x higher throughput compared to Qwen3-1.7B/0.6B, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。