mlsanityData to 14 Oct 2026

Uncertainty-Aware Calibrated Clinical Text Classification with Large Language Models

arXiv 2509.19375 ↗First posted 19 Sep 2025cs.LG cs.AI stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Large language models are increasingly used for clinical text classification, where overconfident misclassifications can directly affect patient care. Existing black-box uncertainty methods attach a confidence score to a fixed LLM prediction using softmax probabilities, verbalised confidence, prompt agreement, or generation consistency. These signals are often poorly calibrated and offer no mechanism for combining model evidence with prior clinical belief. We instead formulate closed-set clinical classification as likelihood-free posterior inference over diagnostic hypotheses. A prompt-conditioned LLM is treated as a class-conditional stochastic simulator: for each candidate diagnosis it generates synthetic clinical descriptions, which are compared with the observed case in an embedding summary space. Sequential Monte Carlo Approximate Bayesian Computation then yields a posterior over diagnoses from which both the prediction and its uncertainty are derived. We instantiate three variants spanning a cost-fidelity spectrum: multinomial SMC (M-SMC), Dirichlet SMC (D-SMC), and an amortised Dirichlet posterior network (ABI) for deployment-time inference. Across three clinical benchmarks and six open LLMs, the inferred posterior separates correct predictions from errors more sharply than black-box confidence, with misclassification-detection AUROC on MultiCare of 0.923 for M-SMC against 0.760 for softmax and 0.584 for a consistency baseline. The posterior also accepts an explicit clinician prior, recovering the correct diagnosis from a deliberately misleading one, and remains diffuse on out-of-distribution presentations. We release OLB-300, an expert-validated clinical uncertainty-quantification benchmark of oral-lesion vignettes.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 36 references that are in the map. First posted on arXiv 19 Sep 2025.

Builds on

19 of 36 references.

References · 19

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2023Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMscited by 974
  3. 2017Selective Classification for Deep Neural Networkscited by 852
  4. 2023Semantic Uncertainty: Linguistic Invariances for Uncertainty Estimation in Natural Language Generationcited by 819
  5. 2023SelfCheckGPT: Zero-Resource Black-Box Hallucination Detection for Generative Large Language Modelscited by 810
  6. 2018Predictive Uncertainty Estimation via Prior Networkscited by 557
  7. 2020Calibration of Pre-trained Transformerscited by 315
  8. 2023Orca 2: Teaching Small Language Models How to Reasoncited by 159
  9. 2023AlpaCare:Instruction-tuned Large Language Models for Medical Applicationcited by 90
  10. 2024Med42-v2: A Suite of Clinical LLMscited by 90
  11. 2022Prompt Consistency for Zero-Shot Task Generalizationcited by 67
  12. 2024SPUQ: Perturbation-Based Uncertainty Quantification for Large Language Modelscited by 56
7 more
  1. 2024Calibrating the Confidence of Large Language Models by Eliciting Fidelitycited by 47
  2. 2023On Second-Order Scoring Rules for Epistemic Uncertainty Quantificationcited by 38
  3. 2025Simulation-Based Inference: A Practical Guidecited by 32
  4. 2023Reliable Gradient-free and Likelihood-free Prompt Tuningcited by 8
  5. 2025Uncertainty-aware abstention in medical diagnosis based on medical textscited by 6
  6. 2025The Consistency Hypothesis in Uncertainty Quantification for Large Language Modelscited by 6
  7. 2016SimBa: An Efficient Tool for Approximating Rips-filtration Persistence via Simplicial Batch-collapsecited by 3

Built on by

Not cited yet.