让AI通过提示内化多任务技能,摆脱依赖长提示的困境
Memento No More: Coaching AI Agents to Master Multiple Tasks via Hints Internalization
- 用迭代反馈+上下文蒸馏,将人类提示转化为模型内部知识
- 仅需几轮训练,就超越GPT-4o和DeepSeek-V3在复杂任务序列上的表现
- 适合需要持续学习多任务的AI系统开发者使用
随着人工智能代理通用能力的发展,如何通过经验掌握多个复杂任务仍是关键挑战。当前基于专有语言模型的LLM代理通常依赖提示来获取目标任务知识,无法真正内化信息,只能依靠不断扩大的提示维持功能,如同顺行性遗忘症患者依赖外部笔记。本文提出一种新方法,使AI代理在无需冗长笔记或高质量示范数据的情况下,掌握多任务能力。该方法采用迭代流程:代理收集新经验,接收人类以提示形式提供的纠正反馈,并通过上下文蒸馏训练将其融入模型权重。我们在基于Llama-3的代理上实现该方法,仅经数轮反馈后,在需正确排序信息检索、工具使用与问答的任务集上,表现优于GPT-4o和DeepSeek-V3。
原文摘要 · Abstract (English)
As the general capabilities of artificial intelligence (AI) agents continue to evolve, their ability to learn to master multiple complex tasks through experience remains a key challenge. Current LLM agents, particularly those based on proprietary language models, typically rely on prompts to incorporate knowledge about the target tasks. This approach does not allow the agent to internalize this information and instead relies on ever-expanding prompts to sustain its functionality in diverse scenarios. This resembles a system of notes used by a person affected by anterograde amnesia, the inability to form new memories. In this paper, we propose a novel method to train AI agents to incorporate knowledge and skills for multiple tasks without the need for either cumbersome note systems or prior high-quality demonstration data. Our approach employs an iterative process where the agent collects new experiences, receives corrective feedback from humans in the form of hints, and integrates this feedback into its weights via a context distillation training procedure. We demonstrate the efficacy of our approach by implementing it in a Llama-3-based agent that, after only a few rounds of feedback, outperforms advanced models GPT-4o and DeepSeek-V3 in tasksets requiring correct sequencing of information retrieval, tool use, and question answering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。