Hospital readmission within 30 days of Intensive Care Unit (ICU) discharge remains a significant challenge, linked to increased mortality and healthcare costs. We present BERTBoost, a novel hybrid machine learning framework that integrates structured electronic health record (EHR) variables (e.g., demographics, diagnoses, lab values) with semantic embeddings derived from discharge summaries using ClinicalBERT. The model was trained on the MIMIC-III dataset, with class imbalance addressed using Synthetic Minority Over-sampling Technique (SMOTE), and evaluated on a held-out test set. The final cohort included 7,917 ICU stays, with approximately 30% experiencing a 30-day readmission. BERTBoost achieved 89.7% accuracy, 95.4% precision, 72.6% recall, an F1-score of 0.83, and a ROC–AUC of 0.922, outperforming structured-only base-lines (∼70% accuracy, AUC 0.75). The inclusion of unstructured text improved recall from 50% to 72.6%, representing a 46% relative gain over the structured-only XGBoost baseline, by capturing clinical context often missed by coded data. Key predictors included comorbidity indices and principal components derived from discharge summaries. These findings highlight the advantages of combining large language models with traditional EHR features to improve ICU readmission prediction, enable earlier interventions, and enhance overall healthcare quality.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A BERTBoost Framework for Risk Stratification of Intensive Care Unit Patients Across Hospital Networks

  • Karansinh Rathod,
  • Kartik Kasana,
  • Janhvi Agrawal,
  • Rahul Katarya

摘要

Hospital readmission within 30 days of Intensive Care Unit (ICU) discharge remains a significant challenge, linked to increased mortality and healthcare costs. We present BERTBoost, a novel hybrid machine learning framework that integrates structured electronic health record (EHR) variables (e.g., demographics, diagnoses, lab values) with semantic embeddings derived from discharge summaries using ClinicalBERT. The model was trained on the MIMIC-III dataset, with class imbalance addressed using Synthetic Minority Over-sampling Technique (SMOTE), and evaluated on a held-out test set. The final cohort included 7,917 ICU stays, with approximately 30% experiencing a 30-day readmission. BERTBoost achieved 89.7% accuracy, 95.4% precision, 72.6% recall, an F1-score of 0.83, and a ROC–AUC of 0.922, outperforming structured-only base-lines (∼70% accuracy, AUC 0.75). The inclusion of unstructured text improved recall from 50% to 72.6%, representing a 46% relative gain over the structured-only XGBoost baseline, by capturing clinical context often missed by coded data. Key predictors included comorbidity indices and principal components derived from discharge summaries. These findings highlight the advantages of combining large language models with traditional EHR features to improve ICU readmission prediction, enable earlier interventions, and enhance overall healthcare quality.