📄 论文信息

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

ReCogDrive 架构总览:从场景感知到驾驶决策的完整流程


🤔 要解决什么问题?

现有把 VLM 用于自动驾驶的方法有三个致命问题:

问题具体表现后果
领域差距VLM 在通用互联网数据上预训练,缺乏驾驶领域知识长尾场景泛化差
模态错配VLM 输出的是离散文本 token,但驾驶需要连续动作格式错误、不可行动作、推理慢
模仿学习次优只靠行为克隆(BC),无法超越专家数据罕见场景容易出事故

核心矛盾:VLM 会"想"但不会"开"——它能把场景描述得很清楚,但直接让它输出方向盘角度,要么格式不对,要么数值不合理。


💡 核心思路:三分天下

ReCogDrive 把驾驶拆成三个阶段,各司其职:

阶段模块作用类比
认知VLM(自回归)理解场景、推理决策人类"想清楚"
规划扩散规划器生成连续、平滑的轨迹人类"打方向盘"
优化DiffGRPO(强化学习)超越专家数据,提升安全人类"练出经验"

这种设计的关键洞察:不要让 VLM 直接输出动作,而是让它输出"认知表征",再由扩散模型翻译成动作。


⚙️ 模块一:层级化数据流水线

为了让 VLM 真正"懂驾驶",作者设计了三阶段数据构建流程:

阶段 1 — 生成(Generation):用已有驾驶数据自动生成大规模 VQA(视觉问答)数据

阶段 2 — 精炼(Refinement):用更强的模型过滤、修正低质量数据

阶段 3 — 质量控制(Quality Control):多维度评估,保留高质量认知标注

层级化数据流水线

这套流程模拟了人类司机的顺序认知过程:先看到什么 → 分析什么 → 做什么决策。数据不是简单的"图片+轨迹",而是包含了推理链的丰富标注。


⚙️ 模块二:认知引导的扩散规划器

这是解决"模态错配"的核心模块。

问题:VLM 输出的是文本/离散表征,但驾驶需要连续的轨迹点(x, y, heading)。

解法:用一个扩散规划器把 VLM 的"认知表征"翻译成连续轨迹。

认知引导扩散规划器架构

工作流程:

  1. VLM 处理多摄像头图像 → 输出认知表征(场景理解 + 驾驶推理)
  2. 认知表征作为条件注入扩散规划器
  3. 扩散规划器从噪声出发,去噪生成连续、平滑的轨迹

相比直接让 VLM 用文本输出轨迹坐标,扩散规划器生成的轨迹物理上更合理(平滑、可执行),且推理速度提升 7.8 倍


⚙️ 模块三:DiffGRPO 强化学习

这是 ReCogDrive 最具创新性的部分——把强化学习引入 VLA 驾驶模型

为什么需要 RL?

纯模仿学习(行为克隆)有三个局限:

  • 分布偏移:只见过专家轨迹,遇到没见过的情况就崩
  • 因果理解不足:只会模仿"怎么做",不理解"为什么"
  • 无法超越专家:上限就是专家数据的水平

DiffGRPO 怎么做?

DiffGRPO = Diffusion + Group Relative Policy Optimization

核心思想是:让扩散规划器在仿真环境中"试错",用奖励信号引导它找到更安全、更舒适的策略。

DiffGRPO 强化学习流程

组件说明
采样一组轨迹在同一场景下,用扩散模型生成多条候选轨迹
组内相对优势不需要绝对的价值函数,只用组内相对排名计算优势
奖励函数安全性(无碰撞)、舒适性(加加速度小)、效率(到达目标)
更新策略用优势信号更新扩散去噪网络

这种设计巧妙地结合了扩散模型的多模态生成能力和 RL 的探索能力——扩散模型天然适合"生成多个候选",而 GRPO 天然适合"从候选中挑好的"。


📊 实验结果

ReCogDrive 在 NAVSIM 上达到SOTA,PDMS(预测驾驶模型得分)大幅领先:

方法PDMS特点
传统 E2E(UniAD 等)~80+无语言推理
VLM 直接输出轨迹~85-88有推理但动作不稳
ReCogDrive90+推理 + 扩散 + 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 的代码已开源,感兴趣的可以直接跑起来试试。