arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.16784cs.ROcs.LG

基于世界模型的视觉足式移动感知

World Model-based Perception for Visual Legged Locomotion

Shanghai Jiao Tong University(上海交通大学);ByteDance Research(字节跳动研究院)

Hang Lai, Jiahang Cao, Jiafeng Xu, Hongtao Wu, Yunfeng Lin, Tao Kong, Yong Yu, Weinan Zhang

AI总结:

针对视觉足式移动中传统模仿学习的性能受限问题,提出基于世界模型的感知方法(WMP),通过构建环境世界模型学习策略,在仿真和真实实验中显著提升了可通行性与鲁棒性。

摘要:

在各种地形上的足式移动具有挑战性,需要通过本体感觉和视觉精确感知机器人及其周围环境。然而,直接从高维视觉输入中学习通常数据效率低下且复杂。为解决此问题,传统方法尝试先学习具有特权信息访问权限的教师策略,然后学习学生策略以用视觉输入模仿教师行为。尽管取得了一些进展,但由于输入间的信息差距,这种模仿框架阻碍了学生策略实现最佳性能。此外,学习过程不自然,因为动物直观地基于对世界的理解来穿越不同地形,无需特权知识。受此自然能力启发,我们提出了一种简单有效的方法,基于世界模型的感知(WMP),该方法构建环境的世界模型并基于该世界模型学习策略。我们说明,尽管完全在仿真中训练,世界模型能对真实世界轨迹做出准确预测,从而为策略控制器提供信息信号。大量仿真和真实世界实验证明,WMP在可通行性和鲁棒性上优于最先进的基线。

英文摘要:

Legged locomotion over various terrains is challenging and requires precise perception of the robot and its surroundings from both proprioception and vision. However, learning directly from high-dimensional visual input is often data-inefficient and intricate. To address this issue, traditional methods attempt to learn a teacher policy with access to privileged information first and then learn a student policy to imitate the teacher's behavior with visual input. Despite some progress, this imitation framework prevents the student policy from achieving optimal performance due to the information gap between inputs. Furthermore, the learning process is unnatural since animals intuitively learn to traverse different terrains based on their understanding of the world without privileged knowledge. Inspired by this natural ability, we propose a simple yet effective method, World Model-based Perception (WMP), which builds a world model of the environment and learns a policy based on the world model. We illustrate that though completely trained in simulation, the world model can make accurate predictions of real-world trajectories, thus providing informative signals for the policy controller. Extensive simulated and real-world experiments demonstrate that WMP outperforms state-of-the-art baselines in traversability and robustness. Videos and Code are available at: https://wmp-loco.github.io/.

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

↑