用3D高斯点云实现高效自监督场景表征学习
Gaussian2Scene: 3D Scene Representation Learning via Self-supervised Learning with 3D Gaussian Splatting
- 设计双分支掩码自编码器,联合学习2D与3D场景表征
- 通过高斯点云重建与渲染图像联合监督,提升几何理解能力
- 适合需要精确3D结构理解的下游任务,如物体检测
自监督学习(SSL)已成为众多3D视觉任务的基础,可从大规模无标注数据中有效学习。现有场景级SSL方法常结合体素渲染,以RGB-D图像为重建信号进行跨模态学习,促进2D与3D模态对齐并利用丰富的视觉线索。然而,这些方法依赖隐式场景表示,计算开销大,且仅在2D空间施加重建目标,难以捕捉深层3D几何结构。为此,我们提出Gaussian2Scene,一种基于3D高斯点云(3DGS)的新型场景级自监督学习框架。3DGS兼具高效性与显式表示特性,降低体积渲染带来的计算负担,支持直接3D场景重建,增强主干网络的几何理解能力。该方法采用渐进式两阶段训练:第一阶段,双分支掩码自编码器学习2D与3D场景表示;第二阶段,以重建点云初始化训练,并利用高斯原语的空间位置和渲染图像进一步监督学习,强化几何与跨模态学习。我们在多个下游3D物体检测任务上验证了Gaussian2Scene的有效性,性能持续优于现有预训练方法。
原文摘要 · Abstract (English)
Self-supervised learning (SSL) for point cloud pre-training has become a cornerstone for many 3D vision tasks, enabling effective learning from large-scale unannotated data. At the scene level, existing SSL methods often incorporate volume rendering into the pre-training framework, using RGB-D images as reconstruction signals to facilitate cross-modal learning. This strategy promotes alignment between 2D and 3D modalities and enables the model to benefit from rich visual cues in the RGB-D inputs. However, these approaches are limited by their reliance on implicit scene representations and high memory demands. Furthermore, since their reconstruction objectives are applied only in 2D space, they often fail to capture underlying 3D geometric structures. To address these challenges, we propose Gaussian2Scene, a novel scene-level SSL framework that leverages the efficiency and explicit nature of 3D Gaussian Splatting (3DGS) for pre-training. The use of 3DGS not only alleviates the computational burden associated with volume rendering but also supports direct 3D scene reconstruction, thereby enhancing the geometric understanding of the backbone network. Our approach follows a progressive two-stage training strategy. In the first stage, a dual-branch masked autoencoder learns both 2D and 3D scene representations. In the second stage, we initialize training with reconstructed point clouds and further supervise learning using the geometric locations of Gaussian primitives and rendered RGB images. This process reinforces both geometric and cross-modal learning. We demonstrate the effectiveness of Gaussian2Scene across several downstream 3D object detection tasks, showing consistent improvements over existing pre-training methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。