Are You Sure Your LLM Is Sure? Repairing and Auditing Confidence
Organizations: MBZUAI · TCS Research Labs, India · IIT Delhi
Abstract
Confidence scores determine whether an LLM answer is accepted, reviewed, or regenerated, yet scalar recalibration can only correct errors visible through the score itself. We introduce RADAR (Repair, Adopt, and Diagnose After Recalibration), a reliability framework that separates confidence repair from deciding whether to deploy the repair and auditing the errors that remain. Under squared loss, we decompose prediction risk into outcome noise, information hidden by the score, and mapping error; restricting repair to order-preserving maps additionally introduces a pooling gap. For repair, we propose Probe-Anchored Isotonic Calibration (PAIC), which learns a monotone map from correctness-probe predictions on unlabeled target prompts. PAIC's excess risk depends on the probe's score-conditional bias and fitting error rather than its overall accuracy. For adoption, we derive a paired procedure with finite-sample control of false improvements. For auditing, we develop a prompt-clustered kernel test that detects feature-aligned residual errors while accounting for dependent answers from the same prompt. Across multiple LLM checkpoints, benchmarks, and distribution shifts, richer predictors can improve confidence estimation, but projecting their information back to a scalar score can lose these gains, and lower prediction risk need not yield better review decisions. Across 47,400 replayed raw-score allocations, paired betting adopts candidates with lower held-out test risk in 42.95% of allocations, versus 15.09% for one-sided empirical Bernstein. RADAR thus provides a principled workflow to repair what a scalar score can express, adopt a repair only when supported by evidence, and diagnose systematic errors that recalibration leaves behind. Code and data will be released upon acceptance.
Figures & tables
| Correction | Supervision | Serving inputs |
|---|---|---|
| Source scalar maps | Source labels fit the map | |
| Importance-weighted isotonic | Source labels, reweighted toward unlabeled target prompts | |
| Source-distilled isotonic | from source labels; isotonic fit to on source prompts | |
| Target-probe Platt, beta | from source labels; parametric fit to on target prompts | |
| PAIC | from source labels; isotonic fit to on target prompts | |
| Target isotonic (yardstick) | Target labels fit the map |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Stage | Inputs and label access |
|---|---|
| Fit the probe | Train on its feature inputs and . Count these labels even when later stages use only its predictions. |
| Fit scalar repairs | Label-fitted maps use ; PAIC uses on repair prompts independent of probe training, without their labels. |
| Compare candidates | Evaluate the frozen candidates and on disjoint and ; use only on . Both pools are independent of fitting. |
| Serve the score | PAIC uses ; group calibration also needs a feature-based group; direct needs its feature inputs. None uses . |
| Audit the score | Use , , and the chosen score on new independent prompts, with a kernel fixed before the audit. |
| 20 | |||
|---|---|---|---|
| 50 | |||
| 100 | |||
| 500 | |||
| 2000 | |||
| Population limit |
| Shift | Model | Original | Source iso. | Platt | Beta | Non- mono. | PAIC ridge | PAIC trees | Group | Direct trees | Target |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Difficulty | Llama | .296066 | .251714 | .251608 | .251833 | .253394 | .237544 | .238951 | .249526 | .193440 | .236763 |
| Qwen | .229475 | .210177 | .209458 | .209722 | .211439 | .194298 | .196389 | .206683 | .138683 | .194654 | |
| Task | Llama | .247051 | .238183 | .237836 | .236746 | .238634 | .234934 | .218722 | .246771 | .199584 | .219169 |
| Qwen | .175655 | .185457 | .185327 | .182487 | .182422 | .166735 | .165191 | .184328 | .148355 | .159922 | |
| Subject | Llama | .325099 | .099458 | .103151 | .102349 | .099871 | .083310 | .083183 | .099059 | .054461 | .082144 |
| Qwen | .253062 | .122240 | .129469 | .128146 | .120107 | .117090 | .116237 | .125438 | .074457 | .112542 |
| Dataset | Model | Original | Label iso. | Flexible scalar | Platt | PAIC | Direct |
|---|---|---|---|---|---|---|---|
| GSM8K | Llama | .104379 | .105213 | .104840 | .104688 | .104812 | .106254 |
| Qwen | .085551 | .085418 | .085300 | .085266 | .085281 | .086493 | |
| 10K | Llama | .049106 | .048715 | .048548 | .048779 | .048607 | .048427 |
| Qwen | .054499 | .054428 | .054160 | .054534 | .054235 | .054067 |
| Target-probe Platt | Target-probe beta | Source- | ||||||
|---|---|---|---|---|---|---|---|---|
| Shift | Model | PAIC | CE | SQ | CE | SQ | distilled | IW iso. |
| Difficulty | Llama | .238951 | .240000 | .240173 | .239466 | .239557 | .251711 | .237170 |
| Qwen | .196389 | .198206 | .198034 | .196995 | .196972 | .207772 | .197005 | |
| Task | Llama | .218722 | .219675 | .219552 | .219042 | .218932 | .236728 | .235725 |
| Qwen | .165191 | .167125 | .166397 | .165519 | .165316 | .180906 | .184300 | |
| Subject | Llama | .083183 | .084982 | .084411 | .084557 | .084132 | .099035 | .092045 |
| Comparator | Lower PAIC estimate | Pointwise interval favors PAIC | Global interval favors PAIC |
|---|---|---|---|
| Target-probe Platt (CE) | 18 | 15 | 10 |
| Target-probe beta (CE) | 16 | 10 | 8 |
| Source-distilled isotonic | 18 | 18 | 18 |
| Importance-weighted isotonic | 15 | 12 | 11 |
| Total | 67/72 | 55/72 | 47/72 |
| Target-probe Platt (SQ) | 18 | 13 | 9 |
| Shift | Model | Domain AUC | Raw ESS | Clipped ESS | Lower-clipped fraction |
|---|---|---|---|---|---|
| Difficulty | Llama | .643 | 210.0 | 225.3 | .026 |
| Qwen | .672 | 195.3 | 229.7 | .023 | |
| Task | Llama | .999 | 46.0 | 347.8 | .956 |
| Qwen | .998 | 26.8 | 137.2 | .902 | |
| Subject | Llama | .994 | 24.6 | 56.3 | .808 |
| Qwen | .993 | 7.9 | 43.3 | .798 |
| Model | Candidate | Squared risk | Probe disagreement |
|---|---|---|---|
| Llama | Unchanged score | ||
| Llama | Source label-isotonic | ||
| Llama | Source PAIC | ||
| Llama | Target PAIC | ||
| Llama | Direct probe | ||
| Qwen | Unchanged score |
| Candidate | Llama 3.2 3B | Qwen 2.5 3B |
|---|---|---|
| Original | 0.295817 | 0.377312 |
| Canonical label-isotonic | 0.247903 | 0.241805 |
| Source PAIC | 0.246479 | 0.244422 |
| Target PAIC | 0.250821 | 0.247072 |
| Direct rich teacher | 0.243782 | 0.231066 |
| Direct score-only teacher | 0.244477 | 0.242851 |
| Candidate | MSE | 95% interval for | |
|---|---|---|---|
| Original score | 0.21274 | ||
| Source isotonic | 0.19748 | ||
| Source Platt | 0.19622 | ||
| Raw verifier | 0.19672 | ||
| Calibrated verifier | 0.15506 | ||
| PAIC | 0.19718 |
| Candidate | Loss | Loss | Loss | |||
|---|---|---|---|---|---|---|
| Original score | 0.14732 | 58.03 | 0.25962 | 97.83 | 0.26756 | 100.00 |
| Source isotonic | 0.14448 | 49.16 | 0.24582 | 95.32 | 0.26505 | 97.83 |
| Source Platt | 0.12659 | 33.61 | 0.22952 | 88.46 | 0.26672 | 99.50 |
| Raw verifier | 0.18846 | 88.80 | 0.20778 | 89.80 | 0.23495 | 90.47 |
| Calibrated verifier | 0.11488 | 55.35 | 0.19064 | 84.62 | 0.23328 | 89.80 |
| Population | Two-sided EB | One-sided EB | Betting | Assisted EB | Assisted betting | PPI++ |
|---|---|---|---|---|---|---|
| Binary constant gain | 2,000 | 2,000 | 1,000 | 2,000 | 1,000 | 500 |
| Graded constant gain | 500 | 500 | 250 | 1,000 | 250 | 25 |
| Binary heterogeneous gain | – | 2,000 | 1,000 | – | 1,000 | 1,000 |
| Graded heterogeneous gain | 1,000 | 1,000 | 250 | 2,000 | 250 | 25 |
| Procedure | Beneficial adoption | PAIC certified | PAIC selected | Richer selected | False adoption |
|---|---|---|---|---|---|
| One-sided EB | 7,155 (15.09%) | 5,794 (12.22%) | 47 | 7,108 | 0 |
| Betting | 20,356 (42.95%) | 15,635 (32.99%) | 125 | 20,212 | 0 |
| Assisted EB | 6,610 (13.95%) | 5,440 (11.48%) | 35 | 6,575 | 0 |
| Assisted betting | 19,012 (40.11%) | 14,598 (30.80%) | 183 | 18,815 | 0 |
| Two-sided EB | 6,358 (13.41%) | 5,207 (10.99%) | 130 | 6,228 | 0 |
| Test | Feature map | Null hypothesis under the prompt-weighted law |
|---|---|---|
| Feature | in every joint state. | |
| Score-only | at each score level. | |
| Subgroup | . | |
| Learned auditor | , conditional on the fitted . |
| Configuration | Feature | Score only | Subgroup | Auditor |
|---|---|---|---|---|
| Identity score , binary outcomes, ; configuration: | ||||
| 0.0590 ∘ | 0.0540 ∘ | 0.0570 ∘ | 0.0510 ∘ | |
| 0.1265 | 0.0490 ∘ | 0.1940 | 0.1475 | |
| 0.3625 | 0.0510 ∘ | 0.5950 | 0.5500 | |
| 0.9475 | 0.0505 ∘ | 0.9930 | 0.9915 | |
| 1.0000 | 0.0515 ∘ | 1.0000 | 1.0000 | |
| Partition | Prompts | Learned answers | Mechanical answers | Jaccard | Learned-group error | Mechanical- group error |
|---|---|---|---|---|---|---|
| Discovery | 572 | 200 | 196 | 98.00% | 97.00% | 98.98% |
| Audit | 572 | 243 | 238 | 97.94% | 99.59% | 100.00% |
| Final evaluation | 1,435 | 510 | 498 | 97.65% | 98.82% | 99.20% |
| Dataset | Model | ||||||
|---|---|---|---|---|---|---|---|
| GSM8K | Llama | ||||||
| Qwen | |||||||
| 10K | Llama | ||||||
| Qwen |
| Benchmark | ||||||
|---|---|---|---|---|---|---|
| BoolQ | 400 | 4,393 | 1,597 | 1,598 | — | 3,270 |
| ARC | 112 | 2,126 | 570 | 1,410 | 2,375 | 1,172 |
| Benchmark | Contrast | Effect | Lower | Upper | Direction |
|---|---|---|---|---|---|
| BoolQ | Rich score | unresolved | |||
| BoolQ | PAIC-R- scalar | unresolved | |||
| BoolQ | PAIC-R- PAIC-S- | unresolved | |||
| BoolQ | PAIC-R- rich | first higher risk | |||
| ARC | Rich score | unresolved | |||
| ARC | PAIC-R- scalar | unresolved |
Confidence Estimation in Language Models
Confidence estimation in language models measures whether a model’s uncertainty signals predict answer correctness and support reliable decisions.
See all 229 papers