dwDPO: Training Multi-Turn LLM Agents via Divergence-Weighted Direct Preference Optimization

Jinghao Lin, Yuhang Wu, Yuanqing Wang, Yuchen Li, Tianxingjian Kang, Baris Coskunuzer, Xiawu Zheng


Abstract
Large language models deployed as multi-step agents are increasingly trained via Direct Preference Optimization (DPO) on trajectory-level comparisons. Yet standard DPO weights every step uniformly, penalizing correct early actions in a losing trajectory as severely as the causal error itself. We propose Divergence-Weighted DPO(dwDPO), a drop-in replacement that detects the first divergence step between competing trajectories using implicit rewards already computed by DPO, then concentrates the loss around it via an asymmetric kernel with a closed-form scale factor that preserves the Bradley-Terry preference model. dwDPO recovers standard DPO as a special case and provably achieves higher gradient concentration at the critical step than both uniform and monotonic-discount baselines. Extensive experimental results across six deep-research benchmarks demonstrate the superiority of dwDPO over state-of-the-art open-source baselines of comparable scale at no additional inference cost.
Anthology ID:
2026.emnlp-main.2406
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
54657–54674
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2406/
DOI:
Bibkey:
Cite (ACL):
Jinghao Lin, Yuhang Wu, Yuanqing Wang, Yuchen Li, Tianxingjian Kang, Baris Coskunuzer, and Xiawu Zheng. 2026. dwDPO: Training Multi-Turn LLM Agents via Divergence-Weighted Direct Preference Optimization. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 54657–54674, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
dwDPO: Training Multi-Turn LLM Agents via Divergence-Weighted Direct Preference Optimization (Lin et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2406.pdf