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

基于特征空间聚类的内容型Transformer点云分类方法

Point Cloud Classification Using Content-based Transformer via Clustering in Feature Space

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Purdue School of Engineering and Technology, Indiana University-Purdue University Indianapolis(印第安纳大学-普渡大学印第安纳波利斯分校普渡工程与技术学院)

Yahui Liu, Bin Tian, Yisheng Lv, Lingxi Li, Feiyue Wang

AI总结:

针对现有3D点云分类Transformer方法依赖局部空间注意力、忽略内容且难建立远点关联的问题,提出基于特征空间聚类的内容型PointConT架构,结合Inception特征聚合器,在ScanObjectNN最难设置下达90.3% Top-1准确率。

摘要:

近年来,Transformer已被尝试应用于三维点云分类任务。为降低计算量,现有方法大多聚焦于局部空间注意力,却忽略了点的内容特征,也无法建立距离较远但具有相关性的点之间的联系。针对局部空间注意力的局限,我们提出一种基于点内容的Transformer架构,简称PointConT。该架构利用特征空间中点的局部性(基于内容),将特征相似的采样点聚为同一类,并在每个类别内计算自注意力,从而在捕获长距离依赖关系与控制计算复杂度之间实现有效权衡。我们还引入了用于点云分类的Inception特征聚合器,采用并行结构在各分支中分别聚合高频与低频信息。大量实验表明,我们的PointConT模型在点云形状分类任务上取得了优异性能。尤其在ScanObjectNN的最难设置下,该方法的Top-1准确率达到90.3%。本文源代码可在https://github.com/yahuiliu99/PointConT 获取。

英文摘要:

Recently, there have been some attempts of Transformer in 3D point cloud classification. In order to reduce computations, most existing methods focus on local spatial attention, but ignore their content and fail to establish relationships between distant but relevant points. To overcome the limitation of local spatial attention, we propose a point content-based Transformer architecture, called PointConT for short. It exploits the locality of points in the feature space (content-based), which clusters the sampled points with similar features into the same class and computes the self-attention within each class, thus enabling an effective trade-off between capturing long-range dependencies and computational complexity. We further introduce an Inception feature aggregator for point cloud classification, which uses parallel structures to aggregate high-frequency and low-frequency information in each branch separately. Extensive experiments show that our PointConT model achieves a remarkable performance on point cloud shape classification. Especially, our method exhibits 90.3% Top-1 accuracy on the hardest setting of ScanObjectNN. Source code of this paper is available at https://github.com/yahuiliu99/PointConT.

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

↑