Journal of East China Normal University(Natural Science) ›› 2024, Vol. 2024 ›› Issue (2): 108-118.doi: 10.3969/j.issn.1000-5641.2024.02.012

• Computer Science • Previous Articles    

Group contrastive learning for weakly-supervised 3D point cloud semantic segmentation

Zhihong ZHENG, Haichuan SONG*()   

  1. 1. School of Computer Science and Technology, East China Normal University, Shanghai 200062, China
  • Received:2023-02-01 Online:2024-03-25 Published:2024-03-18
  • Contact: Haichuan SONG


Three-dimensional point cloud semantic segmentation is an essential task for 3D visual perception and has been widely used in autonomous driving, augmented reality, and robotics. However, most methods work under a fully-supervised setting, which heavily relies on fully annotated datasets. Many weakly-supervised methods have utilized the pseudo-labeling method to retrain the model and reduce the labeling time consumption. However, the previous methods have failed to address the conformation bias induced by false pseudo labels. In this study, we proposed a novel weakly-supervised 3D point cloud semantic segmentation method based on group contrastive learning, constructing contrast between positive and negative sample groups selected from pseudo labels. The pseudo labels will compete with each other within the group contrastive learning, reducing the gradient contribution of falsely predicted pseudo labels. Results on three large-scale datasets show that our method outperforms state-of-the-art weakly-supervised methods with minimal labeling annotations and even surpasses the performance of some classic fully-supervised methods.

Key words: weakly-supervised learning, 3D point cloud, semantic segmentation, contrastive learning

CLC Number: