用SAM引导的掩码预测,提升3D场景理解的自监督学习效果
SAM-Guided Masked Token Prediction for 3D Scene Understanding
- 用SAM生成区域分割替代传统KNN,实现2D-3D特征对齐
- 通过分组平衡重加权缓解3D数据长尾分布问题
- 两阶段掩码令牌预测,让学生模型同时学全局与局部特征
基础模型显著提升了2D任务性能,近期工作如Bridge3D已通过知识蒸馏将此类模型应用于3D场景理解,取得重要进展。然而,2D与3D表示间的错位以及3D数据集中的长尾分布问题,仍限制了从2D到3D的知识蒸馏效果。为此,我们提出一种新型的SAM引导的标记化方法,无缝对接3D Transformer结构与基于区域的知识蒸馏,取代传统的基于KNN的标记化技术。同时,引入分组平衡重加权策略,有效应对知识蒸馏中的长尾问题。此外,受最近掩码特征预测成功的启发,我们的框架采用两阶段掩码标记预测机制:学生模型在第一阶段教师模型训练后,预测全局嵌入和基于教师模型提取的逐标记局部嵌入。该方法在SUN RGB-D、ScanNet和S3DIS等多个数据集上验证,涵盖3D目标检测与语义分割任务,结果显著优于现有自监督最先进方法,建立了该领域的全新基准。
原文摘要 · Abstract (English)
Foundation models have significantly enhanced 2D task performance, and recent works like Bridge3D have successfully applied these models to improve 3D scene understanding through knowledge distillation, marking considerable advancements. Nonetheless, challenges such as the misalignment between 2D and 3D representations and the persistent long-tail distribution in 3D datasets still restrict the effectiveness of knowledge distillation from 2D to 3D using foundation models. To tackle these issues, we introduce a novel SAM-guided tokenization method that seamlessly aligns 3D transformer structures with region-level knowledge distillation, replacing the traditional KNN-based tokenization techniques. Additionally, we implement a group-balanced re-weighting strategy to effectively address the long-tail problem in knowledge distillation. Furthermore, inspired by the recent success of masked feature prediction, our framework incorporates a two-stage masked token prediction process in which the student model predicts both the global embeddings and the token-wise local embeddings derived from the teacher models trained in the first stage. Our methodology has been validated across multiple datasets, including SUN RGB-D, ScanNet, and S3DIS, for tasks like 3D object detection and semantic segmentation. The results demonstrate significant improvements over current State-of-the-art self-supervised methods, establishing new benchmarks in this field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。