搜索
首页科技周边人工智能CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

写在前面&笔者的个人理解

目前,在整个自动驾驶系统当中,感知模块扮演了其中至关重要的角色,行驶在道路上的自动驾驶车辆只有通过感知模块获得到准确的感知结果后,才能让自动驾驶系统中的下游规控模块做出及时、正确的判断和行为决策。目前,具备自动驾驶功能的汽车中通常会配备包括环视相机传感器、激光雷达传感器以及毫米波雷达传感器在内的多种数据信息传感器来收集不同模态的信息,用于实现准确的感知任务。

基于纯视觉的BEV感知算法因其较低的硬件成本和易于部署的特点,以及其输出结果能便捷地应用于各种下游任务,因此受到工业界和学术界的广泛关注。近年来,许多基于BEV空间的视觉感知算法相继涌现,且在公开数据集上展现出色的感知性能。

目前,基于BEV空间的感知算法根据构建BEV特征的方式可以大体分成两类算法模型:

  • 一类是以LSS算法为代表的前向BEV特征构建方式,这类感知算法模型首先是利用感知模型中的深度估计网络来预测特征图的每个像素点的语义特征信息以及离散深度概率分布,然后将得到的语义特征信息与离散深度概率采用外积运算的方式构建语义视锥特征,采用BEV池化等方式最终完成BEV空间特征的构建过程。
  • 另外一类是以BEVFormer算法为代表的反向BEV特征构建方式,这类感知算法模型首先是在感知的BEV空间下显式的生成3D体素坐标点,然后利用相机的内外参将3D体素坐标点投影回图像坐标系下,并对相应特征位置的像素特征进行提取和聚合,从而构建出BEV空间下的BEV特征。

尽管两种算法都能够准确生成BEV空间下的特征并实现3D感知结果,但在目前基于BEV空间的3D目标感知算法中,比如BEVFormer算法,存在以下两个问题:

  • 问题一:由于BEVFormer感知算法模型整体框架采用的是Encoder-Decoder的网络结构,其主要思想是利用Encoder模块获取BEV空间下的特征,然后利用Decoder模块预测最终的感知结果,并通过将输出的感知结果与真值目标计算损失来实现模型预测的BEV空间特征的过程。但通过这种网络模型的参数更新方式会过于依赖Decoder模块的感知性能,导致可能存在模型输出的BEV特征与真值BEV特征并不对齐的问题,从而进一步制约感知模型最终的表现性能。
  • 问题二:由于BEVFormer感知算法模型的Decoder模块依旧沿用Transformer中的自注意力模块->交叉注意力模块->前馈神经网络的步骤完成Query特征的构建输出最终的检测结果,其整个过程依旧是一个黑盒模型,缺乏良好的可解释性。同时,Object Query与真值目标之间的一对一匹配过程在模型训练的过程中也存在着很大的不确定性。

为了解决BEVFormer感知算法模型存在的问题,我们对其进行了改进,提出了基于环视图像的3D检测算法模型CLIP-BEVFormer。我们通过引入对比学习的方法,增强了模型对BEV特征的构建能力,并在nuScenes数据集上实现了领先水平的感知性能。

文章链接:https://arxiv.org/pdf/2403.08919.pdf

网络模型的整体架构&细节梳理

在详细介绍本文提出的CLIP-BEVFormer感知算法模型细节之前,下图展示了CLIP-BEVFormer算法的整体网络结构。

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能本文提出的CLIP-BEVFormer感知算法模型整体流程图

通过算法的整体流程图可以看出,本文提出的CLIP-BEVFormer算法模型是在BEVFormer算法模型的基础上进行改进的,这里先简单回顾一下BEVFormer感知算法模型的实现过程。首先,BEVFormer算法模型输入的是相机传感器采集到的环视图像数据,利用2D的图像特征提取网络提取输入环视图像的多尺度语义特征信息。其次,利用包含时序自注意力和空间交叉注意力的Encoder模块完成2D图像特征向BEV空间特征的转换过程。然后,在3D感知空间中以正态分布的形式生成一组Object Query,并送入到Decoder模块中完成与Encoder模块输出的BEV空间特征的空间特征交互利用。最后利用前馈神经网络预测Object Query查询到的语义特征,输出网络模型最终的分类和回归结果。同时,在BEVFormer算法模型训练的过程中,采用一对一的匈牙利匹配策略完成正负样本的分配过程,并利用分类和回归损失完成整体网络模型参数的更新过程。BEVFormer算法模型整体的检测过程可以用如下的数学公式进行表示:

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

