English

LLM-as-Judge on a Budget

Machine Learning 2026-04-14 v2

Abstract

LLM-as-a-judge has emerged as a cornerstone technique for evaluating large language models by leveraging LLM reasoning to score prompt-response pairs. Since LLM judgments are stochastic, practitioners commonly query each pair multiple times to estimate mean scores accurately. This raises a critical challenge: given a fixed computational budget BB, how to optimally allocate queries across KK prompt-response pairs to minimize estimation error? We present a principled variance-adaptive approach leveraging multi-armed bandit theory and concentration inequalities. Our method dynamically allocates queries based on estimated score variances, concentrating resources where uncertainty is highest. Further, our algorithm is shown to achieve a worst-case score-estimation error of O~(i=1Kσi2B)\tilde{O}\left(\sqrt{\frac{\sum_{i=1}^K \sigma_i^2}{B}}\right), σi2\sigma_i^2 being the unknown score variance for pair i[K]i \in [K] with near-optimal budget allocation. Experiments on \emph{Summarize-From-Feedback} and \emph{HelpSteer2} demonstrate that our method significantly outperforms uniform allocation, reducing worst-case estimation error while maintaining identical budgets. Our work establishes a theoretical foundation for efficient LLM evaluation with practical implications for AI safety, model alignment, and automated assessment at scale.

Keywords

Cite

@article{arxiv.2602.15481,
  title  = {LLM-as-Judge on a Budget},
  author = {Aadirupa Saha and Aniket Wagde and Branislav Kveton},
  journal= {arXiv preprint arXiv:2602.15481},
  year   = {2026}
}
R2 v1 2026-07-01T10:39:46.704Z