arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.12673cs.CV

MambaBEV:基于Mamba2的BEV三维检测模型

MambaBEV: An EV-based 3D detection model with Mamba2

School of Instrument Science and Engineering, Southeast University, China(仪器科学与工程学院,东南大学,中国);Amazon Development Center Germany GmbH, Germany(亚马逊德国开发中心,德国);T3CAIC Technology, China(T3CAIC技术,中国);School of Mechanical Engineering, Southeast University, China(机械工程学院,东南大学,中国)

Zihan You, Ni Wang, Hao Wang, Qichao Zhao, Jinxiang Wang

AI总结:

提出MambaBEV模型,利用Mamba2状态空间模型通过TemporalMamba时序融合模块和Mamba-based DETR头增强全局上下文建模,提升自动驾驶中大型物体的3D检测精度。

摘要:

自动驾驶中的精确3D物体检测依赖于鸟瞰图(BEV)感知和有效的时序融合。然而,现有基于卷积层或可变形自注意力的融合策略难以建模BEV空间中的全局上下文,导致大型物体的检测精度降低。为解决这一限制,我们提出了MambaBEV,一种新颖的基于BEV的3D物体检测模型,利用Mamba2——一种针对长序列处理优化的先进状态空间模型(SSM)。我们的关键贡献是TemporalMamba,一种时序融合模块,通过专为序列处理设计的BEV特征离散重排机制增强全局上下文建模。此外,我们引入了一个基于Mamba的DETR头以改进多物体表示。在nuScenes数据集上的评估表明,MambaBEV-base达到了51.7%的NDS和42.7%的mAP。此外,在端到端自动驾驶范式中的评估验证了其在运动预测和规划中的有效性。这些结果突显了状态空间模型在提升自动驾驶感知系统中全局上下文理解和大型物体检测方面的潜力。

英文摘要:

Accurate 3D object detection in autonomous driving relies on Bird's Eye View (BEV) perception and effective temporal fusion. However, existing fusion strategies based on convolutional layers or deformable self-attention struggle to model global context in BEV space, leading to reduced accuracy for large objects.To address this limitation, we propose MambaBEV, a novel BEV-based 3D object detection model that leverages Mamba2, an advanced state-space model (SSM) optimized for long-sequence processing. Our key contribution is TemporalMamba, a temporal fusion module that enhances global context modeling through a BEV feature discrete rearrangement mechanism tailored for sequential processing. In addition, we introduce a Mamba-based DETR head to improve multi-object representation. Evaluations on the nuScenes dataset demonstrate that MambaBEV-base achieves 51.7% NDS and an 42.7% mAP. Furthermore, evaluation within an end-to-end autonomous driving paradigm validates its effectiveness in motion forecasting and planning.These results highlight the potential of state-space models for improving global context understanding and large-object detection in autonomous driving perception systems.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