🤖 AI Summary
This work addresses the degraded performance in detecting small objects and semantically related entities caused by point cloud sparsity and the lack of global structural context. Building upon the 3DETR framework, the authors introduce a multi-scale attention (MSA) mechanism combined with a feature upsampling strategy to effectively integrate local geometric details with global contextual information. This approach not only enhances the model’s local perception capabilities but also reveals that lightweight architectures require carefully matched upsampling strategies to achieve optimal performance. Evaluated on the ScanNetv2 dataset, the proposed method achieves a nearly 1% improvement in mAP@25 and a significant 4.78% gain in mAP@50, substantially outperforming the baseline model.
📝 Abstract
3D object detection in point cloud data remains a challenging task due to the sparsity and lack of global structure inherent in the input. In this work, we propose a novel Multi-Scale Attention (MSA) mechanism integrated into the 3DETR architecture to better capture both local geometry and global context. Our method introduces an upsampling operation that generates high-resolution feature maps, enabling the network to better detect smaller and semantically related objects. Experiments conducted on the ScanNetv2 dataset demonstrate that our 3DETR + MSA model improves detection performance, achieving a gain of almost 1% in mAP@25 and 4.78% in mAP@50 over the baseline. While applying MSA to the 3DETR-m variant shows limited improvement, our analysis reveals the importance of adapting the upsampling strategy for lightweight models. These results highlight the effectiveness of combining hierarchical feature extraction with attention mechanisms in enhancing 3D scene understanding.