← 返回 2026-04-28

World-R1:通过强化学习强化 3D 约束的文本到视频生成 World-R1: Reinforcing 3D Constraints for Text-to-Video Generation

Weijie Wang, Xiaoxuan He, Youping Gu, Yifan Yang, Zeyu Zhang, Yefei He, Yanbo Ding, Xirui Hu, Donny Y. Chen, Zhiyuan He, Yuqing Yang, Bohan Zhuang 📅 2026-04-27 👍 119 2026-07-13 08:36
3D 一致性 Flow-GRPO 世界模型 强化学习 文本到视频生成 相机控制

用强化学习把视频生成模型对齐 3D 几何,无需改架构即可获得 3D 一致性

前置知识

扩散模型与流匹配 (Flow Matching)

扩散模型通过学习去噪过程生成数据;流匹配是其变体,用确定性 ODE $dx_t = v_t dt$ 描述从噪声到数据的轨迹,比传统扩散在采样效率上更具优势。

World-R1 基于 Wan 2.1 这一流匹配视频模型,必须理解其 ODE 求解过程才能把握后续 SDE 改造和 RL 优化的动机。

GRPO 与 Flow-GRPO

GRPO 是一种省去 critic 网络、用组内奖励归一化估计优势的 RL 算法;Flow-GRPO 将其适配到流匹配模型,把 ODE 转为 SDE 以引入随机性,并通过 $L_{clip}$ 目标和 KL 约束更新策略。

World-R1 的核心训练框架正是 Flow-GRPO-Fast,理解其优势归一化公式和 SDE 策略是阅读奖励设计的必要前提。

3D 高斯溅射 (3D Gaussian Splatting)

3DGS 用一组可学习的 3D 高斯椭球表示场景几何与外观,通过可微分光栅化渲染新视角图像,是近年新视角合成的主流表征方式之一。

World-R1 用预训练 3D 基础模型 (Depth Anything 3) 把生成视频 lift 到 3DGS 表达 $\Phi_{GS}$,并以此计算重建奖励和新视角评估,是奖励机制的核心组件。

相机轨迹参数化与光流 warp

用外参矩阵 $E_t$ 描述相机在 3D 空间中的位姿,将相邻帧相对运动投影为 2D 光流场 $f(u) = u' - u$,即可把运动先验嵌入初始噪声 latent。

World-R1 的隐式相机条件化依赖这套几何流程,是把 prompt 中的运动关键词 ('push in'、'orbit left' 等) 转成噪声模式的关键桥梁。

研究动机

尽管 Wan 2.1、CogVideoX 等视频基础模型在视觉保真度上表现惊艳,但它们本质上仍是在 2D 图像空间做事,缺少对真实世界 3D 几何的内生理解。一旦提示词涉及复杂相机运动 (例如 push-in、orbit) 或长时序驾驶场景,生成内容常出现物体莫名消失、墙面扭曲、地形穿模等几何幻觉。论文 Table 2 数据印证了这一点:Wan2.1-T2V-1.3B 在 3D 重建 PSNR 上仅得 17.40 dB,Wan2.1-T2V-14B 也只有 19.76 dB,LPIPS 高达 0.467 和 0.405,意味着生成视频与基于 3DGS 的真实几何之间存在显著差距。现有 3D 注入方法 (CameraCtrl、ReCamMaster、DAS、Trajectory-Attention 等) 虽然能贴合轨迹,但 Table 1 显示它们 Aesthetic Quality 普遍跌到 38–43 区间,远低于 Wan 2.1 的 62.43;且它们只支持 I2V 任务、需要额外训练控制模块、推理延迟高,难以扩展到大模型。

本文的目标是本文提出 World-R1,目标是让一个现成的文生视频基础模型在不修改架构、不依赖昂贵 3D 数据集的前提下,通过后训练获得世界建模能力——即生成的视频要满足多视角几何一致、相机轨迹精准可控、视觉质量不退化这三个要求。论文以 Wan 2.1 1.3B / 14B 为基座,分别训练出 World-R1-Small 和 World-R1-Large 两个版本,希望在 832×480 分辨率下同时实现 3D 重建指标的大幅跃升和 VBench 通用质量的保持甚至超越。

与已有工作不同的是,作者的关键洞察是:视频基础模型内部已经隐式编码了丰富的 3D 几何信息 [16, 17],问题不在数据量而在于没有被激活。与其堆砌 3D 数据或在推理时硬塞几何约束,不如用强化学习作为「激活器」——通过 3D 基础模型和 VLM 的反馈信号,把潜在几何知识显化出来。这种 RL-only、architecture-free、3D-asset-free 的切入角度与现有 3D-aware 生成 (Fantasyworld 等) 和相机控制 (CameraCtrl 等) 方法形成鲜明区别,避免了高推理开销和任务域限制。

核心方法

