UniBEV: Multi-modal 3D Object Detection with Uniform BEV Encoders for Robustness against Missing Sensor Modalities

Shiming Wang,Holger Caesar,Liangliang Nan,Julian F. P. Kooij

2024-05-08

Abstract:Multi-sensor object detection is an active research topic in automated driving, but the robustness of such detection models against missing sensor input (modality missing), e.g., due to a sudden sensor failure, is a critical problem which remains under-studied. In this work, we propose UniBEV, an end-to-end multi-modal 3D object detection framework designed for robustness against missing modalities: UniBEV can operate on LiDAR plus camera input, but also on LiDAR-only or camera-only input without retraining. To facilitate its detector head to handle different input combinations, UniBEV aims to create well-aligned Bird's Eye View (BEV) feature maps from each available modality. Unlike prior BEV-based multi-modal detection methods, all sensor modalities follow a uniform approach to resample features from the native sensor coordinate systems to the BEV features. We furthermore investigate the robustness of various fusion strategies w.r.t. missing modalities: the commonly used feature concatenation, but also channel-wise averaging, and a generalization to weighted averaging termed Channel Normalized Weights. To validate its effectiveness, we compare UniBEV to state-of-the-art BEVFusion and MetaBEV on nuScenes over all sensor input combinations. In this setting, UniBEV achieves $52.5 \%$ mAP on average over all input combinations, significantly improving over the baselines ($43.5 \%$ mAP on average for BEVFusion, $48.7 \%$ mAP on average for MetaBEV). An ablation study shows the robustness benefits of fusing by weighted averaging over regular concatenation, and of sharing queries between the BEV encoders of each modality. Our code will be released upon paper acceptance.

Computer Vision and Pattern Recognition,Robotics

What problem does this paper attempt to address?

### Problems Addressed by the Paper

The paper aims to address the robustness issue of multimodal 3D object detection in the absence of sensor inputs. Specifically, the researchers propose a new framework called UniBEV, which can maintain high detection performance even when LiDAR and camera data are missing.

#### Main Contributions:
1. **Unified Encoder Design**: UniBEV adopts a unified approach to handle data from different modalities (such as LiDAR and camera), ensuring feature alignment between different modalities. This design allows the model to function properly with single-modality input without the need for retraining.

2. **Improved Fusion Strategy**: The researchers compared different feature fusion methods, including simple concatenation, average fusion, and a new method called Channel Normalized Weights (CNW). Experiments show that the CNW method outperforms traditional feature concatenation methods in the case of missing modalities.

3. **Shared Query Mechanism**: The study found that sharing a unified Bird’s Eye View (BEV) query between different modalities can further improve feature alignment, thereby enhancing the overall performance of the model.

Through experiments on the nuScenes dataset, UniBEV achieved an average mAP of 52.5% across all input combinations, significantly outperforming existing BEVFusion (43.5% mAP) and MetaBEV (48.7% mAP). These results indicate that UniBEV has higher robustness and flexibility in the absence of sensor inputs.