Directed Mix Contrast for Lidar Point Cloud Segmentation

Yuheng Lu,Fangping Chen,Ziwei Zhang,Fan Yang,Xiaodong Xie
DOI: https://doi.org/10.1109/icme52920.2022.9859891
2022-01-01
Abstract:Comprehensive and real-time scene understanding are crucial for autonomous driving, where LiDAR semantic segmentation plays an indispensable role. However, existing segmentation algorithms are exposed with extremely imbalanced dataset. Besides, due to the sparseness of point cloud, some hard negative samples are intrinsically similar. In this paper, we propose a contrastive learning framework named Directed Mix Contrast (DMC) for LiDAR point cloud segmentation. There are two key contributions in this framework. Firstly, a simple pre-processing step is proposed to maintain the balance between classes. In addition, to deal with the hard negative samples, contrastive learning strategy is introduced to facilitate feature learning. More specifically, we propose a directed mix-up training strategy to take control of the contrasting procedure. To demonstrate the effectiveness of DMC, we conduct experiments on two large-scale LiDAR Segmentation datasets and achieve mIoU of 71.0% on SemanticK-ITTI, which outperforms most existing methods. Furthermore, we extend our method to several backbone networks, and the results show good generalization ability of DMC.
What problem does this paper attempt to address?