World-R1 整体采用「离线几何工具 + 在线 RL 后训练」的范式。直觉上,它把文生视频模型当作一个策略 $\pi_\theta$,每条 prompt 是一回合交互:用 Gemini 生成的纯文本 prompt 配合从关键词解析出的相机轨迹 $\mathcal{E}$,通过 Go-With-The-Flow 式的噪声 wrap 注入相机先验,模型在 Flow-GRPO-Fast 的随机 SDE 采样下生成若干候选视频;然后用一个综合奖励函数从 3D 一致性和通用美学两个角度打分,最后按 GRPO 优势 $\hat{A}_i^t$ 更新参数。技术路线包含四大模块:隐式相机条件化、复合奖励设计、纯文本数据集、周期性解耦训练。

与已有方法最本质的区别在于「不修改模型、不堆 3D 数据,仅通过奖励信号引导 RL 后训练」。具体包含三个创新:(1) 用 prompt 关键词 → 相机轨迹 → 光流场 → 离散噪声 transport 的链路,把运动先验塞进初始 latent 而不引入任何控制网络;(2) 借助 Depth Anything 3 把生成视频 lift 到 3DGS 表征 $\Phi_{GS}$,再渲染新视角让 Qwen3-VL 评判,得到几何一致性的可微信号;(3) 周期性解耦训练每 100 步在动态子集上只用通用奖励微调一次,避免模型过度僵硬而丢失非刚性动态。

方法步骤详情

流程分五步:(1) Prompt 解析。定义关键词集 $K$ 涵盖 'push in'、'pan left' 等,由 $\phi(c)$ 检测 token,按 $E_t=E_{t-1}\cdot T_{action}(t)$ 递归生成相机外参。(2) 轨迹到光流投影。用平面单应把 3D 运动投影为像素对应,再按离散质量传输把高斯噪声 wrap 成相机条件 latent。(3) Rollout。Wan 2.1 在 SDE 采样下生成 $G=8$ 条候选视频。(4) 奖励评估。$R=R_{3D}+\lambda_{gen}R_{gen}$,$R_{3D}=S_{meta}+S_{recon}+S_{traj}$(Qwen3-VL 评 meta-view、$1-LPIPS$ 对比 3DGS 重渲染、轨迹对齐度量),按组内归一化得优势 $\hat{A}_i^t$。(5) 策略更新。以 $J(\theta)=E_c\frac{1}{T}\sum_t[L_{clip}(r_i^t,\hat{A}_i^t)-\beta D_{KL}(\pi_\theta\|\pi_{ref})]$ 更新参数。

技术新颖性

技术新颖性体现在四个层面。架构层面,World-R1 完全冻结 Wan 2.1 的 DiT 主干,仅通过 RL 微调参数,避免了 CameraCtrl 等方法额外训练 pose encoder 带来的延迟和泛化损失。信号层面,作者首创「3D 基础模型 lift + VLM meta-view 评判」的组合,把几何一致性的复杂判定转化为可微的标量奖励,是分析-合成 (analysis-by-synthesis) 思想在视频 RL 奖励中的首次系统化应用。数据层面,纯文本 prompt 数据集 (~3000 条) 解耦了 3D 学习与视觉分布偏差,避免依赖受限的 3D-aware 视频数据集。训练层面,周期性解耦策略以 100 步为周期在通用奖励和 3D 奖励间切换,用约 500 条动态子集作为正则器,是抑制 RL 奖励 hacking 与刚体过拟合的工程巧思。

Overview of World-R1.
Figure 2: Overview of World-R1.

实验结果

Table 2 显示 World-R1-Small PSNR 27.63 dB,比 Wan2.1-T2V-1.3B 的 17.40 dB 提升 10.23 dB;World-R1-Large 27.67 dB,比 Wan2.1-T2V-14B 的 19.76 dB 提升 7.91 dB;LPIPS 从 0.467/0.405 降至 0.201/0.162,SSIM 从 0.550/0.629 升至 0.858/0.865。Table 1 表明 World-R1-Small 在 VBench 的 Aesthetic 65.74、Imaging 67.53、Subject Consistency 97.58 均超 Wan 2.1 1.3B (62.43/66.51/96.34);显式控制方法 (GCD/ReCamMaster) Aesthetic 仅 38–43。用户盲测 (25 人、30 组 prompt) 几何胜率 92%、相机控制 76%、整体偏好 86%。Figure 4 消融表明去噪声 wrap 后 3D 奖励收敛变慢,去周期解耦则曲线饱和退化,验证四大组件必要性。

