DeFM: Learning Foundation Representations from Depth for Robotics

AK's Picks Embodied AI and Robotics SRT RP ML SSL
深度传感器已广泛部署于各类机器人平台;而快速、高保真深度图像仿真技术的进步,使得仅基于深度观测数据训练的机器人策略,得以在大量任务中实现稳健的“仿真到真实”(sim-to-real)迁移。尽管如此,相较于RGB模态——其大规模基础模型目前已代表该领域的最先进水平——针对深度模态的表征学习仍鲜有探索。为填补这一空白,我们提出了DeFM:一种专为机器人应用设计、完全基于深度图像进行自监督训练的基础模型。DeFM在精心筛选的6000万张深度图像数据集上,采用类似DINO的自蒸馏目标函数进行训练,从而习得兼具几何结构理解能力与语义感知能力的通用表征,可泛化至多样化的环境、任务及传感器类型。为确保模型在多尺度下始终具备对度量空间(metric space)的准确感知能力,我们提出了一种新颖的输入归一化策略。此外,我们还进一步将DeFM知识蒸馏为轻量化模型,以适配资源受限的机器人系统。在涵盖深度图像分类、分割、导航、运动控制与灵巧操作等多类基准任务的评测中,DeFM均达到当前最优性能,并展现出从仿真环境到真实世界场景的卓越泛化能力。我们已开源全部预训练模型,用户可直接将其用于基于深度图像的机器人学习任务,无需任何面向特定任务的微调。项目主页:https://de-fm.github.io/
Depth sensors are widely deployed across robotic platforms, and advances in fast, high-fidelity depth simulation have enabled robotic policies trained on depth observations to achieve robust sim-to-real transfer for a wide range of tasks. Despite this, representation learning for depth modality remains underexplored compared to RGB, where large-scale foundation models now define the state of the art. To address this gap, we present DeFM, a self-supervised foundation model trained entirely on depth images for robotic applications. Using a DINO-style self-distillation objective on a curated dataset of 60M depth images, DeFM learns geometric and semantic representations that generalize to diverse environments, tasks, and sensors. To retain metric awareness across multiple scales, we introduce a novel input normalization strategy. We further distill DeFM into compact models suitable for resource-constrained robotic systems. When evaluated on depth-based classification, segmentation, navigation, locomotion, and manipulation benchmarks, DeFM achieves state-of-the-art performance and demonstrates strong generalization from simulation to real-world environments. We release all our pretrained models, which can be adopted off-the-shelf for depth-based robotic learning without task-specific fine-tuning. Webpage: https://de-fm.github.io/
许愿