JCM, Vol. 13, Pages 5503: Predicting ICU Readmission from Electronic Health Records via BERTopic with Long Short Term Memory Network Approach

JCM, Vol. 13, Pages 5503: Predicting ICU Readmission from Electronic Health Records via BERTopic with Long Short Term Memory Network Approach

Journal of Clinical Medicine doi: 10.3390/jcm13185503

Authors: Chih-Chou Chiu Chung-Min Wu Te-Nien Chien Ling-Jing Kao Chengcheng Li

Background: The increasing rate of intensive care unit (ICU) readmissions poses significant challenges in healthcare, impacting both costs and patient outcomes. Predicting patient readmission after discharge is crucial for improving medical quality and reducing expenses. Traditional analyses of electronic health record (EHR) data have primarily focused on numerical data, often neglecting valuable text data. Methods: This study employs a hybrid model combining BERTopic and Long Short-Term Memory (LSTM) networks to predict ICU readmissions. Leveraging the MIMIC-III database, we utilize both quantitative and text data to enhance predictive capabilities. Our approach integrates the strengths of unsupervised topic modeling with supervised deep learning, extracting potential topics from patient records and transforming discharge summaries into topic vectors for more interpretable and personalized predictions. Results: Utilizing a comprehensive dataset of 36,232 ICU patient records, our model achieved an AUROC score of 0.80, thereby surpassing the performance of traditional machine learning models. The implementation of BERTopic facilitated effective utilization of unstructured data, generating themes that effectively guide the selection of relevant predictive factors for patient readmission prognosis. This significantly enhanced the model’s interpretative accuracy and predictive capability. Additionally, the integration of importance ranking methods into our machine learning framework allowed for an in-depth analysis of the significance of various variables. This approach provided crucial insights into how different input variables interact and impact predictions of patient readmission across various clinical contexts. Conclusions: The practical application of BERTopic technology in our hybrid model contributes to more efficient patient management and serves as a valuable tool for developing tailored treatment strategies and resource optimization. This study highlights the significance of integrating unstructured text data with traditional quantitative data to develop more accurate and interpretable predictive models in healthcare, emphasizing the importance of individualized care and cost-effective healthcare paradigms.