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

GS-PT:通过自监督学习利用3D Gaussian Splatting实现全面的点云理解

GS-PT: Exploiting 3D Gaussian Splatting for Comprehensive Point Cloud Understanding via Self-supervised Learning

School of Computer Science, Fudan University(复旦大学计算机科学学院);School of Mechanical Engineering, Tongji University(同济大学机械工程学院);Academy for Engineering and Technology, Fudan University(复旦大学工程技术研究院)

Keyi Liu, Yeqi Luo, Weidong Yang, Jingyi Xu, Zhijun Li, Wen-Ming Chen, Ben Fei

AI总结:

针对点云自监督学习中数据多样性和增强不足的问题,本文提出GS-PT,首次结合3DGS、transformer掩码重建与多模态对比学习,并在多项下游任务上取得优于现有方法的性能。

摘要:

点云的自监督学习旨在利用未标注的3D数据学习有意义的表示,而不依赖人工标注。然而,当前方法面临数据多样性有限、增强不足以支持有效特征学习等挑战。为应对这些挑战,我们提出GS-PT,首次将3D Gaussian Splatting(3DGS)集成到点云自监督学习中。我们的流程使用transformers作为自监督预训练的骨干网络,并通过3DGS引入新的对比学习任务。具体而言,transformers旨在重建被掩码的点云。3DGS使用多视角渲染图像作为输入,生成增强的点云分布和新视角图像,从而促进数据增强和跨模态对比学习。此外,我们还融合来自深度图的特征。通过联合优化这些任务,我们的方法丰富了三模态自监督学习过程,使模型能够利用3D点云与来自多种模态的2D图像之间的相关性。预训练后我们冻结编码器,并在多个下游任务上测试模型性能。实验结果表明,GS-PT在3D物体分类、真实世界分类以及小样本学习和分割等多种下游任务上优于现有的自监督学习方法。

英文摘要:

Self-supervised learning of point cloud aims to leverage unlabeled 3D data to learn meaningful representations without reliance on manual annotations. However, current approaches face challenges such as limited data diversity and inadequate augmentation for effective feature learning. To address these challenges, we propose GS-PT, which integrates 3D Gaussian Splatting (3DGS) into point cloud self-supervised learning for the first time. Our pipeline utilizes transformers as the backbone for self-supervised pre-training and introduces novel contrastive learning tasks through 3DGS. Specifically, the transformers aim to reconstruct the masked point cloud. 3DGS utilizes multi-view rendered images as input to generate enhanced point cloud distributions and novel view images, facilitating data augmentation and cross-modal contrastive learning. Additionally, we incorporate features from depth maps. By optimizing these tasks collectively, our method enriches the tri-modal self-supervised learning process, enabling the model to leverage the correlation across 3D point clouds and 2D images from various modalities. We freeze the encoder after pre-training and test the model's performance on multiple downstream tasks. Experimental results indicate that GS-PT outperforms the off-the-shelf self-supervised learning methods on various downstream tasks including 3D object classification, real-world classifications, and few-shot learning and segmentation.

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

↑