arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.14872cs.CVcs.RO

Adapt2Reward:通过失败提示将视频-语言模型适配为可泛化的机器人奖励函数

Adapt2Reward: Adapting Video-Language Models to Generalizable Robotic Rewards via Failure Prompts

School of Software Technology, Zhejiang University(浙江大学软件学院);School of Computer Sciene and Technology, Hangzhou Dianzi University(杭州电子科技大学计算机学院);China Mobile (Zhejiang) Research & Innovation Institute(中国移动(浙江)研究院);State Key Lab of CAD&CG, Zhejiang University(浙江大学CAD&CG国家重点实验室);The Hong Kong Polytechnic University(香港理工大学);Zhiyuan Research Institute(智源研究院);School of Computer Sciene and Technology, Xidian University(西安电子科技大学计算机学院)

Yanting Yang, Minghao Chen, Qibo Qiu, Jiahao Wu, Wenxiao Wang, Binbin Lin, Ziyu Guan, Xiaofei He

AI总结:

本文提出Adapt2Reward方法,利用少量单一环境机器人视频数据,通过聚类失败视频特征并集成失败提示,将视频-语言模型转化为可泛化的语言条件奖励函数,在新环境和新指令上表现卓越。

摘要:

为了让通用机器人在现实中运行,必须在各种环境中执行广泛的指令。对于此类机器人智能体的强化学习和规划而言,核心是可泛化的奖励函数。视觉-语言模型(如CLIP)的最新进展在深度学习领域展现了卓越的性能,为开放域视觉识别铺平了道路。然而,收集机器人在多种环境中执行各种语言指令的数据仍然是一个挑战。本文旨在将具有强大泛化能力的视频-语言模型转化为可泛化的语言条件奖励函数,仅利用单一环境中少量任务的机器人视频数据。与用于训练奖励函数的常见机器人数据集不同,人类视频-语言数据集很少包含琐碎的失败视频。为了增强模型区分成功与失败机器人执行的能力,我们对失败视频特征进行聚类,使模型能够识别其中的模式。对于每个聚类,我们在文本编码器中集成一个新训练的失败提示,以表示相应的失败模式。我们的语言条件奖励函数在机器人规划和强化学习中对新环境和新指令展现出卓越的泛化能力。

英文摘要:

For a general-purpose robot to operate in reality, executing a broad range of instructions across various environments is imperative. Central to the reinforcement learning and planning for such robotic agents is a generalizable reward function. Recent advances in vision-language models, such as CLIP, have shown remarkable performance in the domain of deep learning, paving the way for open-domain visual recognition. However, collecting data on robots executing various language instructions across multiple environments remains a challenge. This paper aims to transfer video-language models with robust generalization into a generalizable language-conditioned reward function, only utilizing robot video data from a minimal amount of tasks in a singular environment. Unlike common robotic datasets used for training reward functions, human video-language datasets rarely contain trivial failure videos. To enhance the model's ability to distinguish between successful and failed robot executions, we cluster failure video features to enable the model to identify patterns within. For each cluster, we integrate a newly trained failure prompt into the text encoder to represent the corresponding failure mode. Our language-conditioned reward function shows outstanding generalization to new environments and new instructions for robot planning and reinforcement learning.

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

↑