让语言模型学会主动探索,跨任务自适应决策。
Training a Generally Curious Agent
- 用多任务合成数据微调,提升模型泛化探索能力。
- 在新任务上仅靠环境反馈即可自适应,无需再训练。
- 适合需要自主交互与策略生成的智能系统研究者。
高效探索对智能体与环境互动至关重要,但现有语言模型在需战略性信息收集的场景中表现不足。本文提出Paprika,一种微调方法,使语言模型具备不局限于特定环境的通用决策能力。通过在需多样化策略的不同任务的合成交互数据上训练,Paprika使模型能基于环境反馈,在上下文中调整行为,无需额外梯度更新。实验表明,经Paprika微调的模型可有效将学习到的决策能力迁移至完全未见过的任务。与传统训练不同,本方法的主要瓶颈在于采样高质量交互数据而非模型更新。为提升样本效率,我们提出课程学习策略,优先采样高学习潜力任务的轨迹。结果表明,这为实现可自主解决新型序列决策问题的AI系统提供了可行路径。
原文摘要 · Abstract (English)
Efficient exploration is essential for intelligent systems interacting with their environment, but existing language models often fall short in scenarios that require strategic information gathering. In this paper, we present Paprika, a fine-tuning approach that enables language models to develop general decision-making capabilities that are not confined to particular environments. By training on synthetic interaction data from different tasks that require diverse strategies, Paprika teaches models to explore and adapt their behavior on a new task based on environment feedback in-context without more gradient updates. Experimental results show that models fine-tuned with Paprika can effectively transfer their learned decision-making capabilities to entirely unseen tasks without additional training. Unlike traditional training, our approach's primary bottleneck lies in sampling useful interaction data instead of model updates. To improve sample efficiency, we propose a curriculum learning strategy that prioritizes sampling trajectories from tasks with high learning potential. These results suggest a promising path towards AI systems that can autonomously solve novel sequential decision-making problems that require interactions with the external world.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。