其中,公式中的代表BEVFormer算法中的Encoder特征提取模块,代表BEVFormer算法中的Decoder解码模块,代表数据集中的真值目标标签,代表当前BEVFormer算法模型输出的3D感知结果。

真值BEV的生成

在上文中已经有提到,现有的绝大多数基于BEV空间的3D目标检测算法没有显式的对生成的BEV空间特征进行监督,导致模型生成的BEV特征可能存在与真实的BEV特征不一致的问题,这种BEV空间特征的分布差异会制约模型最终的感知性能。基于这一考虑出发,我们提出了Ground Truth BEV模块,我们设计该模块的核心思路是想让模型生成的BEV特征可以和当前真值BEV特征进行对齐,从而提高模型的表现性能。

具体而言,如整体网络框架图所示,我们使用了一个真值编码器()用来对BEV特征图上的任意一个真值实例的类别标签和空间边界框位置信息进行编码,该过程可以用公式表述成如下的形式:

其中公式中的具有和生成的BEV特征图同等大小的特征维度,代表某个真值目标被编码后的特征信息。在编码的过程中,我们采用了两种形式,一种是大语言模型(LLM),另外一种是多层感知机(MLP),通过实验结果发现,两种方式基本达到了同样的性能。

除此之外,我们为了进一步增强真值目标在BEV特征图上的边界信息,我们在BEV特征图上根据真值目标所在的空间位置将其裁剪下来,并对裁剪后的特征采用池化操作构建对应的特征信息表示,该过程可以表述成如下的形式:

最后,我们为了实现模型生成的BEV特征与真值BEV特征的进一步对齐,我们采用了对比学习的方法来优化两类BEV特征之间的元素关系和距离,其优化过程可以表述成如下的形式:

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

其中公式中的和分别代表生成的BEV特征和真值BEV特征之间的相似度矩阵,代表对比学习中的逻辑尺度因子,表示矩阵间的乘法运算,代表交叉熵损失函数。通过上述对比学习的方式,我们提出的方法可以为生成的BEV特征提供更加明确的特征指导,提高模型的感知能力。

真值目标查询交互

这部分在前文中也有提到,BEVFormer感知算法模型中的Object Query通过Decoder模块与生成的BEV特征进行交互,获得对应的目标查询特征,但该过程整体还是一个黑盒过程,缺少一个完整的流程理解。针对这一问题,我们引入了真值查询交互模块,通过将真值目标来执行Decoder模块的BEV特征交互来激发模型参数的学习过程。具体而言,我们将真值编码器()模块输出的真值目标编码信息引入到Object Query当中参与Decoder模块的解码过程,与正常的Object Query参与相同的自注意力模块,交叉注意力模块以及前馈神经网络输出最终的感知结果。但需要注意的是,在解码的过程中,所有的Object Query均是采用了并行计算的方式,防止发生真值目标信息的泄露。整个真值目标查询交互过程,可以抽象表述成如下的形式:

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

其中,公式中的代表初始化的Object Query,和分别代表真值Object Query经过Decoder模块和感知检测头的输出结果。通过在模型训练的过程中引入真值目标的交互过程,我们提出的真值目标查询交互模块可以实现真值目标查询与真值BEV特征进行交互,从而辅助模型Decoder模块的参数更新过程。

实验结果&评价指标

定量分析部分

为了验证我们提出的CLIP-BEVFormer算法模型的有效性,我们分别在nuScenes数据集上从3D感知效果、数据集中目标类别的长尾分布情况以及鲁棒性等角度出发进行了相关实验,下表是我们提出的算法模型与其他3D感知算法模型在nuScenes数据集上的精度对比情况。

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

本文提出的方法与其他感知算法模型的对比结果

在这部分实验中,我们分别评估了不同模型配置情况下的感知性能,具体而言,我们将CLIP-BEVFormer算法模型应用于BEVFormer的tiny和base变体中。此外,我们还探索了将预训练的CLIP模型或者MLP层作为真值目标编码器对于模型感知性能的影响。通过实验结果可以看出,无论是原先的tiny还是base变体,在应用了我们提出的CLIP-BEVFormer算法后,NDS和mAP指标均有稳定的性能提升。除此之外,通过实验结果我们可以发现,对于真值目标编码器选择MLP层还是语言模型,我们提出的算法模型对于此并不敏感,这种灵活性可以使得我们提出的CLIP-BEVFormer算法更具有适应能力并且方便上车部署。总之,我们提出的算法模型的各类变体的性能指标一致表明提出的CLIP-BEVFormer算法模型具有很好的感知鲁棒性,可以在不同模型复杂度和参数量的情况下实现出色的检测性能。

