arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.05085cs.RO
收藏

未来的关键点:基于模型的强化学习中的自监督对应

Keypoints into the Future: Self-Supervised Correspondence in Model-Based Reinforcement Learning

CSAIL, Massachusetts Institute of Technology(麻省理工学院计算机科学与人工智能实验室);Robotics at Google(谷歌机器人部门)

Lucas Manuelli, Yunzhu Li, Pete Florence, Russ Tedrake

摘要:

预测模型一直是许多机器人系统的核心,从四旋翼机器人到步行机器人。然而,它一直具有挑战性的开发和应用这些模型实际的机器人操作由于高维的感官观察,如图像。以前在机器人操作的背景下学习模型的方法要么学习整个图像动态,要么使用自动编码器来学习低维潜在状态下的动态。在这项工作中,我们引入了基于模型的预测和自监督视觉对应学习,并证明这不仅是可能的,而且证明了这些类型的预测模型与基于视觉的RL的替代方法相比具有引人注目的性能改进。通过仿真实验,我们证明了我们的模型提供了更好的泛化精度,特别是在3D场景,场景涉及遮挡,并在类别泛化。此外,我们验证了我们的方法有效地转移到现实世界中,通过硬件实验。视频和补充材料,请访问https://sites.google.com/view/keypointsintothefuture

英文摘要:

Predictive models have been at the core of many robotic systems, from quadrotors to walking robots. However, it has been challenging to develop and apply such models to practical robotic manipulation due to high-dimensional sensory observations such as images. Previous approaches to learning models in the context of robotic manipulation have either learned whole image dynamics or used autoencoders to learn dynamics in a low-dimensional latent state. In this work, we introduce model-based prediction with self-supervised visual correspondence learning, and show that not only is this indeed possible, but demonstrate that these types of predictive models show compelling performance improvements over alternative methods for vision-based RL with autoencoder-type vision training. Through simulation experiments, we demonstrate that our models provide better generalization precision, particularly in 3D scenes, scenes involving occlusion, and in category-generalization. Additionally, we validate that our method effectively transfers to the real world through hardware experiments. Videos and supplementary materials available at https://sites.google.com/view/keypointsintothefuture

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