FreeMU: Free Marginal Utility for Reward-Aware Direct Preference Optimization

Lei Zhang, Jian Ma, Xianpeng Zhang, Dongyu Chen, Haonan Lu, Tianhuang Su, Chuangchuang Wang, Kai Tang, Zhixun Su


Abstract
Reward-aware variants of Direct Preference Optimization (DPO), such as ODPO and MODPO, implicitly assume linear marginal utility over the preference signal: identical signal gaps induce identical gradient updates regardless of whether the pair lies in a low- or high-quality region. We propose FreeMU, a lightweight plug-and-play training module that introduces non-linear marginal utility through a pair-aware divided-difference weight applied directly to the preference margin. The same weight formula integrates with three representative DPO variants—ODPO, MODPO, and β-DPO—with each instantiation differing only in the reward source and its normalisation. We first instantiate FreeMU on the two explicit-score variants (Likert labels for ODPO and reward-model outputs for MODPO) and then, for variants without externally annotated scores, on β-DPO using the implicit Bradley–Terry reward induced by the policy and reference models. Across summarization, multi-objective safety alignment, and dialogue, FreeMU consistently improves over its host in LLM-based win rates and task-specific metrics with near-zero training overhead; a controlled ablation confirms that the convex curvature direction is the operative ingredient—reversing the sign to a concave utility removes the gain.
Anthology ID:
2026.findings-emnlp.1871
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
39932–39953
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1871/
DOI:
Bibkey:
Cite (ACL):
Lei Zhang, Jian Ma, Xianpeng Zhang, Dongyu Chen, Haonan Lu, Tianhuang Su, Chuangchuang Wang, Kai Tang, and Zhixun Su. 2026. FreeMU: Free Marginal Utility for Reward-Aware Direct Preference Optimization. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 39932–39953, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
FreeMU: Free Marginal Utility for Reward-Aware Direct Preference Optimization (Zhang et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1871.pdf