📄 论文信息
- 标题:DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving(迈向真实场景驱动的自动驾驶世界模型)
- 团队: CASIA(中科院自动化所)× 清华大学 × 鉴智机器人(Xiaofeng Wang, Zheng Zhu, Guan Huang, Xinze Chen, Jiagang Zhu, Jiwen Lu)
- arXiv:2309.09777(2023 年 9 月)
- 一句话总结:首个完全从真实驾驶数据构建的世界模型,用扩散模型吃下结构化交通约束(HDMap、3D 框、文本、动作),既能生成可控的驾驶视频,又能预测未来动作——把"理解世界"和"生成世界"统一起来了。
🤔 要解决什么问题?为什么需要驾驶世界模型?
世界模型(World Model) 的核心能力是理解环境如何随时间演化——给定当前状态和动作,预测未来会发生什么。这对自动驾驶至关重要:能"预见未来"的策略,才能提前避让、从容决策。
但 DriveDreamer 之前,驾驶世界模型有一个致命短板:
| 已有世界模型 | 问题 |
|---|---|
| 基于游戏/仿真环境(如 CARLA、DriveGAN) | 与真实驾驶差距巨大,迁移困难 |
| 缺乏真实场景理解 | 只会"画画",不懂交通规则与结构 |
DriveDreamer 的判断:必须从真实驾驶数据出发构建世界模型,而且要让它理解结构化的交通约束(车道线、3D 框、交通规则),而不只是像素层面的"像"。这是它区别于一切前作的根本所在。

💡 核心思想:用扩散模型"理解"驾驶世界
驾驶场景的搜索空间是天文数字——同一条路在不同天气、光照、车流下千变万化。DriveDreamer 的核心选择是:用扩散模型(Diffusion Model)来建模这个复杂的联合分布。
为什么是扩散?因为扩散模型在图像/视频生成上已被证明有最强的分布拟合能力,而且它的条件生成天然适合"给定约束生成未来"的范式。DriveDreamer 把扩散模型当成一个驾驶世界的"模拟器":喂给它当前状态和约束,它就能"演绎"出合理的未来。
多模态条件输入
DriveDreamer 接受多种结构化条件,而不只是原始像素:
| 条件类型 | 内容 | 作用 |
|---|---|---|
| 文本(text) | 天气、时段、场景描述 | 控制风格与环境 |
| HDMap | 车道边界、车道分隔线、人行横道 | 几何结构约束 |
| 3D 框(3D box) | 八角点 3D 检测框 | 障碍物位置约束 |
| 动作(action) | 驾驶操作(转向、加减速) | 演绎"如果我这么做" |
这些条件被各自的编码器编码后,通过 ControlNet 式 的机制注入扩散主干,让生成既逼真又可控。
🏗️ 两阶段训练:先懂规则,再会预言
DriveDreamer 最精彩的设计是它的两阶段训练流水线(two-stage pipeline)。直接端到端学"从约束生成视频"太难,于是拆成两步:

