用3D高斯点云加速无监督预训练,提升效率与性能
Point Cloud Unsupervised Pre-training via 3D Gaussian Splatting
- 将3D高斯点云融入渲染框架,实现快速自监督学习
- 预训练速度提升9倍,内存占用不足原方法的1/4
- 适用于3D分割、检测与重建等下游任务
在标注数据有限的情况下,大规模无标签数据上的预训练能显著提升3D视觉模型性能。然而,现有基于渲染的自监督框架因体素渲染固有特性,存在计算开销大、内存消耗高的问题。本文提出高效框架GS$^3$,通过将快速3D高斯点云渲染无缝集成至渲染类框架中,实现点云表示学习。核心思路是:仅当高斯点携带丰富的几何与外观信息时,才能生成高质量图像,因此通过对比渲染图像与真实图像来优化点云编码器。具体流程为:将输入的RGB-D图像反投影至3D空间,由点云编码器提取点级特征;随后从学习到的特征预测3D高斯点,并使用基于瓦片的光栅化器进行图像渲染。预训练后的编码器可微调用于多种下游3D任务,包括3D分割、检测等高层感知任务及3D场景重建等低层任务。大量实验表明,该编码器具备强泛化能力,且所提框架效率极高,相较此前渲染框架Ponder,预训练速度提升约9倍,内存消耗低于0.25倍。
原文摘要 · Abstract (English)
Pre-training on large-scale unlabeled datasets contribute to the model achieving powerful performance on 3D vision tasks, especially when annotations are limited. However, existing rendering-based self-supervised frameworks are computationally demanding and memory-intensive during pre-training due to the inherent nature of volume rendering. In this paper, we propose an efficient framework named GS$^3$ to learn point cloud representation, which seamlessly integrates fast 3D Gaussian Splatting into the rendering-based framework. The core idea behind our framework is to pre-train the point cloud encoder by comparing rendered RGB images with real RGB images, as only Gaussian points enriched with learned rich geometric and appearance information can produce high-quality renderings. Specifically, we back-project the input RGB-D images into 3D space and use a point cloud encoder to extract point-wise features. Then, we predict 3D Gaussian points of the scene from the learned point cloud features and uses a tile-based rasterizer for image rendering. Finally, the pre-trained point cloud encoder can be fine-tuned to adapt to various downstream 3D tasks, including high-level perception tasks such as 3D segmentation and detection, as well as low-level tasks such as 3D scene reconstruction. Extensive experiments on downstream tasks demonstrate the strong transferability of the pre-trained point cloud encoder and the effectiveness of our self-supervised learning framework. In addition, our GS$^3$ framework is highly efficient, achieving approximately 9$\times$ pre-training speedup and less than 0.25$\times$ memory cost compared to the previous rendering-based framework Ponder.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。