自动驾驶中基于潜在世界模型学习多概率决策(LatentDriver)

Learning Multiple Probabilistic Decisions from Latent World Model in Autonomous Driving

问题

图a是将模型与规划器结合,但是这样对不确定性的考虑不全面,尤其是在涉及到自车与环境交互时的动作

图b是将世界模型的知识隐式地转移到规划器中,并对它们进行联合优化。这些方法未能充分发挥自回归模型的潜力

第一个问题是对不确定性的考虑不全面,尤其是在涉及到自车与环境交互时的动作。驾驶场景本质上是随机的,决策不应被视为单模态问题。可能存在多种有效的选择,每个选择代表分布的不同模式。另一个挑战是自欺问题。

第二个问题是自欺问题
自欺问题:过度依赖于基于历史动作的预测(例如:车辆前方突然出现障碍物,自回归世界模型依据之前的行驶动作预测车辆可以继续按原路线行驶,但规划器需要根据当前观测到的障碍物来决策,若规划器受到模型预测的干扰,忽视当前观测,就会导致危险,这就是自欺问题的体现)

创新点

核心观点是:假设动作和状态的分布以及它们的组合是多概率的

世界模型和规划器之间的交互是双向且完全随机的,最终动作是从它们的混合分布中推导出来的(图c)

具体来说,我们引入了多概率规划器(MPP),它通过混合高斯分布 [18,19] 将自车的动作建模为一个随机过程。MPP 采用多层 Transformer 结构,每一层都根据潜在世界模型(LWM)的输出优化动作分布。因此,它自然地捕捉到了自车的随机动作。为了缓解联合优化过程中的自欺问题,从 MPP 中间层采样得到的动作被用作真实动作的估计,减少了最终决策对历史动作的依赖

具体内容

LatentDriver框架:原始观测首先被矢量化,然后输入到场景编码器中。中间动作分布由多概率规划器(MPP)的中间层生成。

潜在世界模型(LWM)在接收到中间动作后,预测下一个潜在状态并将其表示为一种分布。

然后,动作分布和潜在状态分布通过 MPP 的后续层进行组合,形成一个混合分布,最终的控制信号由此导出

问题公式化

这段写的真挺好

输入表示和上下文编码

BERT作为场景编码器,处理一系列观测 O 1 : t O_{1:t} O1:t,得到类令牌和环境令牌的集合h

潜在特征的世界模型

潜在世界模型(LWM)旨在使用动作令牌和先前的潜在状态令牌(均由一个接受( A ˉ \bar{A} Aˉ和h的适配器生成)来预测下一个潜在状态令牌

适配器

动作令牌

潜在状态令牌

潜在世界模型

多概率规划器

将路标点视为动作空间,真实动作表示为 a ^ = [ a ^ x , a ^ y , a ^ y a w ] ∈ R 3 \hat{a} = [\hat{a}_x, \hat{a}y, \hat{a}{yaw}] \in \mathbb{R}^3 a^=[a^x,a^y,a^yaw]∈R3,其中yaw表示车辆的偏航角,用于描述车辆的行驶方向

MPA模块

损失函数

两个损失函数

一个是利用KL三度使得世界模型的适配器输出的潜在状态分布 ( s ^ i ) (\hat{s}{i}) (s^i)与估计的潜在状态分布 ( s ‾ i ) (\overline{s}{i}) (si)之间的 KL 散度,通过最小化这个散度,世界模型可以学习到更接近真实潜在状态的分布,提高预测的准确性。

另一个负对数似然损失来最大化自车真实动作 a ^ \hat{a} a^的可能性

相关推荐
格林威40 分钟前
常规线扫描镜头有哪些类型?能做什么?
人工智能·深度学习·数码相机·算法·计算机视觉·视觉检测·工业镜头
007php0071 小时前
某大厂跳动面试:计算机网络相关问题解析与总结
java·开发语言·学习·计算机网络·mysql·面试·职场和发展
倔强青铜三1 小时前
苦练Python第63天:零基础玩转TOML配置读写,tomllib模块实战
人工智能·python·面试
知识分享小能手1 小时前
微信小程序入门学习教程,从入门到精通,微信小程序核心 API 详解与案例(13)
前端·javascript·学习·react.js·微信小程序·小程序·vue
递归不收敛1 小时前
吴恩达机器学习课程(PyTorch 适配)学习笔记:3.3 推荐系统全面解析
pytorch·学习·机器学习
B站计算机毕业设计之家2 小时前
智慧交通项目:Python+YOLOv8 实时交通标志系统 深度学习实战(TT100K+PySide6 源码+文档)✅
人工智能·python·深度学习·yolo·计算机视觉·智慧交通·交通标志
高工智能汽车2 小时前
棱镜观察|极氪销量遇阻?千里智驾左手服务吉利、右手对标华为
人工智能·华为
txwtech2 小时前
第6篇 OpenCV RotatedRect如何判断矩形的角度
人工智能·opencv·计算机视觉
正牌强哥2 小时前
Futures_ML——机器学习在期货量化交易中的应用与实践
人工智能·python·机器学习·ai·交易·akshare
倔强青铜三2 小时前
苦练Python第62天:零基础玩转CSV文件读写,csv模块实战
人工智能·python·面试