HuaiCheng Yan | Searcharxiv

arXiv2024

Unleashing the Potential of Mamba: Boosting a LiDAR 3D Sparse Detector by Using Cross-Model Knowledge Distillation

The LiDAR 3D object detector that balances accuracy and speed is crucial for achieving real-time perception in autonomous driving. However, many existing LiDAR detection models rely on complex feature transformations, leading to poor real-time performance and high resource consumption, which limits their practical effectiveness. In this work, we propose a Faster LiDAR 3D object detection framework that Adaptively aligns Sparse voxels to enable efficient heterogeneous knowledge Distillation, called FASD. We aim to distill the Transformer's sequence modeling capability into Mamba models, significantly boosting accuracy through knowledge transfer. Specifically, we first design a cross-model knowledge distillation architecture to convey the global contextual understanding capabilities of the Transformer to Mamba. The Transformer-based teacher model employs a scale-adaptive attention mechanism to enhance multi-scale fusion. In contrast, the Mamba-based student model leverages feature alignment through spatial alignment adapters, supervised with latent-space features and span-head logit distributions, leading to improved performance and efficiency. We evaluated FASD on the Waymo and nuScenes datasets, achieving up to a 2x reduction in FLOPs and a 4x reduction in memory consumption, while improving baseline performance by 1-2 percentage points and maintaining high deployment efficiency.

cs.CV↗

arXiv2024

Future Does Matter: Boosting 3D Object Detection with Temporal Motion Estimation in Point Cloud Sequences

Accurate and robust LiDAR 3D object detection is essential for comprehensive scene understanding in autonomous driving. Despite its importance, LiDAR detection performance is limited by inherent constraints of point cloud data, particularly under conditions of extended distances and occlusions. Recently, temporal aggregation has been proven to significantly enhance detection accuracy by fusing multi-frame viewpoint information and enriching the spatial representation of objects. In this work, we introduce a novel LiDAR 3D object detection framework, namely LiSTM, to facilitate spatial-temporal feature learning with cross-frame motion forecasting information. We aim to improve the spatial-temporal interpretation capabilities of the LiDAR detector by incorporating a dynamic prior, generated from a non-learnable motion estimation model. Specifically, Motion-Guided Feature Aggregation (MGFA) is proposed to utilize the object trajectory from previous and future motion states to model spatial-temporal correlations into gaussian heatmap over a driving sequence. This motion-based heatmap then guides the temporal feature fusion, enriching the proposed object features. Moreover, we design a Dual Correlation Weighting Module (DCWM) that effectively facilitates the interaction between past and prospective frames through scene- and channel-wise feature abstraction. In the end, a cascade cross-attention-based decoder is employed to refine the 3D prediction. We have conducted experiments on the Waymo and nuScenes datasets to demonstrate that the proposed framework achieves superior 3D detection performance with effective spatial-temporal feature learning.

cs.CV↗