Laser & Infrared, Volume. 54, Issue 8, 1216(2024)

Point cloud classification model based on graph neural network and attention mechanism

XU Hai-tao1,2,3,4, HAO Xiao-ping5, CHAO Xin5, DONG Shao-feng5, and LI Xiang1,2,4
Author Affiliations
  • 1Shenyang Institute of Automation, Chinese Academy of Sciences, Shenyang 110016, China
  • 2Institutes for Robotics and Intelligent Manufacturing, Chinese Academy of Sciences, Shenyang 110169, China
  • 3University of Chinese Academy of Sciences, Beijing 100049, China
  • 4Key Laboratory on Intelligent Detection and Equipment Technology of Liaoning Province, Shenyang 110179, China
  • 5China Aviation Engine Power Co., Ltd, Xi'an 710021, China
  • show less

    In order to enhance the modeling capability of global features in deep learning-based 3D point cloud classification models and improve their generalization performance, a point cloud classification model based on the fusion of graph neural network and attention mechanism is proposed on the basis of PointNet. Firstly, the extracted features are used to make the model pay more attention to the global context information, suppress the noise information, reduce the redundant parameters, and enhance the modelling ability of the global features by increasing the channel attention module and the spatial attention module, respectively. Secondly, different K-values nearest neighbor searches are performed within multiple scales of sphere radius to construct the input features for encoding, which not only reduces the scale of the graph and training overhead but also enables the model to learn features at different levels. Finally, neighborhood information is aggregated and node features are updated through graph convolutional neural networks. The output features of different graph convolutional neural network layers are summed up to fuse multi-level features and improve classification accuracy. The proposed model is trained and tested on the public dataset ModelNet40, achieving an overall classification accuracy of 88.6%, which outperforms the commonly used 3DShapeNets, VoxNet, ECC, and PointNet models, demonstrating its superiority in point cloud classification.

    Tools

    Get Citation

    Copy Citation Text

    XU Hai-tao, HAO Xiao-ping, CHAO Xin, DONG Shao-feng, LI Xiang. Point cloud classification model based on graph neural network and attention mechanism[J]. Laser & Infrared, 2024, 54(8): 1216

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Nov. 16, 2023

    Accepted: Apr. 30, 2025

    Published Online: Apr. 30, 2025

    The Author Email:

    DOI:10.3969/j.issn.1001-5078.2024.08.006

    Topics