第一阶段:理解结构化交通约束(Auto-DM)
第一阶段(论文里称为 Auto-DM,Auto-driving Decision Model)的目标是让模型先学会"看懂"交通结构:
- 输入:图像 + HDMap + 3D 框 + 文本
- 任务:从结构化条件重建/生成当前帧图像
- 收获:模型被迫学会"车道线、3D 框、文本"与视觉外观之间的对应关系
这一步是打地基——模型先理解"世界长什么样、规则是什么",才能在第二阶段去预测未来。
第二阶段:未来状态预测
有了第一阶段的"世界理解",第二阶段才进入真正的世界模型能力:
- 输入:当前帧 + 历史信息 + 动作
- 任务:预测未来多帧视频(世界演化)和未来动作(决策)
- 收获:模型学会"动作 → 状态变化"的因果,能推演未来
| 阶段 | 学什么 | 类比 |
|---|---|---|
| 阶段一(Auto-DM) | 读懂交通结构(约束 ↔ 画面) | “认字” |
| 阶段二 | 预测未来(动作 → 演化) | “写故事” |
这种"先理解、后预测“的渐进式设计,避免了让模型一上来就硬啃"从约束直接生成未来"的难题,是工程上的关键智慧。
Auto-DM 的内部:ControlNet 式条件注入
Auto-DM(第一阶段模型)的实现借鉴了 ControlNet 的设计哲学:冻结一个预训练的扩散主干(负责生成能力),再外接一组条件编码器(HDMap 编码器、3D 框编码器、文本编码器),通过 zero-conv 注入主干。这种设计的好处是生成先验与条件控制解耦——主干保留了从海量图像学到的"世界长什么样"的常识,条件编码器只负责"告诉它这次要遵守什么约束”。
| 条件 | 编码方式 | 控制的内容 |
|---|---|---|
| HDMap | 矢量化后光栅化到图像平面 | 车道、人行横道的几何 |
| 3D 框 | 八角点投影到 2D | 障碍物的位置与尺度 |
| 文本 | CLIP/文本编码器 | 天气、时段、场景风格 |
| 动作 | ActionFormer 时序编码 | 自车未来的运动轨迹 |
第一阶段训练时,模型在真实帧上学习"约束 ↔ 画面"的映射;第二阶段则扩展到时序维度,让模型在已有结构理解的基础上去推演未来帧与动作。这种分阶段的课程,让每个阶段的学习目标都足够清晰、足够"可学"。
🎬 两大核心能力:可控视频生成 + 动作预测
1. 可控驾驶视频生成
DriveDreamer 能根据多种条件组合生成高质量的驾驶视频:
- 结构可控:给定 HDMap 和 3D 框,生成符合几何约束的画面
- 风格可控:调整文本提示,改变天气、时段(白天/夜晚、晴/雨)
- 动作可控:给定驾驶动作,演绎"如果转向/变道会看到什么"
这让 DriveDreamer 成为一个强大的数据生成器——可以按需生成各种长尾场景(暴雨夜、施工段),用于增强训练数据。
2. 未来动作预测
除了生成视频,DriveDreamer 还能预测未来的驾驶动作,这直接对接规划任务。论文中用 ActionFormer 等模块来预测自车未来的动作序列,让世界模型不只是一个"放映机",更是一个"决策助手"。
直觉:一个真正理解驾驶世界的模型,应该既会"演未来"(视频生成),也会"想下一步该干嘛"(动作预测)——DriveDreamer 把两者统一在一个框架里。
📊 实验结果:nuScenes 上的 FID/FVD
DriveDreamer 在 nuScenes 上验证。由于 nuScenes 原生只提供 2Hz 的 3D 框标注,作者补充了 12Hz 的高频标注以满足视频生成需求。
主要评测指标:
| 指标 | 含义 |
|---|---|
| FID(Fréchet Inception Distance) | 生成图像质量(越低越好) |
| FVD(Fréchet Video Distance) | 生成视频质量与时序一致性(越低越好) |
关键结论:
- 显著优于 DriveGAN 等仿真驱动方法:FID/FVD 大幅改善,证明用真实数据 + 强扩散模型的组合远胜游戏/仿真环境
- 可控性强:HDMap/3D 框/动作/文本都能有效控制生成内容,与真实交通结构高度对齐
- 动作预测合理:生成的驾驶动作/策略在真实场景下合理可行
这些结果共同验证了"真实数据 + 扩散 + 结构化条件 + 两阶段训练“这条路线的可行性,也确立了 DriveDreamer 作为驾驶世界模型奠基工作的地位。
⚔️ DriveDreamer 在世界模型谱系中的位置
理解 DriveDreamer,要看它在驾驶世界模型演进中的坐标:
| 代际 | 代表 | 数据来源 | 核心能力 |
|---|---|---|---|
| 第一代(仿真) | DriveGAN、CARLA 世界模型 | 游戏/仿真 | 生成但不真实 |
| 第二代(真实) | DriveDreamer | 真实驾驶 | 可控生成 + 动作预测 |
| 后续演进 | DriveDreamer-2、Drive-WM、Cosmos、GAIA | 真实驾驶 | 更高分辨率、更长时序、多视角 |
DriveDreamer 的开创性在于两点:第一次用真实数据、第一次把结构化交通约束显式引入。这两个"第一次"定义了后续所有真实驾驶世界模型的基本范式。
🔗 世界模型如何反哺自动驾驶?
世界模型不只是"炫技的视频生成器”,它对自动驾驶有实实在在的价值:
| 应用 | 价值 |
|---|---|
| 数据增强 | 生成长尾/罕见场景补充训练数据 |
| 闭环仿真 | 提供比游戏更真实的训练/评测环境 |
| 决策推演 | “如果我变道会怎样”——what-if 分析 |
| 安全校验 | 规划前先用世界模型预演结果 |
这也解释了为什么 2024-2025 年世界模型在自动驾驶领域爆发:它是解决"长尾数据稀缺"和"闭环仿真真实性"两大痛点的钥匙。
⚠️ 局限性与未来方向
DriveDreamer 作为早期工作,也有明显局限:
- 分辨率与时序长度有限:生成视频不够长、不够清晰
- 多视角一致性:nuScenes 六路相机的一致性维护仍有挑战
- 物理精确性:扩散生成不保证严格遵循物理规律(如碰撞反弹)
- 闭环保真度:生成数据与真实分布仍有 gap,直接训练可能有问题
后续工作(DriveDreamer-2、GAIA-1、Cosmos、Drive-WM)沿着更高分辨率、更长时序、更强物理一致性、更好可控性的方向不断推进,但 DriveDreamer 开创的"真实数据 + 结构化条件 + 扩散"范式始终是主线。
📝 个人思考
读 DriveDreamer,最打动我的是它对 “世界模型到底要学什么” 的清醒认识。很多人把世界模型简单等同于"会生成视频",但 DriveDreamer 把它讲清楚了:世界模型的本质不是"画得像",而是"懂规则、能推演"。它显式地把 HDMap、3D 框这些结构化约束作为条件,迫使模型理解交通的几何与逻辑,而不只是像素统计。这个选择非常关键——一个只会"画"不懂"规则"的模型,生成的视频再好看,也无法用于严肃的决策与仿真。我倾向于认为,世界模型的价值不在生成质量,而在因果理解,DriveDreamer 用结构化条件把这一点落到了实处。
第二点启发在 两阶段训练的工程智慧。直接学"从约束生成未来视频"是个超级难题,DriveDreamer 把它拆成"先懂结构、再会预言"两步,这种渐进式课程的思想在 AI 里反复出现——从预训练到微调、从简单到复杂。它提醒我们:遇到一个太难的端到端目标时,最好的办法往往是给它搭一个合理的台阶。第一阶段让模型先建立"世界长什么样"的常识,第二阶段再去学"世界怎么变",这种先静态后动态、先理解后预测的顺序,符合人类认知发展的规律,也降低了优化难度。
最后,DriveDreamer 让我看到了 世界模型与端到端驾驶融合的必然趋势。当世界模型能逼真地推演未来、能生成无限长尾数据、能给策略提供 what-if 的推演能力时,它就不再只是感知的附属品,而是自动驾驶系统的"想象引擎"。未来的端到端框架很可能是:感知 → 世界模型推演 → 规划,世界模型成为连接"看见"和"决策"的中枢。DriveDreamer 在 2023 年就播下了这颗种子,今天 Cosmos、GAIA 等大型世界基础模型的繁荣,都是这颗种子的生长。理解世界,才能驾驭世界——这是 DriveDreamer 给自动驾驶留下的最深刻的启示。
📖 这是论文精读系列的第 23 篇。世界模型正在成为自动驾驶的"想象引擎",你认为它会取代传统仿真器吗?欢迎留言讨论。