Quantitative comparison of 3D consistency.
Table 2: Quantitative comparison of 3D consistency.
Qualitative Comparison.
Figure 3: Qualitative Comparison.
Ablation study.
Figure 4: Ablation study.
查看结构化数据
任务指标本文基线提升
3D 一致性重建 (PSNR) PSNR (dB) ↑ World-R1-Small: 27.63;World-R1-Large: 27.67 Wan2.1-T2V-1.3B: 17.40;Wan2.1-T2V-14B: 19.76;CogVideoX-1.5-5B: 24.44 相对 1.3B 基座 +10.23 dB;相对 14B 基座 +7.91 dB
3D 一致性重建 (LPIPS) LPIPS ↓ World-R1-Small: 0.201;World-R1-Large: 0.162 Wan2.1-T2V-1.3B: 0.467;Wan2.1-T2V-14B: 0.405;CogVideoX-1.5-5B: 0.242 降幅约 57% (Small) / 60% (Large)
VBench Aesthetic Quality Aesthetic Quality ↑ World-R1-Small: 65.74 Wan2.1-T2V-1.3B: 62.43;CogVideoX-1.5-5B: 62.07;ReCamMaster: 42.70 相对最强 1.3B 基座 +3.31,相对显式控制方法 +23+
VBench Imaging Quality Imaging Quality ↑ World-R1-Small: 67.53 Wan2.1-T2V-1.3B: 66.51;CogVideoX-1.5-5B: 65.34;DAS: 51.55 +1.02 (vs. Wan 2.1 1.3B),远超显式控制方法
用户盲测几何一致性偏好 Win Rate World-R1 92% Wan 2.1 系列 8% 提升 84 个百分点

局限与改进

作者在 Limitations 与 Future Work 中明确指出了两点。首先是训练成本:RL 后训练要求反复 video rollout 和 reward evaluation,远比普通 SFT 昂贵;World-R1-Small 用 48 块 H200、World-R1-Large 用 96 块 H200 训练,48 个并行组、每组 8 条 rollout,进一步推高了算力门槛。其次是模型能力天花板:World-R1 受限于 Wan 2.1 自身的生成能力,在密集多目标、精细非刚性 (如人手动态)、超长时序场景中仍可能继承基座的瑕疵。从 Figure 3 也能观察到,作者自己承认「复杂相机动作下物体永久性和刚性」虽大幅改善,但远未达到完美。此外,Depth Anything 3 的 lift 精度本身是奖励信号可信度的上限,若 3DGS 重建失败会引发奖励 hacking;论文 Section D.7 虽做了相关分析但仍未完全解决。

独立分析的弱点

独立审视论文,仍可指出几点可改进之处。第一,奖励函数高度依赖 Depth Anything 3 lift 的 3DGS 重建质量,一旦视频含大量动态或细粒度遮挡 (如密集人群、烟雾),3DGS 可能直接崩塌,进而导致奖励信号失真,模型可能陷入 reward hacking;改进方向是引入不依赖几何重建的纯语义级多视角一致性奖励。第二,相机运动关键词集合 $K$ 由作者手工定义 (push in、pan left、orbit left 等),覆盖范围有限,对新颖复合运动 (例如「spiral descent with tilt」)的泛化能力存疑;可考虑用 LLM 自动扩展 token 集合或直接以文本嵌入驱动轨迹。第三,纯文本数据集虽然解耦视觉偏置,但 prompt 由 Gemini 生成,分布与真实用户 query 存在差异,可能限制在真实下游应用中的零样本表现;可补充真实用户 query 数据。第四,Figure 3 的可视化主要展示静态建筑与风景,对动态人物、运动场景的对照较少,削弱了「保持动态生成能力」这一论点的说服力。

未来方向

作者已经指明两个方向:降低 RL 训练成本 (例如开发更高效的 rollout 策略、低成本奖励评估、稳定的视频 RL 优化器) 和等待更强的视频基础模型承接这套后训练框架。从延伸角度看,未来还有几个有潜力的方向:(1) 把 World-R1 的范式推广到视频预测、自动驾驶仿真等具身场景,让生成视频直接接入物理模拟器;(2) 用奖励组合引入物理约束 (刚体动力学、流体守恒) 来约束非刚性内容;(3) 与 World Model 类研究 (GAIA-1、DriveDreamer) 结合,把视频生成器作为可微环境服务于 RL 智能体训练;(4) 探索 LoRA 或 adapter 式 RL 后训练,进一步降低算力门槛使其可被中小团队复现。

复现评估

复现评估需要打折扣。论文没有明确声明开源代码与模型权重,只给出了项目页 https://aka.ms/world-r1;训练数据集为作者私有 (Gemini 生成的 3000 条纯文本 + 500 条动态子集),需要复现者自行用类似 LLM 生成;实现细节方面,论文 Section A.2 仅提到 48/96 张 H200、832×480 分辨率、Flow-GRPO-Fast、48 组、组大小 8,但 GRPO 关键超参 $\beta$、$\lambda_{gen}$、奖励权重、训练步数等要么放在附录要么未完整披露;整体复现难度高、算力门槛极高 (96 张 H200 对大多数实验室不现实)。不过 VBench、3DGS 重建评估协议清晰且可复现,纯文本 prompt 配合相机轨迹生成逻辑也有据可依,从零实现核心 pipeline 在工程上是可行的。