Cited by 50Artificial Intelligence in Healthcare and EducationMachine Learning in HealthcareArtificial Intelligence in Healthcare

Probabilistic medical predictions of large language models

Open access
Gu, Bowen, Desai, Rishi J., Lin, Kueiyu Joshua, Yang, Jie·2024·npj Digital Medicine

Abstract

Large Language Models (LLMs) have shown promise in clinical applications through prompt engineering, allowing flexible clinical predictions. However, they struggle to produce reliable prediction probabilities, which are crucial for transparency and decision-making. While explicit prompts can lead LLMs to generate probability estimates, their numerical reasoning limitations raise concerns about reliability. We compared explicit probabilities from text generation to implicit probabilities derived from the likelihood of predicting the correct label token. Across six advanced open-source LLMs and five medical datasets, explicit probabilities consistently underperformed implicit probabilities in discrimination, precision, and recall. This discrepancy is more pronounced with smaller LLMs and imbalanced datasets, highlighting the need for cautious interpretation, improved probability estimation methods, and further research for clinical use of LLMs.

AI snapshot

Written from the paper’s full text.

TL;DR

Across six open-source LLMs and five medical datasets, explicit probabilities generated in text were consistently less reliable than implicit probabilities derived from token likelihood for discrimination, precision, and recall. The gap was larger for smaller LLMs and imbalanced datasets.

Key findings

  • Implicit probabilities had higher AUROC and AUPRC than explicit probabilities across all LLMs and datasets, with most differences statistically significant at p < 0.05. For example, gemma-2-27b-it on USMLE had explicit AUROC 85.4% vs implicit 90.2% (difference 4.9%), and Phi-3-medium on MMLU-CM had explicit AUPRC 82.5% vs implicit 91.3% (difference 8.8%).
  • All LLMs achieved at least 0.8 average accuracy on all datasets; Meta-Llama-3.1-70B-Instruct had the highest average accuracy at 91.7%, followed by Qwen2-72B-Instruct at 91.3% and Mistral-Large-Instruct-2407 at 90.0%. Qwen2-72B-Instruct performed best on MCMLE at 96.7%, while Phi-3-medium-128k-instruct performed worst on MCMLE at 69.8%.
  • Small LLMs generally showed greater explicit-vs-implicit AUROC differences than their larger counterparts, while implicit probability differences for small LLMs were much smaller.
  • On imbalanced USMLE reconstructions, AUPRC differences increased as datasets became more imbalanced for most LLMs; explicit probabilities were sometimes below 50% even when the model gave an answer, whereas implicit probability always selected the most probable token.

Methods

  • Evaluated six large open-source LLMs (Qwen2-72B-Instruct, Meta-Llama-3.1-70B-Instruct, gemma-2-27b-it, Mistral-Large-Instruct-2407, Yi-1.5-34B-Chat, Phi-3-medium-128k-instruct) and smaller counterparts on five datasets: MMLU-CK (299 questions), MMLU-CM (200 questions), USMLE (1,000-question subset), MCMLE (1,000-question subset), and MGB-SDoH (200 patients, 9 SDoH aspects).
  • Converted multiple-choice questions to binary A/B tasks, randomly assigning 50% of correct options as Option A and 50% as Option B; extracted explicit probability from generated text and implicit probability from the token probability of the predicted label token (“A” or “B”).
  • Used accuracy, AUROC, and AUPRC as metrics; ran experiments three times at temperatures 0, 0.3, and 0.7; used one-sided paired t-tests; reconstructed USMLE with option-A-correct ratios of 5%, 10%, 30%, 50%, 70%, and 90%; and conducted sensitivity analysis with two additional prompts.

Limitations

  • The paper states: experiments were simplified to binary classification, so conclusions do not carry over to multiple-choice question answering; Chain-of-Thought prompting was not examined because implicit probability extraction is challenging there; only open-source LLMs were studied because proprietary LLMs do not provide token probabilities; and results are primarily based on medical datasets, so generalizability to other domains needs further validation.
  • The text also makes evident that implicit probability extraction is only feasible in limited settings, such as option-selection tasks or tasks without a chain of thought, and is mostly available in open-source LLMs.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

Crossref · 10.1038/s41746-024-01366-4

Cites

41 of 41 references verified0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Claims the paper makes

