ProUDA: Progressive Unsupervised Data Augmentation for Semi-Supervised 3D Object Detection on Point Cloud

Pei An,Junxiong Liang,Tao Ma,Yanfei Chen,Liheng Wang,Jie Ma
DOI: https://doi.org/10.1016/j.patrec.2023.04.002
IF: 4.757
2023-04-17
Pattern Recognition Letters
Abstract:Unsupervised data augmentation (UDA) is a simple and general semi-supervised learning (SSL) framework. However, for the task of semi-supervised 3D object detection (SSOD-3D), due to the impact of object occlusion and point cloud resolution, the quality of pseudo labels is uncertain so that the performance of UDA is limited. In this paper, We propose an efficient and novel SSL framework pro gressive u nsupervised d ata a ugmentation (ProUDA). At first, to minimize the overfitting risk of the inaccurate pseudo labels, we sort the unlabeled samples by prediction complexity, and present a progressive consistency loss to adjust the usage ratio of the unlabeled samples. After that, we employ a strategy of iteration check to select the best learning result using a few but representative validation dataset annotated from the unlabeled samples. It ensures the safe model weight updating. Extensive experiments are conducted on both the public indoor and outdoor 3D object detection datasets. Results demonstrate that ProUDA has better 3D average precision than UDA and the proposed method benefits to 3D object detector training.
computer science, artificial intelligence
What problem does this paper attempt to address?