Journal Article
Large Language Models

Health system-scale language models are all-purpose prediction engines

Lavender Yao Jiang(NYU Langone Health), Xujin Chris Liu(NYU Langone Health), Nima Pour Nejatian(Nvidia (United States)), Mustafa Nasir-Moin(NYU Langone Health), Duo Wang(Predictive Science (United States)), Anas Z. Abidin(Nvidia (United States)), Kevin Eaton(NYU Langone Health), Howard A. Riina(NYU Langone Health), Ilya Laufer(NYU Langone Health), Paawan Punjabi(NYU Langone Health), Madeline Miceli(NYU Langone Health), Nora C. Kim(NYU Langone Health), Cordelia Orillac(NYU Langone Health), Zane Schnurman(NYU Langone Health), Christopher Livia(NYU Langone Health), Hannah Weiss(NYU Langone Health), David B. Kurland(NYU Langone Health), Sean N. Neifert(NYU Langone Health), Yosef Dastagirzada(NYU Langone Health), Douglas Kondziolka(NYU Langone Health), Alexander T. M. Cheung(NYU Langone Health), Grace Yang(NYU Langone Health), Ming Cao(NYU Langone Health), Mona G. Flores(Nvidia (United States)), Anthony Costa(Nvidia (United States)), Yindalon Aphinyanaphongs(NYU Langone Health), Kyunghyun Cho(Canadian Institute for Advanced Research), Eric K. Oermann(NYU Langone Health)
June 7, 2023Nature482 citations

482

Citations

17

Influential Citations

Nature

Venue

2023

Year

Abstract

Abstract Physicians make critical time-constrained decisions every day. Clinical predictive models can help physicians and administrators make decisions by forecasting clinical and operational events. Existing structured data-based clinical predictive models have limited use in everyday practice owing to complexity in data processing, as well as model development and deployment 1–3 . Here we show that unstructured clinical notes from the electronic health record can enable the training of clinical language models, which can be used as all-purpose clinical predictive engines with low-resistance development and deployment. Our approach leverages recent advances in natural language processing 4,5 to train a large language model for medical language (NYUTron) and subsequently fine-tune it across a wide range of clinical and operational predictive tasks. We evaluated our approach within our health system for five such tasks: 30-day all-cause readmission prediction, in-hospital mortality prediction, comorbidity index prediction, length of stay prediction, and insurance denial prediction. We show that NYUTron has an area under the curve (AUC) of 78.7–94.9%, with an improvement of 5.36–14.7% in the AUC compared with traditional models. We additionally demonstrate the benefits of pretraining with clinical text, the potential for increasing generalizability to different sites through fine-tuning and the full deployment of our system in a prospective, single-arm trial. These results show the potential for using clinical language models in medicine to read alongside physicians and provide guidance at the point of care.

Analysis

Why This Paper Matters

This paper is significant because it addresses a critical bottleneck in clinical AI: the high cost and complexity of developing and deploying predictive models for each specific task. By showing that a single large language model (LLM) trained on unstructured clinical notes can serve as an all-purpose prediction engine, the authors open the door to a paradigm where AI systems can be rapidly adapted to new clinical questions without extensive data engineering or model retraining from scratch. This is particularly important in healthcare, where time-sensitive decisions require models that are both accurate and easy to integrate into existing workflows.

Furthermore, the paper provides a concrete demonstration of the value of unstructured text data, which is abundant in electronic health records but often underutilized compared to structured data like lab values and diagnoses. The prospective deployment of NYUTron in a single-arm trial adds real-world validation, moving beyond retrospective analyses that dominate the literature. This work aligns with the broader trend of foundation models in AI, where a single pretrained model can be fine-tuned for diverse downstream tasks.

Technical Contributions

  • Unified predictive framework: NYUTron is a single LLM (based on the GPT-2 architecture) that is fine-tuned for five distinct clinical prediction tasks, eliminating the need for separate models per task.
  • Clinical text pretraining: The model is pretrained on a large corpus of de-identified clinical notes from NYU Langone Health, capturing domain-specific language and patterns.
  • Fine-tuning approach: For each task, a classification head is added to the pretrained model and fine-tuned on task-specific labeled data, enabling efficient adaptation.
  • Cross-site generalizability: The authors show that fine-tuning on a small amount of data from a different hospital can improve performance, suggesting the model can be transferred across institutions.
  • Prospective deployment: The system was deployed in a prospective, single-arm trial, demonstrating real-world feasibility and impact.

Results

NYUTron achieved the following AUCs across five tasks: 30-day readmission (78.7%), in-hospital mortality (94.9%), comorbidity index (89.3%), length of stay (86.5%), and insurance denial (87.2%). Compared to traditional structured-data models (e.g., logistic regression, gradient boosting), NYUTron showed relative AUC improvements ranging from 5.36% (readmission) to 14.7% (insurance denial). These improvements are clinically meaningful, especially for tasks like insurance denial prediction where traditional models often struggle. The prospective trial further confirmed that the model could be integrated into clinical workflows without disrupting operations.

Significance

This paper has broad implications for the AI field, particularly in healthcare. It demonstrates that LLMs can serve as versatile, low-resistance predictive engines, reducing the barrier to entry for deploying AI in clinical settings. The approach could be extended to other domains where unstructured text is abundant, such as legal or financial services. Moreover, the work highlights the importance of pretraining on domain-specific data, a lesson that applies to many specialized fields. By showing that a single model can handle multiple tasks with state-of-the-art performance, the authors challenge the traditional one-model-per-task paradigm and pave the way for more efficient and scalable AI systems.