Skip to yearly menu bar Skip to main content


Poster

Self-Improving Robust Preference Optimization

Eugene Choi · Arash Ahmadian · Matthieu Geist · Olivier Pietquin · Mohammad Gheshlaghi Azar

Hall 3 + Hall 2B #256
[ ]
Fri 25 Apr midnight PDT — 2:30 a.m. PDT

Abstract: Online and offline RLHFRLHF methods, such as PPOPPO and DPODPO, have been highly successful in aligning AI with human preferences. Despite their success, however, these methods suffer from fundamental limitations: (a)(a) Models trained with RLHFRLHF can learn from mistakes or negative examples through RL mechanism or contrastive loss during training. However, at inference time, they lack an innate self-improvement mechanism for error corrections. (b)(b) The optimal solution of existing methods is highly task-dependent, making it difficult for them to generalize to new tasks. To address these challenges, we propose Self-Improving Robust Preference Optimization (SRPOSRPO), a practical and mathematically principled offline RLHFRLHF framework. The key idea behind SRPOSRPO is to cast the problem of learning from human preferences as a self-improvement process, mathematically formulated as a min-max objective that jointly optimizes a self-improvement policy and a generative policy in an adversarial fashion. Crucially, the solution for this optimization problem is independent of the training task, which makes it robust to its changes. We then show that this objective can be reformulated as a non-adversarial offline loss, which can be efficiently optimized using standard supervised learning techniques at scale. To demonstrate SRPOSRPO’s effectiveness, we evaluate it using AI Win-Rate (WR) against human (GOLD) completions. When tested on the XSum dataset, SRPOSRPO outperforms DPODPO by a margin of 1515% after 55 self-revisions, achieving an impressive 9090% WR. Moreover, on the challenging Arena-Hard prompts, SRPOSRPO outperforms both DPODPO and IPOIPO (by 44% without revision and 66% after a single revision), reaching a 5656% WR against against Llama3.18BInstructLlama3.18BInstruct.

Live content is unavailable. Log in and register to view live content