Best Arm Identification with LLM Judges and Limited Human

Read full text
Authors: Ruicheng Ao, Hongyu Chen, Siyang Gao, Hanwei Li, David Simchi‐Levi
Cited by: 0

Abstract

We study fixed-confidence best-arm identification (BAI) where a cheap but potentially biased proxy (e.g., LLM judge) is available for every sample, while an expensive ground-truth label can only be acquired selectively when using a human for auditing. Unlike classical multi-fidelity BAI, the proxy is biased (arm- and context-dependent) and ground truth is selectively observed. Consequently, standard multi-fidelity methods can mis-select the best arm, and uniform auditing, though accurate, wastes scarce resources and is inefficient. We prove that without bias correction and propensity adjustment, mis-selection probability may not vanish (even with unlimited proxy data). We then develop an estimator for the mean of each arm that combines proxy scores with inverse-propensity-weighted residuals and form anytime-valid confidence sequences for that estimator. Based on the estimator and confidence sequence, we propose an algorithm that adaptively selects and audits arms. The algorithm concentrates audits on unreliable contexts and close arms and we prove that a plug-in Neyman rule achieves near-oracle audit efficiency. Numerical experiments confirm the theoretical guarantees and demonstrate the superior empirical performance of the proposed algorithm.

Fields of Study & Subjects

Advanced Bandit Algorithms Research Simulation Techniques and Applications Reinforcement Learning in Robotics

Publication details

Journal / source
arXiv (Cornell University)
Publication date
29 January 2026
Document type
Preprint