Laser & Infrared, Volume. 54, Issue 8, 1216(2024)
Point cloud classification model based on graph neural network and attention mechanism
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.
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
Category:
Received: Nov. 16, 2023
Accepted: Apr. 30, 2025
Published Online: Apr. 30, 2025
The Author Email: