北京大学与EVLO创新团队共同提出面向主动驾驭的四维时空预训练算法DriveWorld。该方法采用世界模型进行预训练，设计记忆情态空间模型进行四维时空建模，通过展望场景的盘踞栅格，降低主动驾驭面临的随机不确定性和知识不确定性。该论文已被CVPR 2024接收。
论文题目：DriveWorld: 4D Pre-trained Scene Understanding via World Models for Autonomous Driving
论文链接：https://arxiv.org/abs/2405.04390
一、动机
主动驾驭的场景了解工作涉及到对场景的感知和展望未来变化等多个层面，这些层面不仅包括空间上的三维结构，还包含时间维度上的动向变化。这种复杂的场景了解要求模型能够捕捉和了解四维时空的内在关联，从而做出准确的决策。由于自然场景的随机性、环境的局部可观测性以及各种下游工作的多样性，学习四维时空暗示是极具挑战性的。预训练在从大量数据中获取通用暗示方面发挥着关键作用，能够构建一个具备通用知识的基础模型。然而，有关主动驾驭中四维时空的预训练研究仍然相对较少。
主动驾驭系统的设计和实现需要面对和处理各种不确定性，这些不确定性主要分为两类：Aleatoric不确定性和Epistemic不确定性。Aleatoric不确定性源自于世界的固有随机性，例如行人的突然移动或车辆的意外行为。Epistemic不确定性则源于对环境不完全的认知，例如由于遮挡或传感器限制导致的信息缺失。为了有效应对这些不确定性，主动驾驭系统必须能够利用过去的经验来展望未来可能的情态，并对不可见的区域进行推测。本工作通过四维时空预训练的世界模型来解决这一挑战，旨在提升主动驾驭系统在感知、展望和规划工作中的性能。
二、方法
对于由主动驾驭环视相机系统观察到的T个视频帧的序列o1:T，以及它们对应的专家行为a1:T和三维盘踞栅格标签y1:T，其中三维盘踞栅格标签可以利用三维激光雷达点云和姿态数据获得。我们的目标是通过世界模型学习一个紧凑的BEV暗示，该暗示通过过去多视角图像和行动展望的当前和未来的三维盘踞栅格。
2.1时序概率模型
为了赋予模型四维时空建模的能力，我们首先引入两个潜在变量(h1:T,s1:T)，其中ht暗示历史信息变量，包含了到时间步t的所有历史信息，st暗示随机情态变量，是模型展望未来情态的关键。ht通过历史信息h1:t−1和随机情态s1:t−1进行更新。为了展望未来情态，我们遵循循环情态空间模型（Recurrent State-Space Model，RSSM），构建后验情态疏散q(st∣o≤t,a<t)和先验情态疏散p(st∣ht−1,st−1)。目标是匹配先验疏散（基于历史信息和随机情态的预期结果）与后验疏散（从观察到的多视角图像和行动中导出的结果）。
考虑到BEV特征的维度很高，我们将其转换为一维向量xt，然后从(ht,at−1,xt)中抽样高斯疏散以生成后验情态疏散：p(st∣ht−1,st−1)∽N(μθ(ht,a^t−1),σθ(ht,a^t−1)I),其中st被参数化为带有对角协方差的正态疏散，初始疏散设置为s1∽N(0,I)。(μϕ,σϕ)是参数化后验情态疏散的多层感知机。
在没有观察到图像的情况下，模型根据历史信息和展望的行动得出先验情态疏散：p(st∣ht−1,st−1)∽N(μθ(ht,a^t−1),σθ(ht,a^t−1)I),其中(μθ,σθ)参数化先验情态疏散。𝜋𝜃是用于展望行动 a^t−1的策略网络，基于历史信息ht−1和随机情态st−1。
2.1.1 动向信息传递
在主动驾驭的场景了解中，考虑物体的静止对于准确展望未来情态至关重要。为了捕捉这种动向信息，我们提出通过引入静止参数来建模物体的静止，从而在动向信息传播过程中实现静止感知。我们引入了静止感知层归一化（MLN）。静止属性包括速度v和相对时间间隔Δt。(v,Δt)被展平并通过两个线性层(ξ1,ξ2)转换为仿射向量γ和β：γ=ξ1(v,Δt),β=ξ2(v,Δt)。 然后执行仿射变换以得到静止感知的潜在随机情态，暗示为st=γ⋅LN(st)+β。随着车辆的静止，确定性历史情态ht可以建立动向记忆库h1:t。通过与动向记忆库进行交叉注意机制计算，可以得到确定性历史情态ht。确定性历史情态为ht+1=fθ(ht,st)。
2.1.2 空间信息传递
在主动驾驭的场景了解中，除了动向变化信息，空间结构信息同样重要。由于连续的场景帧通常只包含微小的变化，而场景的主要内容往往是由静态物体组成的，如道路、树木和交通标志，因此在处理这些信息时，直接将输入图像转换为一维向量可能会导致关键空间结构信息的丢失。我们从1到T帧中随机选择一帧o′，并使用其BEV特征b′构建一个描述空间感知结构的潜在静态暗示b^=zθ(b′)。我们将空间感知的静态暗示b^与动向变化的静止暗示st结合起来，得到了周围场景的综合暗示。
2.2 预训练辅助工作
对周围环境的全面了解对主动驾驭视至关重要的。我们提出将物理世界建模为三维盘踞栅格结构来描述车辆周围的环境。三维盘踞栅格解码器被设置为y^t=lθ(mθ(h~t,st),b^),其中mθ是将一维特征扩展到BEV维度的网络，lθ是用于展望盘踞栅格的三维卷积网络。这种四维盘踞栅格预训练不仅能够捕捉到场景的静态结构，还能够了解场景随时间的动向变化，为主动驾驭系统提供了更加丰富和动向的环境了解。
2.3 工作提醒机制
虽然通过世界模型设计的预训练工作可以学习四维时空暗示，但不同的下游工作关注的信息是不同的。为了缓解这个问题，受少样本图像识别的语义提醒和多工作学习中的视觉示例引导提醒的启发，引入了“工作提醒”机制，为不同的工作提供特定的提醒，以引导它们提取工作相关的特征。由于不同工作之间存在语义关联，我们利用大语言模型gφ(⋅)（例如BERT，CLIP）构建这些工作提醒。例如，针对三维盘踞栅格重建工作的工作提醒，其关注更多的是当前场景，设置为“工作是展望当前场景的三维盘踞栅格”。我们将提醒ptext输入到gφ(⋅)中以获取提醒编码gφ(ptext)。随后将其扩展到BEV的维度，暗示为qφ(gφ(ptext))，将其与学到的时空特征集成在一起。
2.4 预训练目标函数
DriveWorld的预训练目标包括最小化后验情态疏散与先验情态疏散之间的差异（即Kullback-Leibler（KL）散度），以及最小化与过去和未来三维盘踞栅格（即交叉熵损失（CE））和行动（即L1损失）相关的损失。我们采用模型在T个时间步内观察输入，然后展望未来的三维盘踞栅格和L个步骤的行动。
三、实验
3.1 实验设置
我们在主动驾驭数据集上nuScenes和OpenScenes上进行预训练，并在nuScenes上进行微调。我们采用多帧激光雷达点云聚合的方式获得密集的三维盘踞栅格标签。
3.2 实验结果
这里展示部分结果，更多结果请参考论文。
四、总结
DriveWorld通过基于世界模型的四维时空预训练，提高主动驾驭系统对周围环境的了解和展望能力，降低主动驾驭面临的不确定性。DriveWorld提出了记忆情态空间模型进行时空建模，包含动向记忆存储模块用于学习时序感知暗示，静态场景传播模块用于学习空间感知暗示。为了进一步提升模型的适应性和灵活性，DriveWorld还引入了工作提醒机制，允许模型根据当前的工作需求自适应地调整其暗示，从而在不同的主动驾驭工作中实现最佳性能。
参考
[1]Chen Min, et al. Multi-Camera Unified Pre-Training Via 3D Scene Reconstruction[J]. IEEE Robotics and Automation Letters, 2024.
[2]Chen Min, et al. Occupancy-mae: Self-supervised pre-training large-scale lidar point clouds with masked occupancy autoencoders[J]. IEEE Transactions on Intelligent Vehicles, 2023.
EVOL创新团队介绍
赵健，中国电信人工智能研究院多媒体认知学习实验室（EVOL Lab）负责人、青年科学家，西北工业大学光电与智能研究院研究员、博导，博士毕业于新加坡国立大学，研究兴趣包括多媒体分析、临地安防、具身智能。
共发表CCF-A类论文60余篇，含一作T-PAMI×2（IF: 24.314）、IJCV×3（IF: 13.369），第一发明人授权国家发明专利5项。相关技术成果在百度、蚂蚁金服、奇虎360等6个科技行业领军企业得到应用，产生了显著效益。曾入选中国科协及北京市科协“青年人才托举工程”，主持国自然青年科学基金等项目6项。曾获吴文俊人工智能优秀青年奖（2023）、吴文俊人工智能自然科学奖一等奖（2/5，2022）、新加坡模式识别与机器智能协会（PREMIA）Lee Hwee Kuan奖、ACM Multimedia唯一最佳学生论文奖（一作，1/208，CCF-A类会议，2018），7次在国际重要科技赛事中夺冠。
担任北京图象图形学学会理事，国际知名期刊《Artificial Intelligence Advances》、《IET Computer Vision》编委，《Pattern Recognition Letters》、《Electronics》特刊客座编辑，VALSE资深领域主席，ACM Multimedia 2021分论坛主席，CICAI 2022/2023领域主席，CCBR 2024论坛主席，中国人工智能学会/中国图象图形学学会高级会员，“挑战杯”大学生科技作品竞赛评委，中国人工智能大赛专家委委员等。
GitHub主页：https://zhaoj9014.github.io
学院主页：https://iopen.nwpu.edu.cn/info/1252/4626.htm
金磊，北京邮电大学特聘副研究员，主要研究方向包括计算机视觉、数据挖掘、模式识别，其中深入研究人体姿态估计、人体行动识别、人体解析等细分领域，相关成果发表于CVPR, AAAI, NIPS, ACMMM等高水平会议及期刊，共发表SCI/EI索引论文40余篇，其中高水平论文11篇，包括以第一作者发表中科院JCR一区论文（IEEE Transactions on MultiMedia），CCF-A类会议CVPR, ACMMM论文，中科院JCR二区（Sensors, IEEE Sensor Journal）论文等。主持一项国家自然基金青年基金，参与两项国家重点研发项目以及四项自然基金面上项目。多次依托顶会组织ICCV2021/CVPR2023 workshop (Anti-UAV Workshop & Challenge)。指导学生获得全国大学生物联网技术与应用“三创”大赛一等奖（北邮认定A类竞赛）。
闵称，北京大学计算机学院博士，中科院计算所特别研究助理，主要研究方向包括主动驾驭、具身智能、三维重建，相关成果发表于CVPR、ICCV、ICRA、RAL等高水平会议与期刊，包括以第一作者发表CCF-A类会议CVPR，机器人顶级会议ICRA，机器人权威期刊RAL等。参与多项国家重点研发项目。