提出SlotMIM方法,让视觉模型更好理解机器人场景中的物体
A Data-Centric Revisit of Pre-Trained Vision Models for Robot Learning
- 用语义瓶颈和跨视图正则化强制模型学习物体为中心表征
- 在多类数据上训练均优于现有预训练模型,尤其在物体识别任务提升显著
- 适合做机器人视觉感知与迁移学习的研究者参考
预训练视觉模型(PVMs)是现代机器人技术的核心,但其最优配置仍不明确。系统评估发现,尽管DINO和iBOT在视觉运动控制与感知任务中优于MAE,但在非物体中心(NOC)数据上表现不佳,这与其学习物体中心表征能力下降强相关。研究指出,从非物体中心的机器人数据中形成物体中心表征是成功的关键。为此,我们设计了SlotMIM方法,通过引入语义瓶颈减少原型数量以促进物体性显现,并加入跨视图一致性正则化增强多视角不变性。实验涵盖物体中心、场景中心、网络爬取和自我中心数据的预训练。在所有设置下,该方法均学习到可迁移的表征,在图像识别、场景理解及机器人学习评估中显著优于先前工作。当扩展至百万级数据集时,该方法展现出更优的数据效率与可扩展性。代码与模型已公开于https://github.com/CVMI-Lab/SlotMIM。
原文摘要 · Abstract (English)
Pre-trained vision models (PVMs) are fundamental to modern robotics, yet their optimal configuration remains unclear. Through systematic evaluation, we find that while DINO and iBOT outperform MAE across visuomotor control and perception tasks, they struggle when trained on non-(single-)object-centric (NOC) data--a limitation strongly correlated with their diminished ability to learn object-centric representations. This investigation indicates that the ability to form object-centric representations from the non-object-centric robotics dataset is the key to success for PVMs. Motivated by this discovery, we designed SlotMIM, a method that induces object-centric representations by introducing a semantic bottleneck to reduce the number of prototypes to encourage the emergence of objectness as well as cross-view consistency regularization for encouraging multiview invariance. Our experiments encompass pre-training on object-centric, scene-centric, web-crawled, and ego-centric data. Across all settings, our approach learns transferrable representations and achieves significant improvements over prior work in image recognition, scene understanding, and robot learning evaluations. When scaled up with million-scale datasets, our method also demonstrates superior data efficiency and scalability. Our code and models are publicly available at https://github.com/CVMI-Lab/SlotMIM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。