Hard-Soft Pseudo Labels Guided Semi-Supervised Learning for Point Cloud Classification

Yuan He,Guyue Hu,Shan Yu
DOI: https://doi.org/10.1109/lsp.2024.3386115
2024-04-20
IEEE Signal Processing Letters
Abstract:Point clouds are widely applied in 3D visual sensing and perception. However, manually annotating point clouds is much more tedious and time-consuming than that for 2D images. Fortunately, semi-supervised learning can leverage massive unlabeled data to alleviate this issue, which is becoming a promising technique nowadays. In this letter, we propose a novel semi-supervised learning (SSL) framework for point cloud classification, named HPSSL. Its unsupervised learning branch performs both the representation embedding and pseudo-classification tasks. Specifically, both hard and soft pseudo labels of unlabeled samples are generated from a shared classifier to guide the class-aware contrastive learning in our SSL framework. Besides, a prediction consistency strategy is proposed to enhance the discrimination of feature representation and the exactness of pseudo labels. Furthermore, we force the supervised learning branch to interact with the unsupervised learning branch via distribution alignment, thus achieving representation consistency. Extensive experiments on three 3D shape recognition benchmarks demonstrate the effectiveness of the proposed approach.
engineering, electrical & electronic
What problem does this paper attempt to address?