0 of 18 cited sources checked · 0 supported · 0 flagged

  • Findingown data

    In some cases, while the LLM provides an answer and an explicit probability as instructed, this probability is less than 50%.

    “We find that in some cases, while the LLM provides an answer and an explicit probability as instructed, this probability is less than 50% (Supplementary Table 3).”
    Quote in the paperSubstantial claim
  • Findingown data

    The ROC and PRC curves for the explicit probability is a coarse step function because the LLMs usually give their probabilities in their nearest tenth.

    “We observe that the ROC and PRC curves for the explicit probability is a coarse step function because the LLMs usually give their probabilities in their nearest tenth (e.g., 90%, 80%, etc.), which makes the explicit probability coarse-grained.”
    Quote in the paperSubstantial claim
  • Findingown data

    The Qwen2-72B model shows a clear advantage over other models on the MCMLE dataset.

    “The Qwen2-72B model, which is trained using a rich Chinese corpus, shows a clear advantage over other models on the MCMLE dataset, which consists of Chinese medical examination questions.”
    Quote in the paperNot a substantial claim
  • Findingown data

    The Mistral-Large model's performance is much worse than the Llama-3.1-70B model on the non-English MCMLE dataset (85.2% vs 93.0%).

    “The Mistral-Large model, which has comparable capability to the Llama-3.1-70B model, achieve the highest accuracy for the MMLU-CM and the MGB-SDoH dataset but its performance is much worse than the Llama-3.1-70B model on the non-English MCMLE dataset (85.2% VS 93.0%).”
    Quote in the paperNot a substantial claim
  • Findingown data

    For most LLMs, both their explicit and implicit probability distributions are very polarized, even if their predictions are incorrect.

    “For most LLMs, both their explicit and implicit probability distributions are very polarized, even if their predictions are incorrect.”
    Quote in the paperSubstantial claim
  • May combine two claims. 18
  • Classified as probably not a substantial claim. 13
  • May combine two claims.

    Findingown data

    The Mistral-Large model achieves the highest accuracy for the MMLU-CM and the MGB-SDoH dataset.

    “The Mistral-Large model, which has comparable capability to the Llama-3.1-70B model, achieve the highest accuracy for the MMLU-CM and the MGB-SDoH dataset but its performance is much worse than the Llama-3.1-70B model on the non-English MCMLE dataset (85.2% VS 93.0%).”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingown data

    The only exception is the Llama-3.1-70B model on the USMLE dataset, which is mainly due to the high standard deviation of the implicit probability.

    “The only exception is the Llama-3.1-70B model on the USMLE dataset, which is mainly due to the high standard deviation of the implicit probability.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingown data

    The discrepancy between explicit and implicit probabilities is more pronounced with smaller LLMs and imbalanced datasets.

    “This discrepancy is more pronounced with smaller LLMs and imbalanced datasets, highlighting the need for cautious interpretation, improved probability estimation methods, and further research for clinical use of LLMs.”
    • Also stated in: § Large LLM vs. small LLM · plain
      “Furthermore, small LLMs generally exhibit greater differences between explicit and implicit probability AUROC, as their limited parameters make them less sensitive to explicit probabilities, reducing reliability.”
    • Also stated in: § Imbalanced datasets analysis · hedged
      “Based on Fig. 2, for most LLMs the AUPRC difference increases as the dataset got more imbalanced.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    The explicit probability, even with its narrow distribution, may give values that are close to 50%, while the more widely distributed implicit probability values rarely fall within this area.

    “However, we notice that the explicit probability, even with its narrow distribution, may give values that are close to 50%, while the more widely distributed implicit probability values rarely fall within this area.”
    Quote in the paperNot a substantial claim
  • Classified as probably not a substantial claim.

    Findingreasoning

    The steps of the implicit probability are barely discernible in the plot.

    “In contrast, the steps of the implicit probability are barely discernible in the plot.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    This finding is valid on both a non-English dataset (MCMLE) and a dataset from private electronic health records (MGB-SDoH).

    “This finding is valid on both a non-English dataset (MCMLE) and a dataset from private electronic health records (MGB-SDoH).”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim.

    Premisecitation

    During LLM inference, there is a correlation between predictive uncertainty and hallucination.

    “Recent work has shown that during LLM inference, there is a correlation between predictive uncertainty and hallucination — a behavior where LLMs generate false facts or knowledge not supported by input39.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [39] Xiao, Y. & Wang, W. Y. Proceedings of the 16th Conference of the European Chapter of the Association for Computational Linguistics: Main Volume.

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    In the clinical domain, LLMs often underperform due to limited training data for minority languages, lack of domain knowledge, and dataset imbalances.

    “In the clinical domain, due to limited training data for minority languages, lack of domain knowledge, and dataset imbalances, LLMs often underperforms34.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [34] Wu, J. et al. Clinical text datasets for medical artificial intelligence and large language models—a systematic review. NEJM AI 1, AIra2400012 (2024). source

      Not checked in the free run.

    • Rests on an assumption: Many countries’ medical systems use minority languages that most LLMs are not familiar with.
  • May combine two claims.

    Interpretationreasoning

    The implicit probability always selects the most probable token, avoiding such contradictions and supporting its reliability over the explicit probability.

    “In contrast, the implicit probability always selects the most probable token, avoiding such contradictions and supporting its reliability over the explicit probability.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Study design and conductown data

    All other models have a less than 10% fail rate on the validation process, with 78.6% of the model inference on one dataset have a less than 1% fail rate on the validation process.

    “All other models have a less than 10% fail rate on the validation process, with 78.6% of the model inference on one dataset have a less than 1% fail rate on the validation process.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Most LLMs are trained mainly on English corpora.

    “Most LLMs are trained mainly on English corpora35, while many countries’ medical systems use minority languages that most LLMs are not familiar with.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [35] Blevins, T. & Zettlemoyer, L. Language contamination explains the cross-lingual capabilities of English pretrained models. In Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing, 3563–3574, Abu Dhabi, United Arab Emirates (Association for Computational Linguistics 2022). source

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    Advanced LLMs often rely on general domain datasets and lack exposure to real-world EHR notes, which limits their effectiveness in clinical settings.

    “Furthermore, due to data access and privacy concerns34, advanced LLMs often rely on general domain datasets and lack exposure to real-world EHR notes, which limits their effectiveness in clinical settings24,36.”
    Quote in the paperNot a substantial claim0 of 3 sources checked
    • cites [34] Wu, J. et al. Clinical text datasets for medical artificial intelligence and large language models—a systematic review. NEJM AI 1, AIra2400012 (2024). source

      Not checked in the free run.

    • cites [24] Wu, J. et al. Large language models leverage external knowledge to extend clinical insight beyond language boundaries. J. American Medical Informatics Association, ocae079 (2024). source

      Not checked in the free run.

    • cites [36] Wu, J., Wu, X. & Yang, J. in Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence, {IJCAI-24} (ed Kate L.) 7491–7499 (International Joint Conferences on Artificial Intelligence Organization, 2024). source

      Not checked in the free run.

  • Classified as probably not a substantial claim. May combine two claims.

    Premisecitation

    Incorrect estimation can lead to misinformed decisions, compromising patient safety and treatment efficacy.

    “Incorrect estimation can lead to misinformed decisions, compromising patient safety and treatment efficacy32,33.”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [32] Koçak, B. et al. Bias in artificial intelligence for medical imaging: fundamentals, detection, avoidance, mitigation, challenges, ethics, and prospects. Diagn Interv Radiol (2024). source

      Not checked in the free run.

    • cites [33] Albahri, A. S. et al. A systematic review of trustworthy and explainable artificial intelligence in healthcare: Assessment of quality, bias risk, and data fusion. Inf. Fusion 96, 156–191 (2023). source

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    Medical datasets frequently suffer from imbalances in phenotypes, like rare diseases or serious clinical outcomes, and cohort identification.

    “Medical datasets also frequently suffer from imbalances in phenotypes, like rare diseases or serious clinical outcomes37, and cohort identification7.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [37] Schubach, M., Re, M., Robinson, P. N. & Valentini, G. Imbalance-aware machine learning for predicting rare and common disease-associated non-coding variants. Sci. Rep. 7, 2959 (2017). source

      Not checked in the free run.

    • cites [7] Yang, J. et al. Development and validation of a deep learning model for detection of allergic reactions using safety event reports across hospitals. JAMA Netw. Open 3, e2022836–e2022836 (2020). source

      Not checked in the free run.

  • May combine two claims.

    Study design and conductown data

    All LLMs follow instructions well, with outputs strictly adhering to the format in the prompt, except for the Qwen2-7B model on the USMLE dataset.

    “All LLMs follow instructions well, with outputs strictly adhering to the format in the prompt, except for the Qwen2-7B model on the USMLE dataset.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Interpretationreasoning

    The decrease in AUROC when using the second auxiliary prompt is due to the introduction of the setting of a student proposing answers to the LLM, which introduces additional complexity and lowered the model's performance.

    “We propose that this is due to the introduction of the setting of a student proposing answers to the LLM, which introduces additional complexity and lowered the model’s performance.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Premisecitation

    A primary cause of low explicit probability is insufficient knowledge, which can drive LLMs to produce hallucinated information for predictions.

    “A primary cause of low explicit probability is insufficient knowledge, which can drive LLMs to produce hallucinated information for predictions40,41.”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [40] Martino, A., Iannelli, M. & Truong, C. in European Semantic Web Conference. 182–185 (Springer).

      Not checked in the free run.

    • cites [41] Liu, X. et al. Uncovering language disparity of ChatGPT on retinal vascular disease classification: cross-sectional study. J. Med. Internet Res. 26, e51926 (2024). source

      Not checked in the free run.

  • May combine two claims.

    Interpretationown data

    This implies that although the model proposes an answer, it assigns a higher probability to the alternative, undermining its reliability.

    “This implies that although the model proposes an answer, it assigns a higher probability to the alternative, undermining its reliability.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductnone given

    Mass General Brigham Institutional Review Board approved the study protocol (2020P001486).

    “Mass General Brigham (MGB) Institutional Review Board approved the study protocol (2020P001486).”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Study design and conductreasoning

    The study only applies to open-sourced LLMs since it is difficult to obtain the implicit probability of tokens using proprietary LLMs.

    “Additionally, our study only applies to open-sourced LLMs since it is difficult to obtain the implicit probability of the tokens using proprietary LLMs.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim.

    Study design and conductcitation

    To eliminate the impact of option order on LLM performance, 50% of correct options were randomly assigned as Option A and the remaining 50% as Option B.

    “To eliminate the impact of the option order to LLMs performance55, we randomly assign 50% of the correct options as Option A and the remaining 50% of the correct options as Option B.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [55] Pezeshkpour, P. & Hruschka, E. in Findings of the Association for Computational Linguistics: NAACL 2024. 2006–2017.

      Not checked in the free run.

  • Classified as probably not a substantial claim.

    Study design and conductown data

    To ensure robustness, a sensitivity analysis was conducted using two additional prompts on the USMLE dataset to analyze if the relative difference between explicit and implicit probability persists under different prompt settings.

    “To ensure the robustness of our results, we conduct a sensitivity analysis by using two additional prompts (Supplementary Table 1) on the USMLE dataset and analyze if the relative difference between the explicit and the implicit probability persists under different prompt settings.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Study design and conductreasoning

    The study does not examine probability performance using Chain of Thought (CoT) prompting, as it is challenging to extract implicit probabilities in these cases.

    “Besides, we do not examine the probability performance using Chain of Thought (CoT) prompting, as it is challenging to extract implicit probabilities in these cases.”
    Quote in the paperNot a substantial claim
  • Classified as probably not a substantial claim. May combine two claims.

    Study design and conductnone given

    Validation scripts were implemented to ensure that extracted content aligns with model predictions and falls within the two candidate options.

    “Validation scripts are implemented to ensure that the extracted content aligns with the model predictions and falls within the two candidate options.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim.

    Premisecitation

    AUPRC is a commonly used metric to evaluate model performance on datasets with imbalanced labels.

    “Since the metric of AUPRC is a commonly used metric to evaluate model performance on datasets with imbalanced labels56,57, we calculate the AUPRC of the two models and compare their differences.”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [56] Wu, X., Huang, F. & Huang, H. in 2022 IEEE International Conference on Data Mining (ICDM). 578–587 (IEEE). source

      Not checked in the free run.

    • cites [57] Qi, Q., Luo, Y., Xu, Z., Ji, S. & Yang, T. Stochastic optimization of areas under precision-recall curves with provable convergence. Adv. neural Inf. Process. Syst. 34, 1752–1765 (2021).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Interpretationown data

    The results are primarily based on medical datasets, and generalizability to different domains needs confirmation by further validation studies.

    “Finally, our results are primarily based on medical datasets, the generalizability to different domains needs to be confirmed by further validation studies.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductnone given

    Consent to participate was not deemed required for this observational investigation.

    “Consent to participate was not deemed required for this observational investigation.”
    Quote in the paperNot a substantial claim
