COMAL: A Convergent Meta-Algorithm for Aligning LLMs with General Preferences

Published: 10 Oct 2024, Last Modified: 01 Nov 2024FITML 2024 OralEveryoneRevisionsBibTeXCC BY 4.0
Keywords: Alignment, General Preferences, Large Language Model, Nash Equilibrium
TL;DR: We propose a simple, integrable meta-algorithm inspired by game theory that provably converges to an exact Nash policy, effectively aligning language models with general human preferences beyond the limitations of existing methods.
Abstract: Many alignment methods, including reinforcement learning from human feedback (RLHF), rely on the Bradley-Terry reward assumption, which is insufficient to capture the full range of general human preferences. To achieve robust alignment with general preferences, we model the alignment problem as a two-player zero-sum game, where the Nash equilibrium policy guarantees a 50\% win rate against any competing policy. However, previous algorithms for finding the Nash policy either diverge or converge to a Nash policy in a modified game, even in a simple synthetic setting, thereby failing to maintain the 50\% win rate guarantee against all other policies. We propose a meta-algorithm, **Co**nvergent **M**eta **Al**ignment Algorithm (COMAL), for language model alignment with general preferences, inspired by convergent algorithms in game theory. Theoretically, we prove that our meta-algorithm converges to an exact Nash policy in the last iterate. Additionally, our meta-algorithm is simple and can be integrated with many existing methods designed for RLHF and preference optimization with minimal changes. Experimental results demonstrate the effectiveness of the proposed framework when combined with existing preference policy optimization methods.
Submission Number: 106
Loading