MambaBEV: An efficient 3D detection model with Mamba2

Zihan You, Ni Wang, Hao Wang, Qichao Zhao, Jinxiang Wang
10/16/2024
cs.CV

No confident code implementation found for this paper.

Abstract

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 with global context modeling in BEV space,leading to lower accuracy for large objects. To address this, we introduce 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 awareness by introducing a BEV feature discrete rearrangement mechanism tailored for Mamba's sequential processing. Additionally, we propose Mamba based DETR as the detection head to improve multi object representation.Evaluations on the nuScenes dataset demonstrate that MambaBEV base achieves an NDS of 51.7\% and an mAP of 42.7\%.Furthermore, an end to end autonomous driving paradigm validates its effectiveness in motion forecasting and planning.Our results highlight the potential of SSMs in autonomous driving perception, particularly in enhancing global context understanding and large object detection.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{you2024mambabev,
  title  = {MambaBEV: An efficient 3D detection model with Mamba2},
  author = {Zihan You and Ni Wang and Hao Wang and Qichao Zhao and Jinxiang Wang},
  year   = {2024},
  eprint = {2410.12673},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.12673v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.12673.svg)](https://codeofpaper.com/paper/2410.12673)

Discussion