Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training

Junwon Ko, Dong-Jae Lee, Minchan Kwon, Sunghyun Baek, Junmo Kim


Abstract
LLM agents for sequential decision tasks are often post-trained with trajectory-level outcome labels, but such labels provide little supervision for preserving multiple successful branches from the same decision state. We study this problem as successful strategy coverage: how broadly a model realizes distinct successful strategies under a fixed rollout budget. We present Direct Diversity Optimization (DDO), an offline post-training method that combines Divergence-Tree Collection (DTC) with the Reference-Relative Target-Odds Objective (RTO). DTC constructs state-aligned branch sets rooted at shared decision states, and RTO trains the model to match reference-relative targets over successful alternatives. DDO achieves the strongest task success and successful strategy coverage among the compared post-training methods across BabyAI, BabaIsAI, and WebShop. It also achieves the highest recovery rate after local action replacement and higher task success and coverage than successful-only imitation and decoding-time diversification controls.
Anthology ID:
2026.emnlp-main.1738
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:
39645–39663
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1738/
DOI:
Bibkey:
Cite (ACL):
Junwon Ko, Dong-Jae Lee, Minchan Kwon, Sunghyun Baek, and Junmo Kim. 2026. Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 39645–39663, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training (Ko et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1738.pdf