27 of 27
  • The paper relies on this without support.

    The MGB-SDoH dataset was not used in LLM training.

    “Since such polarization still persists on the MGB-SDoH dataset, a private dataset that can’t be touched during LLM training, we argue that such confidence does not originate from the data leakage31, but is an intrinsic property of the LLMs.”
  • The paper relies on this without support.

    Labels in clinical practice often have imbalanced distributions, and the performance of AI models is affected by such imbalance.

    “Labels in clinical practice often have imbalanced distributions, and the performance of AI models are affected by such imbalance.”
  • The paper relies on this without support.

    Many countries’ medical systems use minority languages that most LLMs are not familiar with.

    “Most LLMs are trained mainly on English corpora35, while many countries’ medical systems use minority languages that most LLMs are not familiar with.”
3 of 3

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

  • Qwen2-72B is an unacknowledged exception to higher implicit AUROC

    Minor

    The text names Llama-3.1-70B on USMLE as the only exception, but Table 2 shows another: on USMLE, Qwen2-72B has a lower implicit AUROC (89.5%) than explicit AUROC (90.5%).

1 of 1

Cites

Flagged first, then by citations

Research works printed but not linked: 11

  1. Devlin, J., Chang, M.-W., Lee, K. & Toutanova, K. in Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers). 4171–4186.
  2. Brown, T. et al. Language models are few-shot learners. Adv. neural Inf. Process. Syst. 33, 1877–1901 (2020).
  3. Zhou, W. & Ngo, T. H. Using Pretrained Large Language Model with Prompt Engineering to Answer Biomedical Questions. In Proceedings of the Conference and Labs of the Evaluation Forum, 3740, 253–268 (2024).
  4. Radford, A. et al. Language models are unsupervised multitask learners. OpenAI blog 1, 9 (2019).
  5. Wightman, G. P., Delucia, A. & Dredze, M. in Proceedings of the 3rd Workshop on Trustworthy Natural Language Processing (TrustNLP 2023). 326–362.
  6. Wang, X. et al. The Eleventh International Conference on Learning Representations.
  7. Xiao, Y. & Wang, W. Y. Proceedings of the 16th Conference of the European Chapter of the Association for Computational Linguistics: Main Volume.
  8. Martino, A., Iannelli, M. & Truong, C. in European Semantic Web Conference. 182–185 (Springer).
  9. Hendrycks, D. et al. Aligning AI with shared human values. In Proceedings of the International Conference on Learning Representations (2021).
  10. Pezeshkpour, P. & Hruschka, E. in Findings of the Association for Computational Linguistics: NAACL 2024. 2006–2017.
  11. Qi, Q., Luo, Y., Xu, Z., Ji, S. & Yang, T. Stochastic optimization of areas under precision-recall curves with provable convergence. Adv. neural Inf. Process. Syst. 34, 1752–1765 (2021).

