Bandits with Preference Feedback: A Stackelberg Game Perspective

Barna Pásztor, Parnian Kassraie, Andreas Krause
6/24/2024
cs.LGcs.AIcs.GT

No official code found. Best implementation: lasgroup/MaxMinLCB (4 stars, Apache-2.0, last commit 2024-12-17).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

lasgroup/MaxMinLCBCommunity100%
Apache-2.0

Code for our paper "Bandits with Preference Feedback: A Stackelberg Game Perspective"

41Python, Jupyter NotebookJun 17, 20241 years agoApache-2.0
banditsfine-tuningpreference-learning

Similar Papers

Cite this paper

@article{psztor2024bandits,
  title  = {Bandits with Preference Feedback: A Stackelberg Game Perspective},
  author = {Barna Pásztor and Parnian Kassraie and Andreas Krause},
  year   = {2024},
  eprint = {2406.16745},
  archivePrefix = {arXiv},
  doi    = {10.52202/079017-0383},
  url    = {http://arxiv.org/abs/2406.16745v3},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.16745.svg)](https://codeofpaper.com/paper/2406.16745)

Discussion