arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.06110cs.CV

RISurConv:用于3D点云分类与分割的旋转不变表面注意力增强卷积

RISurConv: Rotation Invariant Surface Attention-Augmented Convolutions for 3D Point Cloud Classification and Segmentation

School of Computing and Information Systems, Singapore Management University(新加坡管理大学计算与信息系统学院);College of Information Science and Electronic Engineering, Zhejiang University(浙江大学信息与电子工程学院)

Zhiyuan Zhang, Licheng Yang, Zhiyu Xiang

AI总结:

针对现有3D点云深度学习旋转不变性研究不足、现有方法精度偏低的问题,提出构建局部三角面的RISurConv注意力增强卷积架构,在多个分类和分割基准上大幅超越现有SOTA。

摘要:

尽管3D点云深度学习领域已取得进展,但多数现有工作聚焦于学习对平移和点排列具有不变性的特征,针对旋转不变性的研究投入十分有限。近期部分研究虽实现了旋转不变性,却以降低精度为代价。本研究通过提出一种新颖且有效的旋转不变架构来弥补这一差距,该架构用于3D点云的分类与分割任务。我们摒弃传统的逐点操作,构建局部三角表面以捕捉更精细的表面结构,在此基础上提取高表达力的旋转不变表面属性,随后将这些属性整合到名为RISurConv的注意力增强卷积算子中,通过自注意力层生成经过优化的注意力特征。基于RISurConv,我们构建了一个高效的3D点云分析神经网络,该网络对任意旋转具有不变性,同时能保持高精度。我们在多个基准数据集上验证了模型性能,取得了远超现有最先进方法的优异结果:在ModelNet40数据集上总体准确率达96.0%(提升4.7%),在ScanObjectNN数据集上达93.1%(提升12.8%);在FG3D数据集的细粒度分类任务三个类别上,类别准确率分别为91.5%(提升3.6%)、82.7%(提升5.1%)和78.5%(提升9.2%)。此外,在ShapeNet数据集的分割任务上,我们取得了81.5%(提升1.0%)的平均交并比(mIoU)。代码可通过以下链接获取:https://github.com/cszyzhang/RISurConv

英文摘要:

Despite the progress on 3D point cloud deep learning, most prior works focus on learning features that are invariant to translation and point permutation, and very limited efforts have been devoted for rotation invariant property. Several recent studies achieve rotation invariance at the cost of lower accuracies. In this work, we close this gap by proposing a novel yet effective rotation invariant architecture for 3D point cloud classification and segmentation. Instead of traditional pointwise operations, we construct local triangle surfaces to capture more detailed surface structure, based on which we can extract highly expressive rotation invariant surface properties which are then integrated into an attention-augmented convolution operator named RISurConv to generate refined attention features via self-attention layers. Based on RISurConv we build an effective neural network for 3D point cloud analysis that is invariant to arbitrary rotations while maintaining high accuracy. We verify the performance on various benchmarks with supreme results obtained surpassing the previous state-of-the-art by a large margin. We achieve an overall accuracy of 96.0% (+4.7%) on ModelNet40, 93.1% (+12.8%) on ScanObjectNN, and class accuracies of 91.5% (+3.6%), 82.7% (+5.1%), and 78.5% (+9.2%) on the three categories of the FG3D dataset for the fine-grained classification task. Additionally, we achieve 81.5% (+1.0%) mIoU on ShapeNet for the segmentation task. Code is available here: https://github.com/cszyzhang/RISurConv

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