Not All Preference Pairs Are Created Equal: A Recipe for Annotation-Efficient Iterative Preference Learning

S Yang, L Cui, D Cai, X Huang, S Shi…�- arXiv preprint arXiv�…, 2024 - arxiv.org
arXiv preprint arXiv:2406.17312, 2024arxiv.org
Iterative preference learning, though yielding superior performances, requires online
annotated preference labels. In this work, we study strategies to select worth-annotating
response pairs for cost-efficient annotation while achieving competitive or even better
performances compared with the random selection baseline for iterative preference
learning. Built on assumptions regarding uncertainty and distribution shifts, we propose a
comparative view to rank the implicit reward margins as predicted by DPO to select the�…
Iterative preference learning, though yielding superior performances, requires online annotated preference labels. In this work, we study strategies to select worth-annotating response pairs for cost-efficient annotation while achieving competitive or even better performances compared with the random selection baseline for iterative preference learning. Built on assumptions regarding uncertainty and distribution shifts, we propose a comparative view to rank the implicit reward margins as predicted by DPO to select the response pairs that yield more benefits. Through extensive experiments, we show that annotating those response pairs with small margins is generally better than large or random, under both single- and multi-iteration scenarios. Besides, our empirical results suggest allocating more annotation budgets in the earlier iterations rather than later across multiple iterations.
arxiv.org