让大模型学会持续学习与自我更新,提升长期任务表现。
Connect the Dots: Training LLMs for Long-Lifecycle Agents with Cross-Domain Generalization Via Reinforcement Learning
- 设计端到端强化学习框架,交替执行任务求解与环境认知更新。
- 实验证明模型在训练域内及跨域均具备泛化能力,且可迁移到新场景。
- 适合研究长期自主智能体的学者,推动大模型持续进化能力发展。
本文提出一种通用框架,用于训练大语言模型(LLMs)具备「连点成线」(CoD)这一元能力:当基于LLM的智能体部署于环境中时,它能连续完成一系列任务,持续探索、从自身经验中学习,并迭代更新对环境的认知,从而在后续任务中表现更优。该框架包含两大核心:(1) 支持长序列回放的端到端强化学习(RL)算法与基础设施,实现任务求解与上下文更新的交替;(2) 专门设计的任务与环境,用于在训练中激励并激发目标元能力,并在评估中准确衡量进展。我们实现了基于GRPO风格的强化学习算法,采用细粒度信用分配机制,并构建了面向目标能力而非特定领域或任务的训练环境。实验验证了该框架的有效性,展示了所习得元能力在训练域内、跨域以及从CoD设置迁移至Ralph-loop设置中的潜在泛化能力。本研究连接了多个先前工作方向,为推进大模型与智能体的发展开辟新路径。为促进进一步研究与应用,代码已开源。
原文摘要 · Abstract (English)
This work presents a general framework for training large language models (LLMs) to "Connect the Dots" (CoD), a meta-capability required by long-lifecycle agents: as an LLM-based AI agent gets deployed in an environment, it solves a long sequence of tasks while continuously exploring the environment, learning from its own experiences, and iteratively self-updating its context about the environment, thereby achieving progressively better performance on future tasks conditioned on the updated context. Major components of the CoD framework include: (1) algorithm design and infrastructure for end-to-end reinforcement learning (RL) with long rollout sequences interleaving solve-task and update-context episodes; (2) tasks and environments for incentivizing and eliciting the targeted meta-capability in LLMs during training, as well as for faithfully measuring progress during evaluation. We present proof-of-concept implementations of the CoD framework, including a GRPO-style RL algorithm with fine-grained credit assignment, as well as tasks and environments tailored to the targeted meta-capability (rather than domain-specific LLM capabilities or standard task-by-task RL). Empirical results validate the efficacy of end-to-end RL training in the CoD setting, and demonstrate the potential for out-of-distribution generalization -- within the training domains, across different domains, and from CoD to Ralph-loop settings -- of the elicited meta-capability. Our investigation of CoD connects several lines of prior works, and opens up new opportunities for advancing LLMs and AI agents. To facilitate further research and applications, we release our implementations at \url{https://github.com/agentscope-ai/Trinity-RFT/tree/research/cod/examples/research_cod}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。