除了验证我们提出的CLIP-BEVFormer在3D感知任务上的性能外,我们还进行了长尾分布的实验来评估我们的算法在面对数据集中存在长尾分布情况下的鲁棒性和泛化能力,实验结果汇总在下表

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

提出的CLIP-BEVFormer算法模型在长尾问题上的表现性能

通过上表的实验结果可以看出,nuScenes数据集中表现出了极大的类别数量不均衡的问题,其中一些类别如(建筑车辆、公交车、摩托车、自行车等)占比很低,但是对于小汽车的占比非常高。我们通过进行长尾分布的相关实验来评估提出的CLIP-BEVFormer算法模型在特征类别上的感知性能,从而验证其解决不太常见类别的处理能力。通过上述的实验数据可以看出,提出的CLIP-BEVFormer算法模型在所有类别上均实现了性能的提升,并且在对于占比极少的类别上,CLIP-BEVFormer算法模型展示出了明显的实质性改进。

考虑到在真实环境下的自动驾驶系统需要面临硬件故障、恶劣天气状况或者人造障碍物容易引发的传感器故障等问题,我们进一步实验验证了提出的算法模型的鲁棒性。具体而言,我们为了模拟传感器的故障问题,我们在模型实施推理的过程中随机对一个相机的摄像头进行遮挡,从而实现对于相机可能出现故障的场景进行模拟,相关的实验结果如下表所示

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能提出的CLIP-BEVFormer算法模型的鲁棒性实验结果

通过实验结果可以看出,无论是在tiny还是base的模型参数配置下,我们提出的CLIP-BEVFormer算法模型始终要优于BEVFormer的相同配置的基线模型,验证了我们的算法模型在模拟传感器故障情况下的优越性能和优秀的鲁棒性。

定性分析部分

下图展示了我们提出的CLIP-BEVFormer算法模型与BEVFormer算法模型的感知结果可视化对比情况。通过可视化的结果可以看出,我们提出的CLIP-BEVFormer算法模型的感知结果与真值目标更加的接近,表明我们提出的真值BEV特征生成模块与真值目标查询交互模块的有效性。

CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能

提出的CLIP-BEVFormer算法模型与BEVFormer算法模型感知结果的可视化对比情况

结论

在本文中,针对原有的BEVFormer算法当中存在的生成BEV特征图过程中缺少显示监督以及Decoder模块中Object Query与BEV特征交互查询的不确定问题,我们提出了CLIP-BEVFormer算法模型,并从算法模型的3D感知性能、目标长尾分布以及在传感器故障的鲁棒性等方面进行实验,大量的实验结果表明我们提出的CLIP-BEVFormer算法模型的有效性。

以上是CLIP-BEVFormer:显式监督BEVFormer结构,提升长尾检测性能的详细内容。更多信息请关注PHP中文网其他相关文章!

声明
本文转载于:51CTO.COM。如有侵权,请联系admin@php.cn删除
AI技能差距正在减慢供应链AI技能差距正在减慢供应链Apr 26, 2025 am 11:13 AM

经常使用“ AI-Ready劳动力”一词,但是在供应链行业中确实意味着什么? 供应链管理协会(ASCM)首席执行官安倍·埃什肯纳齐(Abe Eshkenazi)表示,它表示能够评论家的专业人员

一家公司如何悄悄地努力改变AI一家公司如何悄悄地努力改变AIApr 26, 2025 am 11:12 AM

分散的AI革命正在悄悄地获得动力。 本周五在德克萨斯州奥斯汀,Bittensor最终游戏峰会标志着一个关键的时刻,将分散的AI(DEAI)从理论转变为实际应用。 与闪闪发光的广告不同

NVIDIA释放NEMO微服务以简化AI代理开发NVIDIA释放NEMO微服务以简化AI代理开发Apr 26, 2025 am 11:11 AM

企业AI面临数据集成挑战 企业AI的应用面临一项重大挑战:构建能够通过持续学习业务数据来保持准确性和实用性的系统。NeMo微服务通过创建Nvidia所描述的“数据飞轮”来解决这个问题,允许AI系统通过持续接触企业信息和用户互动来保持相关性。 这个新推出的工具包包含五个关键微服务: NeMo Customizer 处理大型语言模型的微调,具有更高的训练吞吐量。 NeMo Evaluator 提供针对自定义基准的AI模型简化评估。 NeMo Guardrails 实施安全控制,以保持合规性和适当的

