Abstract
In this paper, we propose an attention-based dynamic graph CNN method for point cloud classification. We introduce an efficient channel attention module into each edge convolution block of dynamic graph CNN (DGCNN) to obtain more discriminative and stable features. Our experimental results show that, compared with DGCNN, our method improves not only the classification accuracy but also the stability.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Qi, C.R., Su, H., Mo, K., et al.: PointNet: deep learning on point sets for 3D classification and segmentation. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, USA, pp. 652–660 (2017)
Qi, C.R., Yi, L., Su, H., et al.: Pointnet++: deep hierarchical feature learning on point sets in a metric space. In: Advances in Neural Information Processing Systems (NIPS), Long Beach, USA (2017)
Wang, Y., Sun, Y., Liu, Z., et al.: Dynamic Graph CNN (DGCNN) for learning on point clouds. ACM Trans. Graph. 38(5), 1–12 (2019)
Vaswani, A., Shazeer, N., Parmar, N., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, Long Beach, USA, p. 30 (2017)
Wang, Q., Wu, B., Zhu, P., et al.: ECA-Net: efficient channel attention for deep convolutional neural networks. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, USA (2020)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, USA, pp. 7132–7141 (2018)
Woo, S., Park, J., Lee, J.-Y., Kweon, I.S.: CBAM: convolutional block attention module. In: Ferrari, V., Hebert, M., Sminchisescu, C., Weiss, Y. (eds.) ECCV 2018. LNCS, vol. 11211, pp. 3–19. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-01234-2_1
Cao, Y., Xu, J., Lin, S., et al.: GCNet: non-local networks meet squeeze-excitation networks and beyond. In: Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops (ICCV), Seoul, Korea (2019)
Wu, Z., Song, S., Khosla, A., et al.: 3D ShapeNets: a deep representation for volumetric shapes. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Boston, USA, pp. 1912–1920 (2015)
Shen, Y., Feng, C., Yang, Y., et al.: Mining point cloud local structures by kernel correlation and graph pooling. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, USA, pp. 4548–4557 (2018)
Duan, Y., Zheng, Y., Lu, J., et al.: Structural Relational Reasoning (SRN) of point clouds. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, USA, pp. 949–958 (2019)
Zhao, H., Jiang, L., Fu, C.W., et al.: PointWeb: enhancing local neighborhood features for point cloud processing. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, USA, pp. 5565–5573 (2019)
Yan, X., Zheng, C., Li, Z., et al.: PointASNL: robust point clouds processing using nonlocal neural networks with adaptive sampling. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 5589–5598 (2020)
Xu, Q., Sun, X., Wu, C.Y., et al.: Grid-GCN for fast and scalable point cloud learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 5661–5670 (2020)
Guo, M.H., Cai, J.X., Liu, Z.N., et al.: PCT: point cloud transformer. Comput. Vis. Med. 7(2), 187–199 (2021)
Xu, M., Zhang, J., Zhou, Z., et al.: Learning geometry-disentangled representation for complementary understanding of 3D object point cloud. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, no. 4, pp. 3056–3064 (2021)
Xu, M., Ding, R., Zhao, H., et al.: PAConv: Position Adaptive Convolution with dynamic kernel assembling on point clouds. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 3173–3182 (2021)
Zhao, H., Jiang, L., Jia, J., et al.: Point transformer. In: Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), pp. 16259–16268 (2021)
Ma, X., Qin, C., You, H., et al.: Rethinking network design and local geometry in point cloud: a simple residual MLP framework. In: Tenth International Conference on Learning Representations (ICLR) (2022)
Zheng, Y., Yang, S., Li, Y., Lu, H.: Global-PBNet: a novel point cloud registration for autonomous driving. IEEE Trans. Intell. Transp. Syst. 23, 22312–22319 (2022)
Li, Y., Yang, S., Zheng, Y., Lu, H.: Improved point-voxel region convolutional neural network: 3D object detectors for autonomous driving. IEEE Trans. Intell. Transp. Syst. 23, 9311–9317 (2021)
Lu, H., Yang, R., Deng, Z., Zhang, Y., Gao, G., Lan, R.: Chinese image captioning via fuzzy attention-based DenseNet-BiLSTM. ACM Trans. Multimed. Comput. Commun. Appl. 17(1s), 1–18 (2021)
Lu, H., Zhang, Y., Li, Y., Jiang, C., Abbas, H.: User-oriented virtual mobile network resource management for vehicle communications. IEEE Trans. Intell. Transp. Syst. 22, 3521–3532 (2020). https://doi.org/10.1109/TITS.2020.2991766
Acknowledgment
This work was supported by the State Key Laboratory of Precision Blasting, Jianghan University (No. PBSKL2022201), the Scientific Research Program of Jianghan University (No. 2021yb052), the Graduate Innovation Fund of the Jianghan University, the National College Students’ Innovation and Entrepreneurship Training Program of China (Grant No. S202211072042), and the Undergraduate Research Projects of the Jianghan University (Grant Nos. 2021Bczd006, 2021Bczd007, and 2022zd096).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Wang, J. et al. (2022). Attention-Based Dynamic Graph CNN for Point Cloud Classification. In: Yang, S., Lu, H. (eds) Artificial Intelligence and Robotics. ISAIR 2022. Communications in Computer and Information Science, vol 1700. Springer, Singapore. https://doi.org/10.1007/978-981-19-7946-0_30
Download citation
DOI: https://doi.org/10.1007/978-981-19-7946-0_30
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-19-7945-3
Online ISBN: 978-981-19-7946-0
eBook Packages: Computer ScienceComputer Science (R0)