Introduction
- Nemotron-cascade 2
- Nemotron-cascade
- Nemotron-4
- Nemotron Nano
- Nemotron 3 Nano
- AceReason-nemotron
| Model | Year | Parameters | Attention | FFN | Training Tokens |
|---|---|---|---|---|---|
| Nemotron 3 Nano | 2025.12 | 30B-A3B | GQA + Mamba | MoE | 25T |
| Nemotron 3 Super | 2026.04 | 120B-A12B | GQA + Mamba | Latent MoE | 25T |
| Nemotron 3 Ultra | 2026.6 | 550B-A55B | GQA + Mamba | Latent MoE | 20T |
架构细节如下表所示
| Model | Nemotron 3 Nano | Nemotron 3 Super | Nemotron 3 Ultra |
|---|---|---|---|
| num layers | 52 | 88 | 108 |
| hidden size | 2688 | 4096 | 8192 |
| num attention heads | 32 | 32 | 64 |
| num key-value heads | 2 | 2 | 2 |
| head dim | 128 | 128 | 128 |
| mamba state dim | 128 | 128 | 128 |
| mamba groups | 8 | 8 | 8 |
| mamba heads | 64 | 128 | 256 |
| mamba head dim | 64 | 64 | 64 |
| expert hidden size | 1856 | 2688 | 5120 |
| total experts | 128 | 512 | 512 |
| activated experts | 6 | 22 | 22 |
| shared experts | 2 | 2 | 2 |
| MoE latent size | - | 1024 | 2048 |
| MTP layers | - | 2 | 2 |
Nemotron 3 Nano
Nvidia 在 25 年 12 月发布了 Nemotron 3 Nano (NVIDIA et al., 2025), 一个基于 Mamba-2 (Dao & Gu, 2024) 和 GQA (Ainslie et al., 2023) 混合注意力,以及 MoE 架构的大语言模型。 其中,MoE 模块包含 128 个专家,激活 6 个专家,参数量为 31.6B-A3.2B.
Architecture
Nemotron 3 Nano 的架构基于 Nemotron-H 和 Nemotron 2 Nano,
Data
Pre-training Data
Pre-training data 包括以下几个部分:
- Nemotron-CC-Code-v1: 基于 Nemotron-CC-Math pipeline 构建了高质量 code pre-training corpus, 使用了 Lynx + LLM pipeline 来进行过滤和优化,最终包含 428B tokens.
- Nemotron-Pretraining-Code-v2: 基于 Nemotron 2 Nano 从 Github 构建了 source-code corpus. 还基于 Nemotron-H 的策略使用 Qwen3-32B 来合成 QA pairs. 作者还是用了 code-rewriting 来提高代码数据的质量
- Nemotron-CC-v2.1: 基于 Nemotron-CC recipe 来构建 CC web crawl data, 加入了 2.1T 新 token.
- Nemotron-Pretraining-Specialized-v1: 包括 Wikipedia, Math Textbook, Scientific Coding, cross-domain code, STEAM reasoning 和一些 SFT 风格的数据,作者使用了 Nemo Data Designer 来实现 data generation pipeline.
最终,pre-training 2 个 phase 的数据配比如下图所示,其中 phase 1 的数据占比为 .
SFT
SFT 阶段的数据包含以下部分:
- math and code: 基于 Nemotron 2 Nano
- tool use
- long context: 128k
- formal proof: 基于社区数据,包含 300k samples
- multilingual: 基于 Nemotron 2 Nano
- terminal use: 基于 Terminal Bench 构造,作者还使用了 SWE-Smith 的数据
- general chat: LMSYS 和 WildChat
- instruction following: 基于 Tulu3 和 IFEval, IFBench
- safety: Nemotron Content Safety v2, Gretel Safety Alignment v1, Harmful Tasks 和 Red-Team-2K
- SWE: 使用 SWE-Gym 和 R2E-Gym 数据集,基于 OpenHands, SWE-Agent 和 Mini-SWE-Agent 来蒸馏轨迹
- Science: 基于 Nemotron 2 Nano
- GenSelect
- CUDA: 基于 HuggingFace Transformers 和 Kernelbook 进行合成
最终 SFT 阶段一共包括 18M samples, 数据配比如下图所示
为了提高模型的动态思考能力,作者采用了与 Nemotron 2 Nano 类似的方法:将 reasoning 数据的 CoT 丢弃掉,将 reasoning 数据根据不同的 reasoning budget 进行截断。
RL
作者使用了 unified RL 的训练方式,数据包括:
- math: DAPO [@DAPO] 和 Skyworks math, 分别有 17K tasks 和 104K tasks
- coding: OpenCodeReasoning, 22K tasks
- QA: 135K tasks
- structured output: 9k tasks, 提高模型的 json schema adherence 能力
- instruction following: 基于 IFEval 和 IFBench, 46K tasks
- long context: multi-document QA, 32k context length, 12K tasks
- agentic tool use: 包含两个个环境,一个是 multi-step verifiable tool-calling env, 使用了 Workplace Assistance, 26 tools, 690 tasks. 另一个是 multi-turn env, 包含 1k tasks.
RLHF 的训练数据主要来原始 HelpSteer3
Training
Pre-training
Pre-training 阶段作者使用了 Warm-Stable-Decay learning rate (8.4B warmup, 20T stable, 5T decay), context length 为 8192, batch size 为 3072.
与 Nemotron 2 Nano 一致,作者加入了一个 long-context phase 来提高模型的长上下文能力,该阶段的数据配比为, document QA: long context data: phase 2 data = 20: 1: 79. 这个阶段使用了 121B tokens.
Post-training
Post-training 分为 SFT 和 RL 两个阶段。 作者将 reasoning flow 进行了如下修改:
- 对于 multi-step 场景,保留模型的 reasoning token
- 对于 multi-turn 场景,一旦用户输入了新的内容,则会将之前的 reasoning content 丢弃掉
这个上下文管理方式与 DeepSeek-V3.2 一致。
对于工具调用,作者使用了 XML 风格的 tags 来减少 character escaping.
RL 阶段训练时,作者使用了 LLaMA-Nemotron 动态调整问题难度的方法。作者将 pass-rate 建模为一个 Gaussian function, 随训练进行逐渐调整问题难度。
RL 的算法为 synchronous GRPO 算法,作者还是用了 MIS 来解决训练推理不一致性。
对于 RLHF, 作者使用了 GenRM 来训练 reward model. 训练时的 reward 包括 ranking difference, helpfulness score, format score.
训练时,作者提出了 group relative length control 机制来减少 redundant thinking.
作者还提出了 PTQ 来进行量化,作者发现基于 post-training reasoning SFT dataset 来进行 calibration 效果更好。
作者通过实验发现,self-attention layers 最敏感,因此作者使用了 BF16 精度。对于 model weights, activations, KV cache, 作者使用了 FP8 精度。
Infra
RL 的训练基于 Nemo RL 和 Nemo Gym.
Experiments
- 第一个结果是 RLVR 的上限要比 SFT 更高
- 第二个结果是 FP8 quantization 的表现与 BF16 精度表现差不多
- FP8 KV cache 对 throughput 提升最明显,其他的 quantization 均会损害模型的性能
- Ainslie, J., Lee-Thorp, J., de Jong, M., Zemlyanskiy, Y., Lebron, F., & Sanghai, S. (2023). GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints. The 2023 Conference on Empirical Methods in Natural Language Processing. https://openreview.net/forum?id=hmOwOZWzYE
- Dao, T., & Gu, A. (2024). Transformers are SSMs: generalized models and efficient algorithms through structured state space duality. Proceedings of the 41st International Conference on Machine Learning.
- NVIDIA, :, Blakeman, A., Grattafiori, A., Basant, A., Gupta, A., Khattar, A., Renduchintala, A., Vavre, A., Shukla, A., Bercovich, A., Ficek, A., Shaposhnikov, A., Kondratenko, A., Bukharin, A., Milesi, A., Taghibakhshi, A., Liu, A., Barton, A., … Yan, Z. (2025). Nemotron 3 Nano: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning. https://arxiv.org/abs/2512.20848
Nemotron 3 Super
NVIDIA 在 2026年 4 月 提出了 Nemotron 3 Super, 一个基于 Mamba-2 (Dao & Gu, 2024) 和 GQA (Ainslie et al., 2023) 混合注意力,以及 MoE 架构的大语言模型。 其中,MoE 模块包含 512 个专家,激活 24 个专家,参数量为 120B-A12B.
相比于 Nemotron 3 Nano, Nemotron 3 Super 主要有三点改进:
Architecture
Nemotron 3 Super 的 Attention 采用了 Mamba 和 GQA 的混合架构,FFN 采用了 LatentMoE 架构。 与其他主流模型一致,MTP 也被用于提高 decoding 效率
关于 LatentMoE, 详情见 LatentMoE.
针对 MTP, Nemotron 3 Super 主要提出了在不同的 MTP heads 之间共享参数来实现变长 speculative decoding 的目标,实验结果发现模型的 acceptance length 比 DeepSeek-R1 (Guo et al., 2025) 和 Qwen3-Next (Team, 2025) 高。
关于 attention, Nemotron 3 Super 将 GQA 作为 global anchor 来提高模型的上下文交互能力
不同模块的精度如下表所示
| Layer Type | Format | Rationale |
|---|---|---|
| All Linear Layers Unless Otherwise Noted | NVFP4 | |
| Final 15% of Network | BF16 | Promote training stability at scale |
| Latent Projections | BF16 | Strategically kept in BF16 as step-time impact is negligible |
| MTP Layers | BF16 | Preserves multi-token prediction capabilities |
| QKV & Attention Projections | BF16 | Maintain fidelity of few attention layers |
| Mamba Output Projection | MXFP8 | Mitigates high incidence of underflows observed when quantizing this layer to NVFP4 at smaller scales |
| Embedding Layers | BF16 |
Data
Pre-training Data
预训练数据在 Nemotron 3 Nano 的基础上额外增加了
- code concepts, 基于 HumanEval 提取 91 code concepts, 然后使用 GPT-oss 来生成相应的编程问题,得到 14M problems, 再生成多样化的答案,得到 23M problem-solution pairs, 最后对数据进行过滤得到 15M problems.
- unconditional algorithmic data, 作者使用一些简单的问题来让模型生成 leetcode 风格的问题以及答案,最终得到 0.2B tokens
- economics, 合成了一批 MCQ 数据
- formal logic
- MCQ, 合成了 3.5M MMLU-style 的 MCQ samples, 大约有 1.6B tokens.
最终,pre-training 2 个 phase 的数据配比如下图所示,其中 phase 1 的数据占比为 .
最终 Nemotron 3 Super 的预训练使用了 25T token.
Post-training Data
SFT 阶段的数据也基于 Nemotron 3 Nano
- SWE, 从 SWE-Gym, R2E-Gym 和 SWE-rebench 数据集中收集 issue, 使用 Qwen3-Coder 在 Openhands agent harness 中来收集轨迹
- agentic programming, NeMo Data Designer 收集 20k queries, 使用 GPT-oss 120b 过滤任务,得到 15k tasks, 增加了 3k 的 SWE 相关任务和 10k web development 相关任务
- long context, concatenate documents 并合成 multi-hop reasoning QA
- financial reasoning, 565 seed questions
- CUDA, 100K samples
- safety, 在 Nemotron 3 Nano 的基础上使用了一个两阶段的 response
- search, 基于 wikidata knowledge graph, 然后合成 search problems
- terminal use, 基于 Nemotron-Terminal 合成了 84k samples.
- SQL, 96.5k records
- conversational tool use, 279k samples
- general tool use, 1.5M samples
最终 SFT 阶段包括 7M samples, 80B tokens, 其中 stage 1 占比 , 数据配比如下所示
RLVR 阶段的数据基于 Nemo Gym 合成得到,最终,作者在 37 个不同的数据集和 21个 environments 上进行训练
SWE-RL 阶段,作者使用 OpenHands 来生产 code patch, 为了提高 harness 多样性,作者还在 OpenHands 中实现了 Codex, OpenCode 等 agent.
Training
Pre-training
作者在预训练阶段探究了使用 model merge 来提高模型的表现,结果发现在 WSD stable stage, 使用 model merge 可以有效提高模型的表现,但是在 decay stage, model merge 的效果提升有限。
在 Phase 2 结果之后,与 Nemotron 3 Nano 一样,作者也加入了一个 long-context phase, 训练数据中 QA 数据占比 , 来源于 LC data, 其余 来自 phase 2 data, 最终一共训练了 17B tokens.
Post-traning
Post-training 分为 SFT, RL, MTP healing 三个阶段。
在 SFT 阶段,作者将训练分为了两个 phase, 其中, phase 1 主要是提高模型的 reasoning 能力, phase 2 用于提高模型的对话质量,防止出现过长的输出。
phase 1 的损失函数为 token level NLL loss:
phase 2 的损失函数为 sample level NLL loss:
其中 为 global batch, 为 response token, 为 NLL loss.
训练时,作者使用了三个 reasoning modes: reasoning-off, regular 和 low-effort, reasoning-off samples 通过 strip CoT 得到。
RL 的训练分为了 4 个阶段:1. RLVR; 2. SWE-RL; 3. RLHF; 4. MTP Healing
RLVR 训练时使用了 21 个 environments, 且根据难度进行了过滤。 训练过程中,作者还加入了 左右的 low-effort reasoning data 来提高模型的 reasoning efficiency.
SWE-RL 阶段,作者在 OpenHands 中实现了 OpenCode, Codex agent classes 来提高 tool diversity.
RLHF 阶段,与 Nemotron 3 Nano 一样,作者训练了一个 GenRM model 来提供 supervision.
RL 训练时,作者使用了 async RL, inference engine 的 staleness 设置为 1 step.
作者还使用了 MIS 来解决 training-inference mismatch 问题
在 agentic RL 训练过程中,作者使用了 PivotRL 策略,训练的 domain 包括 agentic programming, search, terminal use 和 conversational use.
RL 的训练基于 Nemo Gym 和 Nemo RL
Inference
作者使用了 PTQ 来得到 FP8 和 NVFP4 checkpoint, FP8 量化策略如下
| Configuration | FP8 Checkpoint | BF16 Baseline |
|---|---|---|
| Embedding | BF16 | BF16 |
| Attention GEMM (QKV and Out Projection) | BF16 | BF16 |
| KV Cache + Attention BMM1 | FP8 | FP8 |
| Attention BMM2 | BF16 | BF16 |
| MoE GEMM (Sparse Experts and Shared Experts) | FP8 | BF16 |
| MoE Latent Projection GEMM | BF16 | BF16 |
| Router | FP32 | FP32 |
| Mamba GEMM | FP8 | BF16 |
| Mamba SSM Cache | FP16 | FP32 |
| Mamba 1D Conv | BF16 | BF16 |
| Output Layers | BF16 | BF16 |
NVFP4 量化策略如下
以下是根据图片内容转换的 Markdown 表格:
| Configuration | AutoQuantize Searched? | NVFP4 Checkpoint | BF16 Baseline |
|---|---|---|---|
| Embedding | No | BF16 | BF16 |
| Attention QKV Projection GEMM | Yes | BF16 | BF16 |
| Attention Output Projection GEMM | Yes | FP8 / BF16 | BF16 |
| KV Cache + Attention BMM1 | No | FP8 | FP8 |
| Attention BMM2 | No | BF16 | BF16 |
| Sparse Expert (Routed) GEMM | Yes | NVFP4 | BF16 |
| Shared Expert GEMM | Yes | NVFP4 / FP8 / BF16 | BF16 |
| MoE Latent Projection GEMM | Yes | FP8 / BF16 | BF16 |
| Router | No | FP32 | FP32 |
| Mamba Projection GEMM | Yes | FP8 / BF16 | BF16 |
| Mamba 1D Conv | No | BF16 | BF16 |
| Mamba SSM Cache | No | FP16 | FP32 |
| Output Layers | No | BF16 | BF16 |
为了提高 decoding 阶段读取 Mamba state cache 的速度,作者在读取/写入 cache 时,使用 FP16 精度,计算时使用 FP32 精度
- Ainslie, J., Lee-Thorp, J., de Jong, M., Zemlyanskiy, Y., Lebron, F., & Sanghai, S. (2023). GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints. The 2023 Conference on Empirical Methods in Natural Language Processing. https://openreview.net/forum?id=hmOwOZWzYE
- Dao, T., & Gu, A. (2024). Transformers are SSMs: generalized models and efficient algorithms through structured state space duality. Proceedings of the 41st International Conference on Machine Learning.
- Guo, D., Yang, D., Zhang, H., Song, J., Wang, P., Zhu, Q., Xu, R., Zhang, R., Ma, S., Bi, X., Zhang, X., Yu, X., Wu, Y., Wu, Z. F., Gou, Z., Shao, Z., Li, Z., Gao, Z., Liu, A., … Zhang, Z. (2025). DeepSeek-R1 incentivizes reasoning in LLMs through reinforcement learning. Nature, 645(8081), 633–638. 10.1038/s41586-025-09422-z
- Team, Q. (2025). Qwen3-Next: Towards Ultimate Training & Inference Efficiency. https://qwen.ai/blog?id=4074cca80393150c248e508aa62983f9cb7d27cd&from=research.latest-advancements-list
Nemotron 3 Ultra
NVIDIA 在 26 年 6 月提出了 Nemotron 3 Ultra (NVIDIA et al., 2026), 一个 550B-A55B 的 MoE 大语言模型,Nemotron 3 其他模型一样,attention 也采取了 mamba-softmax 的 hybrid 架构
Architecture
Nemotron 3 Ultra 和 Nemotron 3 Super 的架构基本一致,
Data
Pre-training Data
- code: 作者新增了 173B tokens
- MCQ: 足者新增了 Nemotron-Pretraining-Multiple-Choice 和 Nemotron-Pretraining-Generative 两个与 MCQ 和 open ended QA 相关的数据集
- Fact seeking: 作者基于 Finewiki 构建了 fact seeking 数据
- Moral Scenarios
- Legal
最终,pre-training 阶段两个 phase 的数据配比如下图所示
Post-training Data
SFT SFT 阶段,作者使用 GPT-OSS-120B 的 medium-effort mode 以及随机 truncate reasoning traces 来构建动态思考数据。 作者还基于 OpenResearcher 构建了 21.7K search SFT data.
对于 terminal use, 作者基于 OpenCodeReasoning, OpenMathReasoning, SWE-Bench, SWE-Fixer-Train-110K, SWE-rebench 构建了 seed instructions, 然后基于 Nemotron-Cascade 和 DeepSeek-V3.2 合成了一批 task, 最后基于 harbor 来生成轨迹。通过这个流程,做和手机到了 370K multi-turn conversations.
对于 SWE 任务,作者基于 SWE-Gym, R2E-Gym, SWE-rebench, SWE-rebench-V2 来构建数据集,然后使用 MiniMax-M2.5 (thinking) 和 Qwen3-Coder-480B-A35B-instruct 来在 OpenHands, SWE-agent, mini-swe-agent 和 Opencode 这些 harness 中收集轨迹。 轨迹收集完之后,作者基于 submission integrity, disallowed git operations, edit test loop anti patterns, lost-in exploration, tool-call hygiene, debug-artifact detection in the final patch 这些策略来过滤数据
对于 math reasoning, 作者使用了 Nemotron-Cascade-2 的数据,对于 non-proof math data, 作者收集了 1.8M tool-calling, 1.9M non-tool calling; 对于 proof data, 作者从 Nemotron-Math-Proofs-v1 的 AOPS split.
对于 code, 作者基于 Nemotron-Cascade-2, 使用 GPT-OSS-120B 来收集轨迹,最终得到了 1.2M python, 1.0M C++14, 1.3M python tool-calling reasoning traces.
对于 CUDA, 作者构建了包含 100K samples 的数据集;对于 RTL, 作者构建了 1.2M training samples
最终,SFT 阶段数据共包含 40B token, 配比如下图所示
Training
Pre-training
Pre-training 时, Nemotron 3 Ultra 使用了 NVFP4 进行训练,为了提高训练稳定性和表现,作者将最后 15% 的网络使用更高精度进行训练。
训练时,作者使用了两阶段的数据编排策略,前 75% 的 token 主要关注 diversity, 后 25% 的数据主要关注 quality.
作者发现,使用低精度进行预训练存在如下问题:
- 在使用 MTP 进行预训练时,如果采用低精度,MTP block 的
wgrad对 shared output layer 几乎没有贡献,这严重影响了训练稳定性,因此作者采用了 FP32 gradient reduction 来解决这个问题 - MoE experts 的 load balance 可能与训练不稳定性有关,通过观测 experts 的负载可以分析模型训练是否崩溃。
预训练结束之后,作者还加入了一个 long context phase, 将模型的 context length 提升到 1M. 并行策略为 32CP+8TP+128EP+2PP.
long context data 由合成数据与 phase 2 data 融合得到,最后 8% iterations 作者使用 4k context length 来进行训练避免模型短文本能力缺失。
Post-training
Nemotron 3 Ultra 的 post-training 包含:SFT, RLVR, MOPD, MTP boosting 四个阶段
SFT 阶段包括 2 个 phase:
- phase 1, context length 294k, 204K samples
- phase 2, context length 512k, 19K samples
MOPD 训练阶段,作者采用了 async MOPD 的实现方式,另采样策略为 , 参照策略为 (Fu et al., 2026), 为 prompt 对应的 domain expert, 则对应的 importance sampling ratio 和 advantage 分别为
最终损失函数为
其中 时 IcePop 的 token-level masking [@IcePop].
接下来是 experts 的训练:
- SWE teacher: SFT, single-turn RL (PivotRL), multi-turn RL
- Office teacher: 基于 GDPval 构建任务
- Search teacher: 主要使用了 discard-all resets 和 summary-based compression 两种 context management 策略
- Code reasoning teacher: 14K competition problems, 4k OpenCodeReasoning problems
最终,一共训练得到了 10 多个 experts, 部分 experts 会基于训练出来的 student 重新初始化进行第二轮 experts 的训练。
在 MOPD 训练过程中,作者发现直接进行 MOPD 效果比较差,作者猜测原因是不同的教师模型 SFT 训练数据不同,因此其行为也各不相同。 为了解决这个问题,作者提出了 MOPD warmup, 也就是在对应 domain 的 OPD 训练之前,先进行轻量化的 SFT, 来对齐学生和教师的输出分布。
最后,作者提出了 MTP boosting 来解决 MTP 的训练推理不一致性问题,具体做法就是冻结 backbone, 只训练 MTP heads, 每个 head 的输入都从 MTP heads 输出的 hidden states 中随机进行选取,来模仿 inference 阶段不完整的上下文。
- Fu, W., Gao, J., Shen, X., Zhu, C., Mei, Z., He, C., Xu, S., Wei, G., Mei, J., Wang, J., Yang, T., Yuan, B., & Wu, Y. (2026). AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning. https://arxiv.org/abs/2505.24298
- NVIDIA, :, Blakeman, A., Thomas, A., Jhunjhunwala, A., Gupta, A., Khattar, A., Rajfer, A., Renduchintala, A., Asif, A., Vavre, A., Miranda, A. F., Bilal, A., Zaman, A., Hotchandani, A., Shukla, A., Bercovich, A., Ficek, A., Gronskiy, A., … Wertheimer, Z.-A. (2026). Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning. https://arxiv.org/abs/2606.15007
Nemotron-cascade
NVIDIA 在 25 年 12 月提出了 cascaded RL 以及训练得到的模型 Nemotron-Cascade (Wang et al., 2026), 模型包含 instruct 以及 thinking 两种模式。 cascaded RL 使用了 sequential, domain-wise RL 来提高模型的数学和代码能力。
现有的 RLHF 主要分为了 RLHF 和 RLVR 两种范式,前者用于提高模型的 alignment 能力,后者用于提高模型的 reasoning 能力。
但是提高模型的 reasoning 能力涉及到多个 domain, 每个 domain 的 verification 方式不一致,比如 math reasoning 依赖 symbolic rule-based verification, code generation 依赖于 execution based verification, alignment 依赖与 reward model based score.
在 AceReason-nemotron 中,作者提出了使用 Cascade RL 的方式来提高模型的数学和代码能力,作者发现这样做有以下优势:
- rule-based math verification 比 execution based verification 效率更高,可以在训练早期提高模型训练效率
- math reasoning 能力可以泛化到其他 domain 上
- math RL 之后,我们可以通过 code RL 提高模型的 code 能力而不损失模型的 math reasoning 表现
因此,在本文中,作者基于 Qwen3 (Yang et al., 2025) 的 Qwen3-8B-Base 以及 Qwen3-14B-Base 构建了一个开源的 post training recipe.
本文贡献如下:
- 在算法上,作者提出了 cascaded RL 框架,扩展了 RL 的 domain. Cascaded RL 的优势有:
- RLHF 可以有效提高模型输出的质量
- 后续 domain 的 RL 训练不会对前序 RL 产生影响
- 可以针对不同的 domain RL 分别调整超参数
- 作者通过优化数据构建了 Nemotron-Cascade-8B unified reasoning model.
Data
SFT
stage 1 数据构造
- general domain 数据包含 2.8M samples, 3.2B token, knowledge 相关数据有 1.2M samples, 1.5B tokens. 作者对数据进行了处理和增广。
- reasoning domain 数据包含了 AceReason-nemotron 中的数据,包含 AceMath, NuminaMath, OpenMathReasoning 这些数据集,作者收集了 353K prompts, 通过多次采样得到 2.77M samples.
- code reasoning data 来源于 AceReason-nemotron, 包含 TACO, APPs, OpenCoder-Stage-2, OpenCodeReasoning. 数据包含 172K prompts, 1.42M samples, context length 16K
- science reasoning data, 基于 S1K 和 LLaMA-Nemotron, 包含 226K prompts, 289K samples
stage 2 数据构造
- reasoning domain 数据,作者使用了
DeepSeek-R1-0528来生成回答,得到 163k prompts, 1.88M samples. context length 32K - code reasoning data, 作者使用了 OpenCodeReasoning 和 OpenCoder-Stage-2 数据,使用
DeepSeek-R1-0528来生成回答,得到 79K prompts, 1.39M samples, context length 32K - science reasoning data, 226K samples, 345K samples. upsample by 2.
- tool calling data, 基于 LLaMA-Nemotron, 使用
Qwen3-235B-A22B来生成回答,最终包含 320K 对话,1.41M turns. - SWE, 基于 SWE-Bench-Train, SWE-Fixer-Train, SWE-reBench, SWE-Smith. task 包含 localization, repair, test code generation 三个任务。最终 127K repair 数据,92K localization 数据,31K test case generation 数据,upsample by 2.
RLHF
reward model 训练数据包括 82k preference pairs: HelpSteer2 (10k), HelpSteer3 (36K)
数据合成:使用 stronger LLM 生成 bad response, weaker LLM 生成 good responses.
RLHF 训练数据与 reward model 一致,并过滤掉 code 和 math reasoning prompts.
IF RL
数据包括
- LLaMA-Nemotron, 包括 40k samples,
- LMSYS-Chat-1M: 40K samples
- IF-RLVR training data
- Tulu3-SFT
domain-RL
- math: 从 AceReason-Math 数据集中进行优化过滤得到 14k samples
- code: AceReason-Nemotron coding, 9.8k samples
- SWE: 数据与 SFT 阶段相同
Training
SFT
SFT 包含了两个阶段,分别覆盖 reasoning domain 和 general domain.
- stage 1: 16K, 包含 general domain 任务和 math, code 和 science domain 数据,general domain 数据每个 prompt 包含 thinking 和 non-thinking 两种 response, 训练 1 epoch
- stage 2: 32K, 加入了新的 math, science, code 数据,进一步提高模型的 tool use, software engineering skills, 训练 1 epoch
对于 chat template, 作者使用了 ChatML 格式,然后使用了 /think 和 /no_think 两个 flag 来控制是否进行思考。
作者通过实验发现,Qwen3 的 explicit flags 效果更好,并且可以覆盖所有场景,因此,作者舍弃了 <think></think> block.
作者还用了 <tools></tools>, <tool_call></tool_call> tags 来表明可用工具以及模型的工具调用
训练的超参数如下所示
RLHF
reward model 训练数据集为 ,
损失函数为
setup: batch size 256, lr , 1 epoch.
RLHF
基于 GRPO 进行训练,发现模型在下游任务上,除了 instruction following 之外,都有提升,作者认为这是由于 RLHF 要求模型输出较短的,符合人类偏好的回答,而 IF 则严格要求模型遵循输出格式。
IF-RL
两个阶段:第一个阶段关注 IFEval 相关数据,第二个阶段关注 IF-Bench-Train 相关数据。
- unified models: apply IF-RL only in the non-thinking mode
- thinking model: IF-RL with combined reward function
domain-RL
math-RL 训练包括三阶段:
- 24k, compression: 压缩 SFT checkpoint 输出长度
- 32k, extension: 扩展模型的输出长度和表现
- 40k, long reasoning stage: 提升模型解决复杂问题的难度
Code RL 训练使用了 async reward computation.
SWE RL reward function 定义为 generated patch 和 ground truth 之间的相似性
其中 是基于 Unidiff library 的 lexical similarity, 是 LLM 给出的语义相似度。
训练包括 2 个阶段, 第一阶段将输入初始化为 16k 作为 warmup, 第二阶段将输入扩展到 24k.
Experiments
作者对比了 Qwen3 提出的 Qwen3-8B, Qwen3-14B, 表现如下表所示
| Benchmark | Qwen3 8B | Qwen3 14B | Nemotron Cascade-8B | Namotron Cascade 14B-Thinking |
|---|---|---|---|---|
| MMLU | 83.0 | 84.9 | 83.7 | 85.1 |
| MMLU-Pro | 75.1 | 77.6 | 75.7 | 77.0 |
| GPQA-Dimond | 62.0 | 64.0 | 66.5 | 69.6 |
| ArenaHard | 85.8 | 91.7 | 87.9 | 89.5 |
| IFEval | 85.0 | 85.4 | 90.2 | 81.9 |
| IFBench | 34.4 | 33.7 | 40.8 | 41.7 |
| AIME 2024 | 76.0 | 79.3 | 89.5 | 89.7 |
| AIME 2025 | 67.3 | 70/4 | 80.1 | 83.3 |
| LiveCodeBench v5 | 61.2 | 65.2 | 74.3 | 77.5 |
| SWE-verified | 20.5 | 27.4 | 37.2 | 43.1 |
Analysis
reward model:
- benchmark score 低的表现一定差,但是高也不一定好
- 小模型更关注输出的风格而不是质量
- 使用 pre-trained reward model 和基于 SFT model 表现差不多
- reasoning model 比 instruction model 表现差
RLHF
- RLHF 可以降低模型输出重复的概率和 verbosity
- RLHF 可以提高模型在 math 和 code benchmark 上的表现
作者认为 cascaded RL 没有灾难性遗忘的原因:
- Wang, B., Lee, C., Lee, N., Lin, S.-C., Dai, W., Chen, Y., Chen, Y., Yang, Z., Liu, Z., Shoeybi, M., Catanzaro, B., & Ping, W. (2026). Nemotron-Cascade: Scaling Cascaded Reinforcement Learning for General-Purpose Reasoning Models. https://arxiv.org/abs/2512.13607
- Yang, A., Li, A., Yang, B., Zhang, B., Hui, B., Zheng, B., Yu, B., Gao, C., Huang, C., Lv, C., Zheng, C., Liu, D., Zhou, F., Huang, F., Hu, F., Ge, H., Wei, H., Lin, H., Tang, J., … Qiu, Z. (2025). Qwen3 Technical Report. https://arxiv.org/abs/2505.09388