AI为艺术与设计的未来描绘了一幅新图片AI为艺术与设计的未来描绘了一幅新图片Apr 26, 2025 am 11:10 AM

AI:艺术与设计的未来画卷 人工智能(AI)正以前所未有的方式改变艺术与设计领域,其影响已不仅限于业余爱好者,更深刻地波及专业人士。AI生成的艺术作品和设计方案正在迅速取代传统的素材图片和许多交易性设计活动中的设计师,例如广告、社交媒体图片生成和网页设计。 然而,专业艺术家和设计师也发现AI的实用价值。他们将AI作为辅助工具,探索新的美学可能性,融合不同的风格,创造新颖的视觉效果。AI帮助艺术家和设计师自动化重复性任务,提出不同的设计元素并提供创意输入。 AI支持风格迁移,即将一种图像的风格应用

Zoom如何彻底改变与Agent AI的合作:从会议到里程碑Zoom如何彻底改变与Agent AI的合作:从会议到里程碑Apr 26, 2025 am 11:09 AM

Zoom最初以其视频会议平台而闻名,它通过创新使用Agentic AI来引领工作场所革命。 最近与Zoom的CTO XD黄的对话揭示了该公司雄心勃勃的愿景。 定义代理AI 黄d

对大学的存在威胁对大学的存在威胁Apr 26, 2025 am 11:08 AM

AI会彻底改变教育吗? 这个问题是促使教育者和利益相关者的认真反思。 AI融入教育既提出了机遇和挑战。 正如科技Edvocate的马修·林奇(Matthew Lynch)所指出的那样

原型:美国科学家正在国外寻找工作原型:美国科学家正在国外寻找工作Apr 26, 2025 am 11:07 AM

美国科学研究和技术发展或将面临挑战,这或许是由于预算削减导致的。据《自然》杂志报道,2025年1月至3月期间,美国科学家申请海外工作的数量比2024年同期增加了32%。此前一项民意调查显示,75%的受访研究人员正在考虑前往欧洲和加拿大寻找工作。 过去几个月,数百项NIH和NSF的拨款被终止,NIH今年的新拨款减少了约23亿美元,下降幅度接近三分之一。泄露的预算提案显示,特朗普政府正在考虑大幅削减科学机构的预算,削减幅度可能高达50%。 基础研究领域的动荡也影响了美国的一大优势:吸引海外人才。35

所有有关打开AI最新的GPT 4.1家庭的信息 - 分析Vidhya所有有关打开AI最新的GPT 4.1家庭的信息 - 分析VidhyaApr 26, 2025 am 10:19 AM

Openai推出了强大的GPT-4.1系列:一个专为现实世界应用设计的三种高级语言模型家族。 这种巨大的飞跃提供了更快的响应时间,增强的理解和大幅降低了成本

See all articles

热AI工具

Undresser.AI Undress

Undresser.AI Undress

人工智能驱动的应用程序,用于创建逼真的裸体照片

AI Clothes Remover

AI Clothes Remover

用于从照片中去除衣服的在线人工智能工具。

Undress AI Tool

Undress AI Tool

免费脱衣服图片

Clothoff.io

Clothoff.io

AI脱衣机

Video Face Swap

Video Face Swap

使用我们完全免费的人工智能换脸工具轻松在任何视频中换脸!

热工具

mPDF

mPDF

mPDF是一个PHP库,可以从UTF-8编码的HTML生成PDF文件。原作者Ian Back编写mPDF以从他的网站上“即时”输出PDF文件,并处理不同的语言。与原始脚本如HTML2FPDF相比,它的速度较慢,并且在使用Unicode字体时生成的文件较大,但支持CSS样式等,并进行了大量增强。支持几乎所有语言,包括RTL(阿拉伯语和希伯来语)和CJK(中日韩)。支持嵌套的块级元素(如P、DIV),

SublimeText3 Linux新版

SublimeText3 Linux新版

SublimeText3 Linux最新版

VSCode Windows 64位 下载

VSCode Windows 64位 下载

微软推出的免费、功能强大的一款IDE编辑器

适用于 Eclipse 的 SAP NetWeaver 服务器适配器

适用于 Eclipse 的 SAP NetWeaver 服务器适配器

将Eclipse与SAP NetWeaver应用服务器集成。

安全考试浏览器

安全考试浏览器

Safe Exam Browser是一个安全的浏览器环境,用于安全地进行在线考试。该软件将任何计算机变成一个安全的工作站。它控制对任何实用工具的访问,并防止学生使用未经授权的资源。