Evaluating and Recalibrating Probabilistic Classifiers from One-Bit Threshold Queries
Organizations: Institut Polytechnique de Paris · Onepoint, France · Ghent University
Abstract
Standard calibration evaluation and recalibration methods assume access to predicted probabilities. We study one-bit threshold access, a setting in which probabilities are hidden and the evaluator observes only whether the predicted probability exceeds a single chosen threshold per sample. Under this restriction, we show that classifiers can nevertheless be evaluated and recalibrated without reconstructing their hidden scores. We derive a consistent estimator of True Calibration Error, unbiased estimators of strictly proper scoring rules, and a one-bit objective for Temperature Scaling whose expectation matches the standard log-loss objective. Across these settings, we quantify the statistical cost incurred by replacing standard probability-access procedures by our one-bit estimators. Finally, we instantiate the framework for black-box LLMs, where the logit_bias API parameter allows us to realize the required threshold queries in zero-shot binary question answering.
Figures & tables
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Mean Abs. Probability Shift |
|---|---|
| Mistral v0.3-7B | 23.2% |
| Llama 3.1-8B | 19.7% |
| Qwen 2.5-7B | 5.6% |
| Gemma 2-9B | 4.0% |
| Model | ID A | Decoded A | ID B | Decoded B |
|---|---|---|---|---|
| Qwen 2.5-7B | 32 | A | 33 | B |
| Llama 3.1-8B | 32 | A | 33 | B |
| Mistral v0.3-7B | 29509 | A | 29528 | B |
| Gemma 2-9B | 235280 | A | 235305 | B |
| Estimator | Model Calls | Generated Tokens / Call | Logit Interventions |
|---|---|---|---|
| One-bit (ours) | 1 | 1 | 1 |
| Verbalized Confidence | 1 | 10 | 0 |
| Sampling Frequency ( ) | 1 | 0 | |
| Carlini Binary Search ( ) | 1 |
| Single query | Sampling | Carlini | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Model | Dataset | ECE ref. | One-bit | Verb. | ||||||
| Gemma-2-9B | BoolQ | 0.218 | 0.025 | 0.115 | 0.110 | 0.024 | 0.010 | 0.101 | 0.003 | 0.002 |
| Llama-3.1-8B | BoolQ | 0.114 | 0.018 | 0.208 | 0.305 | 0.082 | 0.044 | 0.271 | 0.035 | 0.001 |
| Mistral-7B-v0.3 | BoolQ | 0.265 | 0.011 | 0.089 | 0.201 | 0.047 | 0.011 | 0.192 | 0.009 | 0.001 |
| Qwen2.5-7B | BoolQ | 0.346 | 0.057 | 0.143 | 0.030 | 0.004 | 0.002 | 0.040 | ||
| Gemma-2-9B | MMLU | 0.131 | 0.001 | 0.110 | 0.164 | 0.043 | 0.021 | 0.133 | 0.004 | |
Language Model Calibration
Language model calibration studies whether models’ confidence estimates match their actual accuracy across tasks, prompts, and deployment settings.
See all 381 papers