Cited by

Sorted by citations

Authors
Gu, B., Desai, R. J., Lin, K. J., & Yang, J.
Affiliations
  • Brigham and Women's Hospital (US)
  • Broad Institute (US)
Corresponding author
Yang, Jie
Venue
npj Digital Medicine
Published
2024-12-19
Volume
7
Issue
1
Pages
367-367
ISSN
2398-6352 (ISSN-L)
OpenAlex
W4405597099
Cited by
50
References
38
Citations by year
2026: 30 · 2025: 20
FWCI
4.67
Type
journal article
Language
en
Open access
Locations
  • npj Digital Medicine (publishedVersion, cc-by-nc-nd)
  • PubMed Central (publishedVersion, cc-by-nc-nd)
  • PubMed (publishedVersion)
  • arXiv (Cornell University) (submittedVersion, cc-by-nc-nd)
  • DOAJ (DOAJ: Directory of Open Access Journals) (submittedVersion)
  • arXiv (Cornell University)
Funders
National Institutes of Health (R01LM013204)
Topics
  • Health Sciences › Medicine › Health Informatics › Artificial Intelligence in Healthcare and Education
  • Physical Sciences › Computer Science › Artificial Intelligence › Machine Learning in Healthcare
  • Health Sciences › Health Professions › Health Information Management › Artificial Intelligence in Healthcare
Keywords
large language models, medical prediction, predicted probabilities, clinical decision support, imbalanced medical data, model size effects

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-09.

Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-09.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review