Maintenance Notice

Due to necessary scheduled maintenance, the JMIR Publications website will be unavailable from Wednesday, July 01, 2020 at 8:00 PM to 10:00 PM EST. We apologize in advance for any inconvenience this may cause you.

Who will be affected?

Accepted for/Published in: JMIR AI

Date Submitted: Aug 25, 2023
Date Accepted: Jun 14, 2024

The final, peer-reviewed published version of this preprint can be found here:

Traditional Machine Learning, Deep Learning, and BERT (Large Language Model) Approaches for Predicting Hospitalizations From Nurse Triage Notes: Comparative Evaluation of Resource Management

Patel D, Timsina P, Gorenstein L, Glicksberg BS, Raut G, Cheetirala SN, Santana F, Tamegue J, Kia A, Zimlichman E, Levin MA, Freeman R, Klang E

Traditional Machine Learning, Deep Learning, and BERT (Large Language Model) Approaches for Predicting Hospitalizations From Nurse Triage Notes: Comparative Evaluation of Resource Management

JMIR AI 2024;3:e52190

DOI: 10.2196/52190

PMID: 39190905

PMCID: 11387908

Comparative Evaluation of Traditional Machine Learning, Deep Learning and BERT(LLM) Approaches for Predicting Hospitalizations from Nurse Triage Notes: Insights for Resource Management

  • Dhavalkumar Patel; 
  • Prem Timsina; 
  • Larisa Gorenstein; 
  • Benjamin S Glicksberg; 
  • Ganesh Raut; 
  • Satya Narayan Cheetirala; 
  • Fabio Santana; 
  • Jules Tamegue; 
  • Arash Kia; 
  • Eyal Zimlichman; 
  • Matthew A Levin; 
  • Robert Freeman; 
  • Eyal Klang

ABSTRACT

Predicting hospitalization from nurse triage notes has significant implications in health informatics. To this end, we compared the performance of the deep-learning transformer-based model, bio-clinical-BERT, with a bag-of-words logistic regression model incorporating term frequency-inverse document frequency (BOW-LR-tf-idf). A retrospective analysis was conducted using data from 1,391,988 Emergency Department patients at the Mount Sinai Health System spanning 2017-2022. The models were trained on four hospitals' data and externally validated on a fifth. Bioclinical- BERT achieved higher AUCs (0.82, 0.84, and 0.85) compared to BOW-LR-tfidf (0.81, 0.83, and 0.84) across training sets of 10,000, 100,000, and ~1,000,000 patients respectively. Notably, both models proved effective at utilizing triage notes for prediction, despite the modest performance gap. Importantly, our findings suggest that simpler machine learning models like BOW-LR-tf-idf could serve adequately in resource-limited settings. Given the potential implications for patient care and hospital resource management, further exploration of alternative models and techniques is warranted to enhance predictive performance in this critical domain.


 Citation

Please cite as:

Patel D, Timsina P, Gorenstein L, Glicksberg BS, Raut G, Cheetirala SN, Santana F, Tamegue J, Kia A, Zimlichman E, Levin MA, Freeman R, Klang E

Traditional Machine Learning, Deep Learning, and BERT (Large Language Model) Approaches for Predicting Hospitalizations From Nurse Triage Notes: Comparative Evaluation of Resource Management

JMIR AI 2024;3:e52190

DOI: 10.2196/52190

PMID: 39190905

PMCID: 11387908

Download PDF


Request queued. Please wait while the file is being generated. It may take some time.

© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.