DAPnet: A Double Self-Attention Convolutional Network for Point Cloud Semantic Labeling

被引:8
|
作者
Chen, Li [1 ]
Chen, Weiye [2 ,3 ]
Xu, Zewei [2 ,3 ]
Huang, Haozhe [1 ]
Wang, Shaowen [2 ,3 ]
Zhu, Qing [4 ]
Li, Haifeng [1 ]
机构
[1] Cent South Univ, Sch Geosci & Info Phys, Changsha 410083, Peoples R China
[2] Univ Illinois, Dept Geog & Geog Informat Sci, Champaign, IL 61820 USA
[3] Univ Illinois, CyberGIS Ctr Adv Digital & Spatial Studies, Champaign, IL 61820 USA
[4] Southwest Jiaotong Univ, Dept Geosci & Environm Engn, Chengdu 611756, Peoples R China
基金
中国国家自然科学基金;
关键词
Three-dimensional displays; Feature extraction; Semantics; Labeling; Shape; Training; Solid modeling; Airborne laser scanning (ALS) point clouds; convolutional neural network (CNN); self-attention; semantic labeling; DEEP FEATURES; LIDAR DATA; ALS;
D O I
10.1109/JSTARS.2021.3113047
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Airborne laser scanning (ALS) point clouds have complex structures, and their 3D semantic labeling has been a challenging task. It has three problems: 1) The difficulty of classifying point clouds around boundaries of objects from different classes, 2) the diversity of shapes within the same class, and 3) the scale differences between classes. In this study, we propose a novel double self-attention convolutional network called the DAPnet. The double self-attention includes the point attention module (PAM) and the group attention module (GAM). For problem 1), the PAM can effectively assign different weights based on the relevance of point clouds in adjacent areas. Meanwhile, for problem 2), the GAM enhances the correlation between groups, i.e., grouped features within the same classes. To solve problem 3), we adopt a multiscale radius to construct the groups and concatenate extracted hierarchical features with the output of the corresponding upsampling process. Under the ISPRS 3D Semantic Labeling Contest dataset, the DAPnet outperforms the benchmark by 85.2% with an overall accuracy of 90.7%. By conducting ablation comparisons, we find that the PAM effectively improves the model than the GAM. The incorporation of the double self-attention module has an average of 7% improvement on the pre-class accuracy. Plus, the DAPnet consumes a similar training time to those without the attention modules for model convergence. The DAPnet can assign different weights to features based on the relevance between point clouds and their neighbors, which effectively improves classification performance.
引用
收藏
页码:9680 / 9691
页数:12
相关论文
共 50 条
  • [1] Semantic segmentation of 3D point cloud based on self-attention feature fusion group convolutional neural network
    Yang, Jun
    Li, Bozan
    [J]. Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2022, 30 (07): : 840 - 853
  • [2] Deep Semantic Role Labeling with Self-Attention
    Tan, Zhixing
    Wang, Mingxuan
    Xie, Jun
    Chen, Yidong
    Shi, Xiaodong
    [J]. THIRTY-SECOND AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTIETH INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / EIGHTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2018, : 4929 - 4936
  • [3] Point cloud classification network based on self-attention mechanism
    Li, Yujie
    Cai, Jintong
    [J]. COMPUTERS & ELECTRICAL ENGINEERING, 2022, 104
  • [4] A Deep Neural Network Using Double Self-Attention Mechanism for ALS Point Cloud Segmentation
    Yu, Lili
    Yu, Haiyang
    Yang, Shuai
    [J]. IEEE ACCESS, 2022, 10 : 29878 - 29889
  • [5] 1D Self-Attention Network for Point Cloud Semantic Segmentation Using Omnidirectional LiDAR
    Suzuki, Takahiro
    Hirakawa, Tsubasa
    Yamashita, Takayoshi
    Fujiyoshi, Hironobu
    [J]. PATTERN RECOGNITION, ACPR 2021, PT I, 2022, 13188 : 257 - 270
  • [6] Cross self-attention network for 3D point cloud
    Wang, Gaihua
    Zhai, Qianyu
    Liu, Hong
    [J]. KNOWLEDGE-BASED SYSTEMS, 2022, 247
  • [7] Cyclic Self-attention for Point Cloud Recognition
    Zhu, Guanyu
    Zhou, Yong
    Yao, Rui
    Zhu, Hancheng
    Zhao, Jiaqi
    [J]. ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2023, 19 (01)
  • [8] Lightweight Self-Attention Network for Semantic Segmentation
    Zhou, Yan
    Zhou, Haibin
    Li, Nanjun
    Li, Jianxun
    Wang, Dongli
    [J]. 2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [9] A Multiscale Convolutional Neural Network With Color Vegetation Indices for Semantic Labeling of Point Cloud
    Zhang, Hua
    Ren, Kai
    Zheng, Nanshan
    Hao, Ming
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [10] Adaptive offset self-attention network for 3D point cloud
    Wang, Gaihua
    Wang, Nengyuan
    Li, Qi
    Liu, Hong
    Zhai, Qianyu
    [J]. JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (02)