用3D高斯点云渲染增强点云自监督学习,提升模型泛化能力。
GS-PT: Exploiting 3D Gaussian Splatting for Comprehensive Point Cloud Understanding via Self-supervised Learning
- 利用3D高斯溅射生成多视角图像和增强点云,实现跨模态数据增强
- 通过对比学习在点云与图像间建立关联,提升特征表示能力
- 在分类、分割等下游任务中优于现有自监督方法,尤其适合少样本场景
点云自监督学习旨在利用无标注的3D数据学习有意义的表征,避免依赖人工标注。然而,现有方法面临数据多样性不足、增强手段有限等问题。为此,我们提出GS-PT,首次将3D高斯溅射(3DGS)引入点云自监督学习。该方法以Transformer为骨干网络进行自监督预训练,通过3DGS生成多视角渲染图像与增强点云分布,设计新型对比学习任务。具体而言,模型需重建被遮蔽的点云,同时利用多视图渲染图像生成新视角图像,实现跨模态对比学习,并融合深度图特征。通过联合优化这些任务,丰富了三模态自监督学习过程,使模型能有效利用点云与图像间的多模态相关性。预训练后冻结编码器,在多个下游任务上测试性能。实验结果表明,GS-PT在3D物体分类、真实世界分类及少样本学习与分割任务中均优于现有自监督方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。