Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Calibrating LLM Judges: Linear Probes for Fast and Reliable Uncertainty Estimation

About

As LLM-based judges become integral to industry applications, obtaining well-calibrated uncertainty estimates efficiently has become critical for production deployment. However, existing techniques, such as verbalized confidence and multi-generation methods, are often either poorly calibrated or computationally expensive. We introduce linear probes trained with a Brier score-based loss to provide calibrated uncertainty estimates from reasoning judges' hidden states, requiring no additional model training. We evaluate our approach on both objective tasks (reasoning, mathematics, factuality, coding) and subjective human preference judgments. Our results demonstrate that probes achieve superior calibration compared to existing methods with $\approx10$x computational savings, generalize robustly to unseen evaluation domains, and deliver higher accuracy on high-confidence predictions. However, probes produce conservative estimates that underperform on easier datasets but may benefit safety-critical deployments prioritizing low false-positive rates. Overall, our work demonstrates that interpretability-based uncertainty estimation provides a practical and scalable plug-and-play solution for LLM judges in production.

Bhaktipriya Radharapu, Eshika Saxena, Kenneth Li, Chenxi Whitehouse, Adina Williams, Nicola Cancedda• 2025

Related benchmarks

TaskDatasetResultRank
Uncertainty EstimationJudgeBench (test)
AUROC71.17
77
Correctness CalibrationPPE (Preference Policy Evaluation)
Kuiper0.017
24
Preference CalibrationPPE
Kuiper0.034
24
Uncertainty CalibrationJudgeBench
Kuiper0.037
24
Uncertainty CalibrationRewardBench
Kuiper0.065
24
Correctness EvaluationPPE Correctness (test)
Kuiper0.0643
8
Model EvaluationJudgeBench (test)
Kuiper5.63
8
Preference EvaluationPPE Preference (test)
Kuiper Statistic0.0434
8
Reward Model EvaluationRewardBench (test)
Kuiper6.83
8
Showing 9 of 9 rows

Other info

Follow for update