Agentic-DPO: From Imitation to Agentic Policy Optimization on Expert Trajectories
Summary
Agentic-DPO is a novel, lightweight offline agent policy optimization method designed to enhance Large Language Model (LLM) agents beyond traditional supervised fine-tuning (SFT). While SFT merely imitates expert actions from trajectories, Agentic-DPO addresses the limitation of not learning to select optimal actions against potential errors. It transforms expert trajectories into state-conditioned preference supervision, sampling one-step actions at each expert state and contrasting plausible wrong actions with the expert action using a DPO-style objective. The method incorporates Policy-Preserving Augmentation (PPA) to maintain a fixed expert policy across multiple schemas, preventing policy-schema mixing. Agentic-DPO eliminates the need for online environment rollouts, reward models, or full-trajectory student exploration. Experiments across StableToolBench, tau-bench retail, and Mind2Web demonstrate consistent improvements, notably boosting tau-bench accuracy from 21.7% (SFT) to 41.4% for a 9B model, matching online GRPO performance without environment interaction during gradient steps. The code is available at https://github.com/Schuture/Agentic-DPO.
Key takeaway
For Machine Learning Engineers developing LLM agents, especially when online environment rollouts or reward model training are cost-prohibitive, you should consider Agentic-DPO. This method allows you to significantly improve agent performance and robustness beyond simple imitation learning by converting existing expert trajectories into state-conditioned preference supervision. Integrating Agentic-DPO can boost your agent's accuracy, as demonstrated by a 9B model's tau-bench accuracy rising from 21.7% to 41.4%, without incurring high online interaction costs.
Key insights
Agentic-DPO converts expert trajectories into state-level action preferences for efficient, offline LLM agent policy optimization, surpassing imitation learning.
Principles
- Offline preference learning improves agent robustness.
- State-conditioned preferences enhance action selection.
- Decouple policy and schema for stable learning.
Method
Agentic-DPO samples one-step actions at expert states, contrasting plausible wrong actions with expert actions using a DPO-style objective, aided by Policy-Preserving Augmentation.
In practice
- Apply DPO-style objectives to expert trajectories.
- Use Policy-Preserving Augmentation for schema invariance.
- Optimize LLM agents without costly online rollouts.
Topics
- Agentic-DPO
- LLM Agents
- Policy Optimization
- Preference Learning
- Offline RL
- Policy-Preserving Augmentation
Code references
Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.