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

基于轻量级自注意模型的分层点云编码与解码

Hierarchical Point Cloud Encoding and Decoding with Lightweight Self-Attention based Model

Institute for Infocomm Research, A*STAR(新加坡科技研究局资讯通信研究院);Centre for Frontier AI Research, A*STAR(新加坡科技研究局前沿人工智能研究中心);Alibaba Group(阿里巴巴集团)

En Yen Puang, Hao Zhang, Hongyuan Zhu, Wei Jing

摘要:

在本文中,我们提出了SA-CNN,一个层次化的和轻量级的基于自注意力的编码和解码架构,用于点云数据的表示学习。拟议的SA-CNN引入卷积和转置卷积堆栈来捕获和生成无序3D点之间的上下文信息。遵循传统的分层流水线,编码过程以局部到全局的方式提取特征,而解码过程以粗到细的多分辨率阶段生成特征和点云。我们证明了SA-CNN能够广泛应用,即分类,部分分割,重建,形状检索和无监督分类。虽然在基准测试中达到了最先进或相当的性能,但SA-CNN保持了比其他人低几个数量级的模型复杂性。在定性结果方面,我们可视化的多阶段点云重建和潜在的行走刚性物体以及可变形的非刚性的人类和机器人模型。

英文摘要:

In this paper we present SA-CNN, a hierarchical and lightweight self-attention based encoding and decoding architecture for representation learning of point cloud data. The proposed SA-CNN introduces convolution and transposed convolution stacks to capture and generate contextual information among unordered 3D points. Following conventional hierarchical pipeline, the encoding process extracts feature in local-to-global manner, while the decoding process generates feature and point cloud in coarse-to-fine, multi-resolution stages. We demonstrate that SA-CNN is capable of a wide range of applications, namely classification, part segmentation, reconstruction, shape retrieval, and unsupervised classification. While achieving the state-of-the-art or comparable performance in the benchmarks, SA-CNN maintains its model complexity several order of magnitude lower than the others. In term of qualitative results, we visualize the multi-stage point cloud reconstructions and latent walks on rigid objects as well as deformable non-rigid human and robot models.

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

↑