Energy-based Active Learning for Bringing Beam-induced Domain Gap for 3D Object Detection
In many real-world applications, 16-beam LiDAR-based 3D object detection (3DOD) is indispensable in scene understanding. However, the absence of well-labeled large-scale 16-beam LiDAR datasets impedes the development of these 3DOD methods. To avoid annotation costs in developing datasets, we proposed an energy-based active learning method for cross-beam domain adaptation, which effectively transfers the knowledge from the existing well-labeled 64-beam counterpart. Specifically, the cross-beam domain gap between the source (64-beam) and the target (16-beam) domain is reduced by aligning the deep features based on an energy-based feature-matching loss term during training. Moreover, the proposed energy-based active learning method enables the sampling strategy to shed light on selecting the most valuable 16-beam target samples to be manually labeled, which are then added to the training set. Experimental results show that our method can effectively transfer the knowledge from the 64-beam domain to the 16-beam one, and successfully learns a high-performance 16-beam 3DOD model with only a small portion of unlabeled data to annotate.
Authors
- Le Yang (ORCID: https://orcid.org/0000-0001-8379-4915)
- Fan Li (ORCID: https://orcid.org/0000-0002-7566-1634)
- Yixuan Yan
- Jing Shao
- Hao Cheng
Institutions
- Northeastern University (US)
- Hong Kong University of Science and Technology (HK)
- Xi'an Jiaotong University (CN)
- University of Hong Kong (HK)
Publication Details
- Published
- 2024-12-04
- DOI
- https://doi.org/10.1145/3636534.3694723
- Primary Topic
- Machine Learning and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- National Science and Technology Major Project