LGAIGTMLJun 24, 2024

Bandits with Preference Feedback: A Stackelberg Game Perspective

arXiv:2406.16745v39 citations
AI Analysis

This work addresses a key bottleneck in incorporating human feedback into systems like large language model fine-tuning, representing a significant but incremental advance over prior linear or finite-domain methods.

The paper tackles the challenge of optimizing unknown nonlinear functions using only pairwise preference feedback over infinite domains, proposing the MAXMINLCB algorithm that outperforms existing methods with an anytime-valid rate-optimal regret guarantee.

Bandits with preference feedback present a powerful tool for optimizing unknown target functions when only pairwise comparisons are allowed instead of direct value queries. This model allows for incorporating human feedback into online inference and optimization and has been employed in systems for fine-tuning large language models. The problem is well understood in simplified settings with linear target functions or over finite small domains that limit practical interest. Taking the next step, we consider infinite domains and nonlinear (kernelized) rewards. In this setting, selecting a pair of actions is quite challenging and requires balancing exploration and exploitation at two levels: within the pair, and along the iterations of the algorithm. We propose MAXMINLCB, which emulates this trade-off as a zero-sum Stackelberg game, and chooses action pairs that are informative and yield favorable rewards. MAXMINLCB consistently outperforms existing algorithms and satisfies an anytime-valid rate-optimal regret guarantee. This is due to our novel preference-based confidence sequences for kernelized logistic estimators.

Code Implementations1 repo
Foundations

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

Your Notes