LGAIMLNov 10, 2024

Meta-Learning Objectives for Preference Optimization

arXiv:2411.06568v34 citationsh-index: 14
Originality Highly original
AI Analysis

This work addresses the problem of costly and noisy evaluation in preference optimization for researchers, offering a more controlled benchmark and specialized algorithms, though it is incremental in applying insights from simpler tasks to LLMs.

The paper tackles the challenge of evaluating preference optimization algorithms for LLM alignment by creating a cheaper diagnostic benchmark using MuJoCo tasks and proposes Mirror Preference Optimization (MPO), a novel family of algorithms discovered via evolutionary strategies that outperform existing methods in targeted settings and improve LLM alignment performance.

Evaluating preference optimization (PO) algorithms on LLM alignment is a challenging task that presents prohibitive costs, noise, and several variables like model size and hyper-parameters. In this work, we show that it is possible to gain insights on the efficacy of PO algorithm on simpler benchmarks. We design a diagnostic suite of MuJoCo tasks and datasets, which we use to systematically evaluate PO algorithms, establishing a more controlled and cheaper benchmark. We then propose a novel family of PO algorithms based on mirror descent, which we call Mirror Preference Optimization (MPO). Through evolutionary strategies, we search this class to discover algorithms specialized to specific properties of preference datasets, such as mixed-quality or noisy data. We demonstrate that our discovered PO algorithms outperform all known algorithms in the targeted MuJoCo settings. Finally, based on the insights gained from our MuJoCo experiments, we design a PO algorithm that significantly outperform existing baselines in an LLM alignment task.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes