Search papers, labs, and topics across Lattice.
This paper introduces a framework for training large language models (LLMs) to develop the "Connect the Dots" (CoD) meta-capability, essential for long-lifecycle agents that must adaptively learn and update their context while solving sequential tasks. The authors implement a GRPO-style reinforcement learning algorithm that enables LLMs to interleave task-solving and context-updating episodes, facilitating effective learning across various domains. Empirical results demonstrate that this approach not only enhances performance on future tasks but also allows for significant out-of-distribution generalization.
LLMs can achieve remarkable out-of-distribution generalization by learning to self-update their context through a novel reinforcement learning framework.
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 https://github.com/agentscope-ai/Trinity-RFT/tree/research/cod/examples/research_cod.
