Llama-Nemotron:高效推理模型

Llama-Nemotron: Efficient Reasoning Models

Akhiad Bercovich Akhiad Bercovich · NVIDIA · 2025-05-02 · arXiv:2505.00949 ↗ · 被引 86

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们推出了 Llama-Nemotron 系列模型,这是一个开放的异构推理模型家族,具备卓越的推理能力、推理效率,并采用适合企业使用的开放许可。该系列包含三种尺寸——Nano(8B)、Super(49B)和 Ultra(253B)——在与 DeepSeek-R1 等最先进的推理模型竞争的同时,提供了更优的推理吞吐量和内存效率。在本报告中,我们讨论了这些模型的训练流程,包括使用神经架构搜索从 Llama 3 模型加速推理、知识蒸馏和持续预训练,随后是专注于推理的后训练阶段,该阶段由两个主要部分组成:监督微调和大规模强化学习。Llama-Nemotron 模型是首批支持动态推理切换的开源模型,允许用户在推理过程中在标准聊天模式和推理模式之间切换。为了进一步支持开放研究和促进模型开发,我们提供了以下资源:1. 我们根据商业许可的 NVIDIA 开放模型许可协议发布了 Llama-Nemotron 推理模型——LN-Nano、LN-Super 和 LN-Ultra。2. 我们发布了完整的后训练数据集:Llama-Nemotron-Post-Training-Dataset。3. 我们还发布了我们的训练代码库:NeMo、NeMo-Aligner 和 Megatron-LM。

We introduce the Llama-Nemotron series of models, an open family of heterogeneous reasoning models that deliver exceptional reasoning capabilities, inference efficiency, and an open license for enterprise use. The family comes in three sizes -- Nano (8B), Super (49B), and Ultra (253B) -- and performs competitively with state-of-the-art reasoning models such as DeepSeek-R1 while offering superior inference throughput and memory efficiency. In this report, we discuss the training procedure for these models, which entails using neural architecture search from Llama 3 models for accelerated inference, knowledge distillation, and continued pretraining, followed by a reasoning-focused post-training stage consisting of two main parts: supervised fine-tuning and large scale reinforcement learning. Llama-Nemotron models are the first open-source models to support a dynamic reasoning toggle, allowing users to switch between standard chat and reasoning modes during inference. To further support open research and facilitate model development, we provide the following resources: 1. We release the Llama-Nemotron reasoning models -- LN-Nano, LN-Super, and LN-Ultra -- under the commercially permissive NVIDIA Open Model License Agreement. 2. We release the complete post-training dataset: Llama-Nemotron-Post-Training-Dataset. 3. We also release our training codebases: NeMo, NeMo-Aligner, and Megatron-LM.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 21)

阅读逐段中英对照全文 →