📄 论文信息
- 标题:ReCogDrive: A Reinforced Cognitive Framework for End-to-End Autonomous Driving
- 来源:华中科技大学 × 小米汽车(Xiaomi EV)
- arXiv:2506.08052
- 代码:github.com/xiaomi-research/recogdrive
- 模型:HuggingFace Collection
- 一句话总结:VLM 负责"理解与推理",扩散模型负责"生成轨迹",DiffGRPO 强化学习负责"安全优化",三者合一。

🤔 要解决什么问题?
现有把 VLM 用于自动驾驶的方法有三个致命问题:
| 问题 | 具体表现 | 后果 |
|---|---|---|
| 领域差距 | VLM 在通用互联网数据上预训练,缺乏驾驶领域知识 | 长尾场景泛化差 |
| 模态错配 | VLM 输出的是离散文本 token,但驾驶需要连续动作 | 格式错误、不可行动作、推理慢 |
| 模仿学习次优 | 只靠行为克隆(BC),无法超越专家数据 | 罕见场景容易出事故 |
核心矛盾:VLM 会"想"但不会"开"——它能把场景描述得很清楚,但直接让它输出方向盘角度,要么格式不对,要么数值不合理。
💡 核心思路:三分天下
ReCogDrive 把驾驶拆成三个阶段,各司其职:
| 阶段 | 模块 | 作用 | 类比 |
|---|---|---|---|
| 认知 | VLM(自回归) | 理解场景、推理决策 | 人类"想清楚" |
| 规划 | 扩散规划器 | 生成连续、平滑的轨迹 | 人类"打方向盘" |
| 优化 | DiffGRPO(强化学习) | 超越专家数据,提升安全 | 人类"练出经验" |
这种设计的关键洞察:不要让 VLM 直接输出动作,而是让它输出"认知表征",再由扩散模型翻译成动作。
⚙️ 模块一:层级化数据流水线
为了让 VLM 真正"懂驾驶",作者设计了三阶段数据构建流程:
阶段 1 — 生成(Generation):用已有驾驶数据自动生成大规模 VQA(视觉问答)数据
阶段 2 — 精炼(Refinement):用更强的模型过滤、修正低质量数据
阶段 3 — 质量控制(Quality Control):多维度评估,保留高质量认知标注

这套流程模拟了人类司机的顺序认知过程:先看到什么 → 分析什么 → 做什么决策。数据不是简单的"图片+轨迹",而是包含了推理链的丰富标注。
⚙️ 模块二:认知引导的扩散规划器
这是解决"模态错配"的核心模块。
问题:VLM 输出的是文本/离散表征,但驾驶需要连续的轨迹点(x, y, heading)。
解法:用一个扩散规划器把 VLM 的"认知表征"翻译成连续轨迹。

工作流程:
- VLM 处理多摄像头图像 → 输出认知表征(场景理解 + 驾驶推理)
- 认知表征作为条件注入扩散规划器
- 扩散规划器从噪声出发,去噪生成连续、平滑的轨迹
相比直接让 VLM 用文本输出轨迹坐标,扩散规划器生成的轨迹物理上更合理(平滑、可执行),且推理速度提升 7.8 倍。
⚙️ 模块三:DiffGRPO 强化学习
这是 ReCogDrive 最具创新性的部分——把强化学习引入 VLA 驾驶模型。
为什么需要 RL?
纯模仿学习(行为克隆)有三个局限:
- 分布偏移:只见过专家轨迹,遇到没见过的情况就崩
- 因果理解不足:只会模仿"怎么做",不理解"为什么"
- 无法超越专家:上限就是专家数据的水平
DiffGRPO 怎么做?
DiffGRPO = Diffusion + Group Relative Policy Optimization
核心思想是:让扩散规划器在仿真环境中"试错",用奖励信号引导它找到更安全、更舒适的策略。

| 组件 | 说明 |
|---|---|
| 采样一组轨迹 | 在同一场景下,用扩散模型生成多条候选轨迹 |
| 组内相对优势 | 不需要绝对的价值函数,只用组内相对排名计算优势 |
| 奖励函数 | 安全性(无碰撞)、舒适性(加加速度小)、效率(到达目标) |
| 更新策略 | 用优势信号更新扩散去噪网络 |
这种设计巧妙地结合了扩散模型的多模态生成能力和 RL 的探索能力——扩散模型天然适合"生成多个候选",而 GRPO 天然适合"从候选中挑好的"。
📊 实验结果
NAVSIM 基准(开环)
ReCogDrive 在 NAVSIM 上达到SOTA,PDMS(预测驾驶模型得分)大幅领先:
| 方法 | PDMS | 特点 |
|---|---|---|
| 传统 E2E(UniAD 等) | ~80+ | 无语言推理 |
| VLM 直接输出轨迹 | ~85-88 | 有推理但动作不稳 |
| ReCogDrive | 90+ | 推理 + 扩散 + RL |
Bench2Drive(闭环)
在闭环仿真中,ReCogDrive 同样表现优异,碰撞率显著降低——这正是 DiffGRPO 强化学习的功劳。
DriveBench / DriveLM(VQA)
ReCogDrive 的 VLM 在驾驶问答基准上也超越了开源和闭源模型,证明其认知理解能力强。

消融实验关键发现
- 去掉扩散规划器(让 VLM 直接输出轨迹)→ PDMS 大幅下降,证明模态解耦的必要性
- 去掉 DiffGRPO → 碰撞率上升,证明 RL 优化的价值
- 去掉认知数据 → 长尾场景性能下降,证明层级化数据的价值
🎯 这篇论文为什么重要
| 价值 | 说明 |
|---|---|
| 解耦设计 | 第一次清晰地把"认知"(VLM)和"动作"(扩散)分开,解决了模态错配 |
| RL for VLA | 首次将 GRPO 引入 VLA 驾驶模型,证明 RL 能超越模仿学习 |
| 产业落地 | 小米 EV 出品,有明确的车端部署考量(7.8× 加速) |
| 完全开源 | 代码、模型权重、数据集全部公开 |
📝 个人思考
对工程实践的启发:ReCogDrive 的"三分天下"设计(认知 → 规划 → 强化)是一个非常好的架构模板。我们自己的工作(如 Flow-GRPO)也在做类似的事——在扩散/流匹配策略上做 RL。ReCogDrive 验证了这条路线的可行性。
DiffGRPO 的启示:把 GRPO 和扩散模型结合,这个思路可以迁移到很多场景。核心洞察是:扩散模型的"采样多条轨迹"和 GRPO 的"组内相对优势"天然契合。这比在确定性策略上做 PPO 要优雅得多。
值得关注的后续:ReCogDrive 的 VLM 推理速度仍然是个瓶颈(即使加速了 7.8 倍)。未来如果能把 VLM 也蒸馏成更轻量的模型,或者用快慢双系统(参考 DriveVLM-Dual),实时性可以进一步提升。
📖 论文精读系列。ReCogDrive 的代码已开源,感兴趣的可以直接跑起来试试。