针对点云分割计算量大问题,提出边界感知图网络提升效率与精度。
BAGNet: A Boundary-Aware Graph Attention Network for 3D Point Cloud Semantic Segmentation
- 只关注边界点构建图,减少计算开销。
- 在ScanNet上达90.1%精度,推理速度比SOTA快32%。
- 适合需要高效高精度点云分割的应用场景。
由于点云数据本身不规则且无结构,点云语义分割始终是一项挑战。基于图的方法试图将点云表示为图结构来建模,但需为大规模点云中每个点构建图,导致计算成本高昂。本文发现边界点蕴含更复杂的空间结构信息,提出一种新型图注意力网络——边界感知图注意力网络(BAGNet)。一方面,BAGNet包含边界感知图注意力层(BAGLayer),通过边顶点融合和注意力系数捕捉边界点特征,显著降低计算时间;另一方面,采用轻量级注意力池化层提取点云全局特征,维持模型精度。在标准数据集上的大量实验表明,BAGNet在点云语义分割任务中优于当前最优方法,兼具更高准确率和更短推理时间。
原文摘要 · Abstract (English)
Since the point cloud data is inherently irregular and unstructured, point cloud semantic segmentation has always been a challenging task. The graph-based method attempts to model the irregular point cloud by representing it as a graph; however, this approach incurs substantial computational cost due to the necessity of constructing a graph for every point within a large-scale point cloud. In this paper, we observe that boundary points possess more intricate spatial structural information and develop a novel graph attention network known as the Boundary-Aware Graph attention Network (BAGNet). On one hand, BAGNet contains a boundary-aware graph attention layer (BAGLayer), which employs edge vertex fusion and attention coefficients to capture features of boundary points, reducing the computation time. On the other hand, BAGNet employs a lightweight attention pooling layer to extract the global feature of the point cloud to maintain model accuracy. Extensive experiments on standard datasets demonstrate that BAGNet outperforms state-of-the-art methods in point cloud semantic segmentation with higher accuracy and less inference time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。