arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.08994cs.CV

面向分类与分割的全局注意力引导双域点云特征学习

Global Attention-Guided Dual-Domain Point Cloud Feature Learning for Classification and Segmentation

College of Artificial Intelligence, Nanjing University of Aeronautics and Astronautics(南京航空航天大学人工智能学院);School of Computing, Mathematics, and Engineering, Charles Sturt University(查尔斯斯特大学计算、数学与工程学院);George Mason University(乔治梅森大学)

Zihao Li, Pan Gao, Kang You, Chuan Yan, Manoranjan Paul

AI总结:

针对点云分析中原始点输入嵌入效率低、邻域聚合效果有限的问题,提出全局注意力引导的双域特征学习网络GAD,通过CPT模块生成全局感知嵌入、DKFF模块实现双域特征聚合,在多类点云任务上取得更优性能。

摘要:

已有研究已证明基于点的神经模型在点云分析任务中的有效性。然而,在为原始点坐标生成高效输入嵌入方面仍存在关键问题。此外,另一个问题在于邻域聚合的效率有限,而邻域聚合是网络主干中的关键组件。本文提出一种全局注意力引导的双域特征学习网络(Global Attention-guided Dual-domain Feature Learning network,GAD)以解决上述问题。我们首先设计了上下文位置增强Transformer(Contextual Position-enhanced Transformer,CPT)模块,该模块配备改进的全局注意力机制,用于生成具有全局感知能力的输入嵌入,作为后续聚合的引导。随后,级联双域K近邻特征融合(Dual-domain K-nearest neighbor Feature Fusion,DKFF)模块,通过兼顾局部几何关系与长距离语义关联的新型双域特征学习实现有效特征聚合。在多个点云分析任务(例如分类、部件分割和场景语义分割)上开展的大量实验证明了所提方法的优越性能以及所设计模块的有效性。

英文摘要:

Previous studies have demonstrated the effectiveness of point-based neural models on the point cloud analysis task. However, there remains a crucial issue on producing the efficient input embedding for raw point coordinates. Moreover, another issue lies in the limited efficiency of neighboring aggregations, which is a critical component in the network stem. In this paper, we propose a Global Attention-guided Dual-domain Feature Learning network (GAD) to address the above-mentioned issues. We first devise the Contextual Position-enhanced Transformer (CPT) module, which is armed with an improved global attention mechanism, to produce a global-aware input embedding that serves as the guidance to subsequent aggregations. Then, the Dual-domain K-nearest neighbor Feature Fusion (DKFF) is cascaded to conduct effective feature aggregation through novel dual-domain feature learning which appreciates both local geometric relations and long-distance semantic connections. Extensive experiments on multiple point cloud analysis tasks (e.g., classification, part segmentation, and scene semantic segmentation) demonstrate the superior performance of the proposed method and the efficacy of the devised modules.

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

↑