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 条
[31]   Airborne lidar change detection: An overview of Earth sciences applications [J].
Okyay, Unal ;
Telling, Jennifer ;
Glennie, Craig L. ;
Dietrich, William E. .
EARTH-SCIENCE REVIEWS, 2019, 198
[32]   Combining Active and Semisupervised Learning of Remote Sensing Data Within a Renyi Entropy Regularization Framework [J].
Polewski, Przemyslaw ;
Yao, Wei ;
Heurich, Marco ;
Krzystek, Peter ;
Stilla, Uwe .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2016, 9 (07) :2910-2922
[33]   Detection of fallen trees in ALS point clouds using a Normalized Cut approach trained by simulation [J].
Polewski, Przemyslaw ;
Yao, Wei ;
Heurich, Marco ;
Krzystek, Peter ;
Stilla, Uwe .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2015, 105 :252-271
[34]  
Qi CR, 2017, ADV NEUR IN, V30
[35]   PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation [J].
Qi, Charles R. ;
Su, Hao ;
Mo, Kaichun ;
Guibas, Leonidas J. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :77-85
[36]   Volumetric and Multi-View CNNs for Object Classification on 3D Data [J].
Qi, Charles R. ;
Su, Hao ;
Niessner, Matthias ;
Dai, Angela ;
Yan, Mengyuan ;
Guibas, Leonidas J. .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :5648-5656
[37]   Semantic Labeling of ALS Point Cloud via Learning Voxel and Pixel Representations [J].
Qin, Nannan ;
Hu, Xiangyun ;
Wang, Puzuo ;
Shan, Jie ;
Li, Yijing .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (05) :859-863
[38]   RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Clouds [J].
Hu, Qingyong ;
Yang, Bo ;
Xie, Linhai ;
Rosa, Stefano ;
Guo, Yulan ;
Wang, Zhihua ;
Trigoni, Niki ;
Markham, Andrew .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :11105-11114
[39]  
Rizaldy A., 2018, ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences, VIV-2, P231, DOI [DOI 10.5194/ISPRS-ANNALS-IV-2-231-2018, 10.5194/isprs-annals-IV-2-231-2018]
[40]  
Rottensteiner F., 2012, ISPRS Annals of the Photogrammetry, Remote Sensing and Spatial Information Sciences, VI, P293, DOI [DOI 10.5194/ISPRSANNALS-I-3-293-2012, 10.5194/isprsannals-I-3-293-2012]