
d1
用强化学习给扩散大模型补上推理能力
d1 是论文《d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning》的官方实现,目标是用强化学习提升扩散式大语言模型(dLLM)的推理能力。扩散语言模型并行去噪、生成速度快,但此前在数学、代码等需要多步推理的任务上明显弱于自回归模型。d1 把强化学习(如 GRPO 类策略优化)引入扩散 LLM 的后训练阶段,通过奖励信号优化去噪生成轨迹,让模型学会更长的思维链式推理,并在推理基准上取得可扩展的提升。项目提供训练与评测脚本、配置和预训练模型接口,基于 Python 与主流深度学习框架实现,适合研究扩散语言模型后训练、强化学习与推理增强方向的研究者复现和二次开发。
项目数据
使用教程
核心亮点
- 把 RL 后训练引入扩散 LLM,针对推理任务而非单纯文本生成,方向较新
- 提供官方训练与评测代码,可复现论文中的推理基准结果
- 支持在现有扩散语言模型上做后训练,不要求从零预训练
不足之处
- 早期项目,文档和示例相对简略,上手需要一定 RL 与扩散模型背景
- 依赖特定基座模型与算力,普通开发者复现成本较高
适用场景
- 研究扩散语言模型的强化学习后训练方法
- 在数学、代码等推理任务上评测 dLLM 能力
- 基于已有扩散 LLM 做推理增强的二次开发
替代项目
LLaDA、Dream、open-r1
项目介绍
上一篇:Rethinking-OPD
下一篇:Trinity-RFT
同类项目推荐
Long-RL
开源
让强化学习轻松驾驭超长序列,训练更稳更快
Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)
TPA
开源
把注意力复杂度从平方降到线性,长序列不再卡顿
[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···
minimind
开源
两小时造出你的小模型,练手入门不求人
Train a 64M-parameter LLM from scratch in just 2h!
pruna
开源
一键优化 AI 模型,推理提速又省资源。
Pruna is a model optimization framework built for developers, enabling you to delive···