用大模型当虚拟老师,让机器人学操作更省人力且效果更好
LLM-based Interactive Imitation Learning for Robotic Manipulation
- 用分层提示让大模型生成可执行的控制代码
- 训练中通过相似性反馈机制实时纠错,成功率超传统方法
- 无需真人教师,适合想低成本训练机器人的研究者
近期机器学习进展使得自主智能体能够应对机器人任务中日益复杂的序列决策问题。模仿学习(IL)通过人类示范训练机器人控制,但常违反独立同分布假设。交互式模仿学习(IIL)通过人类教师的实时反馈提升性能,但依赖人力成本高。本文提出 LLM-iTeach——一种基于大语言模型(LLM)的新型交互式模仿学习框架,以大模型作为虚拟教师,减少对真人教师的依赖。该框架采用分层提示策略,引导大模型生成 Python 代码形式的控制策略,并设计基于相似性的反馈机制,在训练过程中提供纠正与评估。在多个机器人抓取任务上,相比行为克隆(BC)和使用真人教师的先进 IIL 方法 CEILing,LLM-iTeach 在成功率上表现更优或相当,证明了大模型作为高效、类人教师的潜力。实验还验证了其在新任务上的泛化能力。代码与提示已开源:https://github.com/Tubicor/LLM-iTeach。
原文摘要 · Abstract (English)
Recent advancements in machine learning provide methods to train autonomous agents capable of handling the increasing complexity of sequential decision-making in robotics. Imitation Learning (IL) is a prominent approach, where agents learn to control robots based on human demonstrations. However, IL commonly suffers from violating the independent and identically distributed (i.i.d) assumption in robotic tasks. Interactive Imitation Learning (IIL) achieves improved performance by allowing agents to learn from interactive feedback from human teachers. Despite these improvements, both approaches come with significant costs due to the necessity of human involvement. Leveraging the emergent capabilities of Large Language Models (LLMs) in reasoning and generating human-like responses, we introduce LLM-iTeach -- a novel IIL framework that utilizes an LLM as an interactive teacher to enhance agent performance while alleviating the dependence on human resources. Firstly, LLM-iTeach uses a hierarchical prompting strategy that guides the LLM in generating a policy in Python code. Then, with a designed similarity-based feedback mechanism, LLM-iTeach provides corrective and evaluative feedback interactively during the agent's training. We evaluate LLM-iTeach against baseline methods such as Behavior Cloning (BC), an IL method, and CEILing, a state-of-the-art IIL method using a human teacher, on various robotic manipulation tasks. Our results demonstrate that LLM-iTeach surpasses BC in the success rate and achieves or even outscores that of CEILing, highlighting the potential of LLMs as cost-effective, human-like teachers in interactive learning environments. We further demonstrate the method's potential for generalization by evaluating it on additional tasks. The code and prompts are provided at: https://github.com/Tubicor/LLM-iTeach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。