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

Point-DAE:用于自监督点云学习的去噪自编码器

Point-DAE: Denoising Autoencoders for Self-supervised Point Cloud Learning

Department of Computing, The Hong Kong Polytechnic University(计算系,香港理工大学);School of Electronic and Information Engineering, South China University of Technology(电子与信息工程学院,华南理工大学)

Yabin Zhang, Jiehong Lin, Ruihuang Li, Kui Jia, Lei Zhang

AI总结:

本文提出Point-DAE,通过研究掩码之外的14种损坏类型(尤其是仿射变换),构建更通用的去噪自编码器进行自监督点云学习,并在多项3D任务上验证了其有效性。

摘要:

掩码自编码器在自监督点云学习中展示了其有效性。考虑到掩码是一种损坏,在这项工作中,我们通过研究掩码之外的更多损坏类型,探索了一种更通用的用于点云学习的去噪自编码器(Point-DAE)。具体而言,我们用特定的损坏来退化点云作为输入,并学习一个编码器-解码器模型,从其损坏版本中重建原始点云。我们使用传统的非Transformer编码器研究了三个损坏族(即密度/掩码、噪声和仿射变换)以及总共十四种损坏类型。除了流行的掩码损坏外,我们发现了另一个有效的损坏族,即仿射变换。仿射变换在全局上干扰所有点,这与丢弃某些局部区域的掩码损坏形成互补。我们还使用Transformer骨干网络验证了仿射变换损坏的有效性,其中我们将完整点云的重建分解为详细局部块和粗糙全局形状的重建,缓解了重建中的位置泄露问题。在对象分类、小样本学习、鲁棒性测试、部件分割和3D对象检测任务上的大量实验验证了所提方法的有效性。代码可在https://github.com/YBZh/Point-DAE获取。

英文摘要:

Masked autoencoder has demonstrated its effectiveness in self-supervised point cloud learning. Considering that masking is a kind of corruption, in this work we explore a more general denoising autoencoder for point cloud learning (Point-DAE) by investigating more types of corruptions beyond masking. Specifically, we degrade the point cloud with certain corruptions as input, and learn an encoder-decoder model to reconstruct the original point cloud from its corrupted version. Three corruption families (\ie, density/masking, noise, and affine transformation) and a total of fourteen corruption types are investigated with traditional non-Transformer encoders. Besides the popular masking corruption, we identify another effective corruption family, \ie, affine transformation. The affine transformation disturbs all points globally, which is complementary to the masking corruption where some local regions are dropped. We also validate the effectiveness of affine transformation corruption with the Transformer backbones, where we decompose the reconstruction of the complete point cloud into the reconstructions of detailed local patches and rough global shape, alleviating the position leakage problem in the reconstruction. Extensive experiments on tasks of object classification, few-shot learning, robustness testing, part segmentation, and 3D object detection validate the effectiveness of the proposed method. The codes are available at \url{https://github.com/YBZh/Point-DAE}.

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

↑