Dependence-Aware Label Aggregation for LLM-as-a-Judge via Ising Models
Krishnakumar Balasubramanian, Aleksandr Podkopaev, Shiva Prasad Kasiviswanathan
TL;DR
This work addresses the gap between conditional-independence label aggregation and the dense dependencies found among LLM-based judges. It develops a hierarchy of dependence-aware models based on Ising graphical models, deriving Bayes posteriors that are quadratic in votes for class-dependent couplings and linear when couplings are shared, and shows that CI-based methods can be strictly suboptimal under dependence. The authors prove separation results—CI predictors can incur nonvanishing excess risk while the Bayes predictor exploiting dependence can achieve vanishing risk as the judge pool grows—and connect these ideas to latent-factor models. Empirically, dependence-aware Ising predictors outperform classical baselines on relevance, toxicity, and summarization tasks across six judge models, with further improvements as training data increases. A practical EM-based inference framework and diagnostic workflow are provided to enable deployment of dependence-aware label aggregation in real-world evaluation pipelines.
Abstract
Large-scale AI evaluation increasingly relies on aggregating binary judgments from $K$ annotators, including LLMs used as judges. Most classical methods, e.g., Dawid-Skene or (weighted) majority voting, assume annotators are conditionally independent given the true label $Y\in\{0,1\}$, an assumption often violated by LLM judges due to shared data, architectures, prompts, and failure modes. Ignoring such dependencies can yield miscalibrated posteriors and even confidently incorrect predictions. We study label aggregation through a hierarchy of dependence-aware models based on Ising graphical models and latent factors. For class-dependent Ising models, the Bayes log-odds is generally quadratic in votes; for class-independent couplings, it reduces to a linear weighted vote with correlation-adjusted parameters. We present finite-$K$ examples showing that methods based on conditional independence can flip the Bayes label despite matching per-annotator marginals. We prove separation results demonstrating that these methods remain strictly suboptimal as the number of judges grows, incurring nonvanishing excess risk under latent factors. Finally, we evaluate the proposed method on three real-world datasets, demonstrating improved performance over the classical baselines.
