Search papers, labs, and topics across Lattice.
SuperIgor is introduced, a framework for instruction-following tasks that uses a language model to generate and refine high-level plans without predefined subtasks. A reinforcement learning agent is trained to follow these plans, while the language model adapts the plans based on RL feedback, creating a co-training loop. Experiments in stochastic environments show that SuperIgor agents adhere to instructions more strictly and generalize better to unseen instructions compared to baselines.
Forget meticulously annotating subtasks – SuperIgor lets language models self-learn to generate and refine instruction-following plans through RL feedback.
We introduce SuperIgor, a framework for instruction-following tasks. Unlike prior methods that rely on predefined subtasks, SuperIgor enables a language model to generate and refine high-level plans through a self-learning mechanism, reducing the need for manual dataset annotation. Our approach involves iterative co-training: an RL agent is trained to follow the generated plans, while the language model adapts and modifies these plans based on RL feedback and preferences. This creates a feedback loop where both the agent and the planner improve jointly. We validate our framework in environments with rich dynamics and stochasticity. Results show that SuperIgor agents adhere to instructions more strictly than baseline methods, while also demonstrating strong generalization to previously unseen instructions.
