A Calibrated Reflection Approach for Enhancing Confidence Estimation in LLMs
Organizations: Amazon.com Seattle, Washington, USA
Abstract
A critical challenge in deploying Large Language Models (LLMs) is developing reliable mechanisms to estimate their confidence, enabling systems to determine when to trust model outputs versus seek human intervention. We present a Calibrated Reflection approach for enhancing confidence estimation in LLMs, a framework that combines structured reasoning with distance-aware calibration technique. Our approach introduces three key innovations: (1) a Maximum Confidence Selection (MCS) method that comprehensively evaluates confidence across all possible labels, (2) a reflection-based prompting mechanism that enhances reasoning reliability, and (3) a distance-aware calibration technique that accounts for ordinal relationships between labels. We evaluate our framework on diverse datasets, including HelpSteer2, Llama T-REx, and a proprietary conversational dataset, demonstrating its effectiveness across both conversational and fact-based classification tasks. This work contributes to the broader goal of developing reliable and well-calibrated confidence estimation methods for LLMs, enabling informed decisions about model trust and human judgement.
Figures & tables
| Method | Brier | ECE | AUROC | AUPRC |
| Log Prob | 0.117 | 0.101 | 0.623 | 0.895 |
| CoT | 0.098 | 0.093 | 0.667 | 0.908 |
| Self-Consistency | 0.095 | 0.086 | 0.672 | 0.913 |
| Debate | 0.104 | 0.078 | 0.692 | 0.927 |
| Reflection | 0.110 | 0.080 | 0.687 | 0.918 |
| Method | Brier | ECE | AUROC | AUPRC |
| MCS | .138 | .169 | .591 | .899 |
| +Reflection | .256 | .301 | .697 | .925 |
| +Debate | .256 | .322 | .700 | .924 |
| Method | Brier Score | ECE | AUROC | AUPRC |
| MCS | 0.13799 | 0.16877 | 0.59146 | 0.89905 |
| MCS+Debate (Ours) | 0.25574 | 0.32192 | 0.69999 | 0.9243 |
| MCS+Reflection (Ours) | 0.25617 | 0.30092 | 0.6969 | 0.92534 |
| MCS+Debate+Calibration (Proposed) | 0.1282 | 0.19243 | 0.73752 | 0.93334 |
| MCS+Reflection+Calibration (Proposed Best) | 0.12502 | 0.17472 | 0.73994 | 0.93516 |
| Method | AUPRC |
| Verbalized Confidence | 0.700 |
| Log Prob | 0.709 |
| Trained Probe (SOTA) | 0.910 |
| Verbalized Debate (Proposed) | 0.887 |
| Verbalized Reflection (Proposed) | 0.890 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
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 247 papers