Probabilistic medical predictions of large language models

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.

Authors

Institutions

Publication Details

Journal
npj Digital Medicine
Published
2024-12-19
DOI
https://doi.org/10.1038/s41746-024-01366-4
Citations
25
Primary Topic
Topic Modeling
Type
article
Field-Weighted Citation Impact
8.68

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Probabilistic medical predictions of large language models

Rishi Desai, Kueiyu Joshua Lin, Jie Yang, Bowen Gu
25 citations
npj Digital Medicine
Topic Modeling
8.68
article

Probabilistic medical predictions of large language models

Rishi Desai, Kueiyu Joshua Lin, Jie Yang, Bowen Gu
article en
25 citations

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.

npj Digital MedicineVol. 7(1)
Broad Institute (US), Brigham and Women's Hospital (US), Harvard University (US), Harvard University Press (US)
National Institutes of Health
Quality Education
Openalex Percentile: Top 2%
Topic Modeling
8.68
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.