arXiv:2508.14444cs.CLcs.AI2025-08被引 79

NVIDIA新模型融合Mamba与Transformer,推理速度提升6倍且精度领先。

NVIDIA Nemotron Nano 2: An Accurate and Efficient Hybrid Mamba-Transformer Reasoning Model

  • 用Mamba替代大部分注意力层,提升长思考链生成效率。
  • 在8k输入16k输出场景下,推理吞吐量最高达同类模型6倍。
  • 支持单卡128k上下文,适合资源受限环境的高效推理任务。

我们提出Nemotron-Nano-9B-v2,一种混合Mamba-Transformer语言模型,旨在提升推理工作负载的吞吐量,同时在同等规模模型中达到顶尖准确率。该模型基于Nemotron-H架构,将通用Transformer中的多数自注意力层替换为Mamba-2层,以加速生成长推理轨迹时的推理速度。我们首先在20万亿个标记上使用FP8训练配方预训练了一个120亿参数模型(Nemotron-Nano-12B-v2-Base)。对齐后,采用Minitron策略进行压缩与蒸馏,目标是在单块NVIDIA A10G GPU(22GiB显存,bfloat16精度)上实现高达128k tokens的推理能力。相比同类模型(如Qwen3-8B),Nemotron-Nano-9B-v2在推理基准测试中表现持平或更优,且在8k输入16k输出的推理场景中,吞吐量最高提升6倍。我们已将Nemotron-Nano-9B-v2、Nemotron-Nano-12B-v2-Base及Nemotron-Nano-9B-v2-Base模型权重,以及大部分预训练与后训练数据集发布至Hugging Face。

原文摘要 · Abstract (English)

We introduce Nemotron-Nano-9B-v2, a hybrid Mamba-Transformer language model designed to increase throughput for reasoning workloads while achieving state-of-the-art accuracy compared to similarly-sized models. Nemotron-Nano-9B-v2 builds on the Nemotron-H architecture, in which the majority of the self-attention layers in the common Transformer architecture are replaced with Mamba-2 layers, to achieve improved inference speed when generating the long thinking traces needed for reasoning. We create Nemotron-Nano-9B-v2 by first pre-training a 12-billion-parameter model (Nemotron-Nano-12B-v2-Base) on 20 trillion tokens using an FP8 training recipe. After aligning Nemotron-Nano-12B-v2-Base, we employ the Minitron strategy to compress and distill the model with the goal of enabling inference on up to 128k tokens on a single NVIDIA A10G GPU (22GiB of memory, bfloat16 precision). Compared to existing similarly-sized models (e.g., Qwen3-8B), we show that Nemotron-Nano-9B-v2 achieves on-par or better accuracy on reasoning benchmarks while achieving up to 6x higher inference throughput in reasoning settings like 8k input and 16k output tokens. We are releasing Nemotron-Nano-9B-v2, Nemotron-Nano12B-v2-Base, and Nemotron-Nano-9B-v2-Base checkpoints along with the majority of our pre- and post-training datasets on Hugging Face.

混合架构推理加速大模型压缩Mamba

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