通过自洽训练提升机器人视觉泛化能力,适应不同视角光照和干扰物。
Invariance Co-training for Robot Visual Generalization
- 引入状态相似性与观测扰动不变性辅助任务,增强模型鲁棒性。
- 在未见视角、光照和干扰条件下性能提升18%,优于现有生成增强方法。
- 结合真实机器人示范数据与合成图像,降低对物理仿真依赖。
通用机器人策略需从多样观测中推理以应对复杂环境。尽管有进展,当前大规模机器人模型仍对相机视角、光照变化及干扰物敏感。我们认为其泛化能力受限于需覆盖的观测变异多样性,以及缺乏丰富多变的大规模机器人数据集。本文提出系统性解决方案:引入状态相似性和观测扰动不变性两个辅助任务,应用于示范数据与静态视觉数据。通过这些任务,利用高成本机器人示范数据与低成本、视觉丰富的非物理仿真合成图像(如Unreal Engine生成),显著提升对未见相机视角、光照配置及干扰物条件的泛化能力。结果表明,联合训练使性能较现有生成增强方法提升18%。更多信息与视频请访问 https://invariance-cotraining.github.io
原文摘要 · Abstract (English)
Reasoning from diverse observations is a fundamental capability for generalist robot policies to operate in a wide range of environments. Despite recent advancements, many large-scale robotic policies still remain sensitive to key sources of observational variation such as changes in camera perspective, lighting, and the presence of distractor objects. We posit that the limited generalizability of these models arises from the substantial diversity required to robustly cover these quasistatic axes, coupled with the current scarcity of large-scale robotic datasets that exhibit rich variation across them. In this work, we propose to systematically examine what robots need to generalize across these challenging axes by introducing two key auxiliary tasks, state similarity and invariance to observational perturbations, applied to both demonstration data and static visual data. We then show that via these auxiliary tasks, leveraging both more-expensive robotic demonstration data and less-expensive, visually rich synthetic images generated from non-physics-based simulation (for example, Unreal Engine) can lead to substantial increases in generalization to unseen camera viewpoints, lighting configurations, and distractor conditions. Our results demonstrate that co-training on this diverse data improves performance by 18 percent over existing generative augmentation methods. For more information and videos, please visit https://invariance-cotraining.github.io
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。