A new weakly supervised approach for ALS point cloud semantic segmentation

被引:43
作者
Wang, Puzuo [1 ]
Yao, Wei [1 ,2 ]
机构
[1] Hong Kong Polytech Univ, Dept Land Surveying & Geoinformat, Hung Hom, Hong Kong, Peoples R China
[2] Hong Kong Polytech Univ, Shenzhen Res Inst, Shenzhen, Peoples R China
基金
中国国家自然科学基金;
关键词
Point cloud semantic segmentation; Weakly supervised learning; Entropy regularization; Consistency constraint; Pseudo-label; ATTENTIONAL NETWORK; CLASSIFICATION;
D O I
10.1016/j.isprsjprs.2022.04.016
中图分类号
P9 [自然地理学];
学科分类号
0705 ; 070501 ;
摘要
Although novel point cloud semantic segmentation schemes that continuously surpass state-of-the-art results exist, the success of learning an effective model typically relies on the availability of abundant labeled data. However, data annotation is a time-consumng and labor-intensive task, particularly for large-scale airborne laser scanning (ALS) point clouds involving multiple classes in urban areas. Therefore, simultaneously obtaining promising results while significantly reducing labeling is crucial. In this study, we propose a deep-learning-based weakly supervised framework for the semantic segmentation of ALS point clouds. This is to exploit implicit information from unlabeled data subject to incomplete and sparse labels. Entropy regularization is introduced to penalize class overlap in the predictive probability. Additionally, a consistency constraint is designed to improve the robustness of the predictions by minimizing the difference between the current and ensemble predictions. Finally, we propose an online soft pseudo-labeling strategy to create additional supervisory sources in an efficient and nonparametric manner. Extensive experimental analysis using three benchmark datasets demonstrates that our proposed method significantly boosts the classification performance without compromising the computational efficiency, considering the sparse point annotations. It outperforms the current weakly supervised methods and achieves a result comparable to that of full supervision competitors. Considering the ISPRS Vaihingen 3D data, using only 1 parts per thousand labels, our method achieved an overall accuracy of 83.0% and an average F1 score of 70.0%. These increased by 6.9% and 12.8%, respectively, compared to the model trained only using sparse label information.
引用
收藏
页码:237 / 254
页数:18
相关论文
共 65 条
[1]   Object-based land cover classification using airborne LiDAR [J].
Antonarakis, A. S. ;
Richards, K. S. ;
Brasington, J. .
REMOTE SENSING OF ENVIRONMENT, 2008, 112 (06) :2988-2998
[2]   What's the Point: Semantic Segmentation with Point Supervision [J].
Bearman, Amy ;
Russakovsky, Olga ;
Ferrari, Vittorio ;
Fei-Fei, Li .
COMPUTER VISION - ECCV 2016, PT VII, 2016, 9911 :549-565
[3]  
Berthelot D, 2019, ADV NEUR IN, V32
[4]   SnapNet: 3D point cloud semantic labeling with 2D deep segmentation networks [J].
Boulch, Alexandre ;
Guerry, Yids ;
Le Saux, Bertrand ;
Audebert, Nicolas .
COMPUTERS & GRAPHICS-UK, 2018, 71 :189-198
[5]   The DGPF-Test on Digital Airborne Camera Evaluation - Overview and Test Design [J].
Cramer, Michael .
PHOTOGRAMMETRIE FERNERKUNDUNG GEOINFORMATION, 2010, (02) :73-82
[6]   Segmentation of Sloped Roofs from Airborne LiDAR Point Clouds Using Ridge-Based Hierarchical Decomposition [J].
Fan, Hongchao ;
Yao, Wei ;
Fu, Qing .
REMOTE SENSING, 2014, 6 (04) :3284-3301
[7]   Dual Attention Network for Scene Segmentation [J].
Fu, Jun ;
Liu, Jing ;
Tian, Haijie ;
Li, Yong ;
Bao, Yongjun ;
Fang, Zhiwei ;
Lu, Hanqing .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :3141-3149
[8]   Are We Hungry for 3D LiDAR Data for Semantic Segmentation? A Survey of Datasets and Methods [J].
Gao, Biao ;
Pan, Yancheng ;
Li, Chengkun ;
Geng, Sibo ;
Zhao, Huijing .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (07) :6063-6081
[9]  
Grandvalet Y., 2005, Advances in Neural Information Processing Systems, P529
[10]   WEAKLY SUPERVISED SEGMENTATION-AIDED CLASSIFICATION OF URBAN SCENES FROM 3D LIDAR POINT CLOUDS [J].
Guinard, Stephan ;
Landrieu, Loic .
ISPRS HANNOVER WORKSHOP: HRIGI 17 - CMRT 17 - ISA 17 - EUROCOW 17, 2017, 42-1 (W1) :151-157