← 返回全部文章
·约 9 分钟PPT

UniRL

图像设计

配套媒体

本页对应的 PPT 与演示视频

UniRL-recommend.mp4
PPT
UniRL-recommend.pptx浏览器不能直接打开 PPT,下载后用 PowerPoint / WPS / Keynote 查看
下载没反应?点这里复制直链

侧边栏预览窗口(Electron webview)会拦截下载,把下面的链接复制到 Chrome / Edge 地址栏打开即可保存。

PPT../../UniRL-recommend.pptx
视频../../UniRL-recommend.mp4
本文目录
  1. 🚀 什么是 UniRL?
  2. ✨ 核心功能一览
  3. 一、两大自研 RL 算法
  4. 二、广泛的多模态模型支持
  5. 三、模块化架构设计
  6. 🛠️ 快速上手:5 分钟跑通示例
  7. 安装
  8. 运行扩散模型 RL 训练
  9. 运行 LLM RL 训练
  10. 运行 VLM RL 训练
  11. 📊 与竞品对比
  12. 💡 适用场景
  13. 场景 1:多模态大模型 RL 对齐训练
  14. 场景 2:扩散模型 RL 算法研究和实验
  15. 场景 3:LLM RL 后训练工程化
  16. 💰 定价方案
  17. 🎯 总结

📌 核心价值:一套RL后训练回路打通文本、图像、视频多模态模型

腾讯混元团队开源的统一多模态强化学习框架,支持扩散模型、LLM、统一架构模型,含 Flow-DPPO 与 DRPO 两大自研算法。

封面图


🚀 什么是 UniRL?

UniRL(Unified + Ni + RL,意为"统一多模态智能所需的强化学习")是腾讯混元团队开源的分布式强化学习后训练框架。

核心定位: 一套 RL 后训练循环,统一覆盖文本、图像、视频等多模态生成模型的 RL 对齐训练。

核心理念:UniRL 将强化学习的五个步骤——生成样本、评分打分、计算优势、更新策略、同步权重回滚动工作节点——抽象为一条通用的后训练回路,不绑定单一模态或单一训练栈,以 Ray Worker Group + Hydra Flat Recipe + 可组合训练后端 + 可插拔 Rollout Engine 为骨架,实现不同模态模型的统一 RL 训练。**

GitHub: Tencent-Hunyuan/UniRL ⭐ Stars: 581+ | 🍴 Forks: 32+ | 📅 发布时间:2026年5月

UniRL 总架构图


✨ 核心功能一览

UniRL 核心功能一览

一、两大自研 RL 算法

1. Flow-DPPO:扩散模型的信赖域 RL

论文:FlowDPPO: Divergence Proximal Policy Optimization for Flow Matching Models

  • 专为扩散模型/流匹配模型设计的 RL 算法
  • 使用基于准确散度的信赖域掩码(exact divergence-based trust region mask)
  • 防止扩散模型 RL 训练中的策略崩溃和模式退化

Flow-DPPO 概览

2. DRPO:LLM RL 的正则化新思路

论文:Rethinking the Divergence Regularization in LLM RL

  • 针对大语言模型 RL 后训练的分层 KL 散度正则化方法
  • 在 Token 级别采用平滑优势加权二次正则化项
  • 缓解 GRPO/PPO 中常见的策略崩溃问题

DRPO 概览

二、广泛的多模态模型支持

模型类型 支持的模型
图像扩散模型 Stable Diffusion 3/3.5, Qwen-Image, FLUX.2-Klein
视频扩散模型 WAN 2.1/2.2, HunyuanVideo 1.0/1.5
自回归视觉语言模型 Qwen-VL
自回归语言模型 Qwen3
统一架构模型 HunyuanImage3, Bagel (AR + Diffusion)
提示增强器 Prompt-Enhancer

三、模块化架构设计

Rollout 引擎架构

  • 可插拔 Rollout Engine:vLLM / SGLang 自由切换
  • 可插拔 Algorithm:Flow-DPPO / DRPO / GRPO / DiffusionNFT / DanceGRPO / MixGRPO
  • 可插拔 Reward Service:不同模态的奖励计算解耦
  • 可插拔 Model Bundle:新模型接入只需注册

🛠️ 快速上手:5 分钟跑通示例

安装

# 克隆仓库
git clone https://github.com/Tencent-Hunyuan/UniRL.git
cd UniRL

# 安装依赖
pip install -e .

# 检查配置是否正确
python -m unirl.train_diffusion --config-name=diffusion/sd3_trainside --cfg job --resolve

运行扩散模型 RL 训练

# 使用 SGLang rollout 在 SD3 上运行示例
bash examples/run_experiment_single_node.sh diffusion/sd3_sglang_rollout_colocate

运行 LLM RL 训练

# Qwen3 + DRPO + SGLang Rollout
bash examples/run_experiment_single_node.sh ar/qwen3_drpo_4b_base_dapo_sglang

运行 VLM RL 训练

# Qwen-VL + GRPO + Geo3K 数据集
bash examples/run_experiment_single_node.sh ar/qwen_vl_grpo_geo3k_mc_4x8

注意:UniRL 为分布式框架,推荐在多 GPU 服务器上运行,需配置 Ray Worker Group。


📊 与竞品对比

UniRL vs 竞品对比


💡 适用场景

场景 1:多模态大模型 RL 对齐训练

如果你有多个不同模态的模型需要统一 RL 后训练,UniRL 提供了标准化的 RL 回路和算法实现,无需从零搭建。

输入:预训练的扩散模型/LLM/VLM + 奖励模型 输出:RL 对齐后的多模态模型权重

适用:AI Lab、大模型研究团队

代码架构图

场景 2:扩散模型 RL 算法研究和实验

Flow-DPPO 的信赖域掩码机制为扩散模型 RL 提供了新的正则化思路。

输入:扩散模型基座 + 自定义奖励函数 输出:Flow-DPPO 训练曲线 + 对齐后模型

适用:扩散模型研究人员、RL 算法研究员

场景 3:LLM RL 后训练工程化

DRPO 算法针对 LLM RL 中的策略崩溃问题,提出了 Token 级别的正则化方案。结合 SGLang/vLLM 的高效 rollout,可实现大规模 LLM RL 训练。

输入:Qwen3 等开源 LLM + 人类偏好数据 输出:RL 对齐后的对齐模型

适用:LLM 厂商、大模型对齐团队


💰 定价方案

完全开源免费,Apache License 2.0 协议。

无需 API Key,无需付费,在本地 GPU 集群上即可运行。

💡 亮点:作为一个来自大厂的高质量开源 RL 框架,UniRL 免费提供所有算法实现和分布式基础设施,大幅降低多模态 RL 训练的研究门槛。**


🎯 总结

UniRL 是目前业界唯一一个将扩散模型、视频模型、LLM、统一架构模型统一在一条 RL 后训练回路中的开源框架。腾讯混元团队在扩散模型 RL(Flow-DPPO)和 LLM RL(DRPO)两个方向上的原创贡献,使它在多模态 RL 领域占据了独特的生态位。

推荐指数: ⭐⭐⭐⭐⭐(满分5星)

适合人群: 大模型研究者、RL 算法工程师、多模态 AI 研发团队

GitHub 仓库: Tencent-Hunyuan/UniRL


数据截至 2026年06月14日,最新信息请以官网为准。