Agentic-DPO: From Imitation to Agentic Policy Optimization on Expert Trajectories

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, quick

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

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

Topics

Code references

Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.