On Extending Direct Preference Optimization to Accommodate Ties

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Jinghong, Yang, Guangyu, Lin, Weizhe, Mei, Jingbiao, Byrne, Bill
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918185175875584
author Chen, Jinghong
Yang, Guangyu
Lin, Weizhe
Mei, Jingbiao
Byrne, Bill
author_facet Chen, Jinghong
Yang, Guangyu
Lin, Weizhe
Mei, Jingbiao
Byrne, Bill
contents We derive and investigate two DPO variants that explicitly model the possibility of declaring a tie in pair-wise comparisons. We replace the Bradley-Terry model in DPO with two well-known modeling extensions, by Rao and Kupper and by Davidson, that assign probability to ties as alternatives to clear preferences. Our experiments in neural machine translation and summarization show that explicitly labeled ties can be added to the datasets for these DPO variants without the degradation in task performance that is observed when the same tied pairs are presented to DPO. We find empirically that the inclusion of ties leads to stronger regularization with respect to the reference policy as measured by KL divergence, and we see this even for DPO in its original form. We provide a theoretical explanation for this regularization effect using ideal DPO policy theory. We further show performance improvements over DPO in translation and mathematical reasoning using our DPO variants. We find it can be beneficial to include ties in preference optimization rather than simply discard them, as is done in common practice.
format Preprint
id arxiv_https___arxiv_org_abs_2409_17431
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On Extending Direct Preference Optimization to Accommodate Ties
Chen, Jinghong
Yang, Guangyu
Lin, Weizhe
Mei, Jingbiao
Byrne, Bill
Computation and Language
We derive and investigate two DPO variants that explicitly model the possibility of declaring a tie in pair-wise comparisons. We replace the Bradley-Terry model in DPO with two well-known modeling extensions, by Rao and Kupper and by Davidson, that assign probability to ties as alternatives to clear preferences. Our experiments in neural machine translation and summarization show that explicitly labeled ties can be added to the datasets for these DPO variants without the degradation in task performance that is observed when the same tied pairs are presented to DPO. We find empirically that the inclusion of ties leads to stronger regularization with respect to the reference policy as measured by KL divergence, and we see this even for DPO in its original form. We provide a theoretical explanation for this regularization effect using ideal DPO policy theory. We further show performance improvements over DPO in translation and mathematical reasoning using our DPO variants. We find it can be beneficial to include ties in preference optimization rather than simply discard them, as is done in common practice.
title On Extending Direct Preference Optimization to Accommodate Ties
topic Computation and Language
url https://arxiv.org/abs/2409.17431