Towards Disentangled Preference Optimization Dynamics: Suppress the Loser, Preserve the Winner
Abstract
Preference optimization is widely used to align large language models (LLMs) with human preferences. However, many margin-based methods also suppress the chosen response when they try to suppress the rejected one, and there is no general way to prevent this across different objectives. We address this issue with a unified incentive-score decomposition of preference optimization, revealing that different objectives share the same local update directions and differ only in their scalar weights. This decomposition provides a common framework for analyzing objectives that were previously studied in separate settings. Building on this decomposition, by analyzing the dynamics of the chosen/rejected likelihoods, we identify the disentanglement band (DB), a simple, testable condition that tells us when training can follow the desired path: suppress the loser while preserving the winner, possibly after an early stage. Using the DB, we propose reward calibration (RC), a plug‑and‑play method that adaptively rebalances the updates for chosen and rejected likelihoods to satisfy the DB, without redesigning the base objective. Empirical results show that RC leads to more disentangled dynamics, with better downstream performance observed across several settings. Our code is available at https://github.com/IceyWuu/DisentangledPreferenceOptimization.
Lay Summary
When training large language models to prefer good responses over bad ones, many methods accidentally hurt the good response while trying to suppress the bad one. This paper explains why this happens and introduces a simple plug‑in fix that rebalances the learning process. The proposed method helps models suppress the bad response without degrading the good one, leading to cleaner training and better performance across different tasks and model sizes.