Point Distance Mask Attention Network for Learning on Point Cloud

YueHui Bao,Yu Gong,ShuKai Duan
DOI: https://doi.org/10.1117/12.2685007
2023-01-01
Abstract:Point cloud data has been widely used in many fields, and processing them using deep learning methods has become a popular research topic. However, the irregularity and unordered data structure of point cloud makes it necessary to design neural networks that are different from those used in image and natural language processing to accommodate its characteristics. In this paper, we propose a novel network called the Point Distance Mask Attention Network. This network utilizes the position relationship between the centroid and the generated neighborhood to weight the features of each point, making the network more focused on the features close to the centroid. Additionally, we use a masking operation to ensure the permutation invariance of the network when using the ball query method to find the neighborhood. Furthermore, we propose a residual-like connection in the network architecture, which achieves better results without changing the network feature extractor and depth. We evaluated the network model on the ModelNet40 dataset and achieved an accuracy of 93.7%. Our experiments show that our network achieves better results than the original baseline network by 1.1%, with almost the same parameters and inference time.
What problem does this paper attempt to address?