1
|
Li J, Hao Y, Liu Y, Wu L, Liang H, Ni L, Wang F, Wang S, Duan Y, Xu Q, Xiao J, Yang D, Gao G, Ding Y, Gao C, Xiao J, Zhao H. Supervised machine learning algorithms to predict the duration and risk of long-term hospitalization in HIV-infected individuals: a retrospective study. Front Public Health 2024; 11:1282324. [PMID: 38249414 PMCID: PMC10796994 DOI: 10.3389/fpubh.2023.1282324] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 08/24/2023] [Accepted: 12/13/2023] [Indexed: 01/23/2024] Open
Abstract
Objective The study aimed to use supervised machine learning models to predict the length and risk of prolonged hospitalization in PLWHs to help physicians timely clinical intervention and avoid waste of health resources. Methods Regression models were established based on RF, KNN, SVM, and XGB to predict the length of hospital stay using RMSE, MAE, MAPE, and R2, while classification models were established based on RF, KNN, SVM, NN, and XGB to predict risk of prolonged hospital stay using accuracy, PPV, NPV, specificity, sensitivity, and kappa, and visualization evaluation based on AUROC, AUPRC, calibration curves and decision curves of all models were used for internally validation. Results In regression models, XGB model performed best in the internal validation (RMSE = 16.81, MAE = 10.39, MAPE = 0.98, R2 = 0.47) to predict the length of hospital stay, while in classification models, NN model presented good fitting and stable features and performed best in testing sets, with excellent accuracy (0.7623), PPV (0.7853), NPV (0.7092), sensitivity (0.8754), specificity (0.5882), and kappa (0.4672), and further visualization evaluation indicated that the largest AUROC (0.9779), AUPRC (0.773) and well-performed calibration curve and decision curve in the internal validation. Conclusion This study showed that XGB model was effective in predicting the length of hospital stay, while NN model was effective in predicting the risk of prolonged hospitalization in PLWH. Based on predictive models, an intelligent medical prediction system may be developed to effectively predict the length of stay and risk of HIV patients according to their medical records, which helped reduce the waste of healthcare resources.
Collapse
Affiliation(s)
- Jialu Li
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Yiwei Hao
- Division of Medical Record and Statistics, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Ying Liu
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Liang Wu
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Hongyuan Liang
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Liang Ni
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Fang Wang
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Sa Wang
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Yujiao Duan
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Qiuhua Xu
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Jinjing Xiao
- Department of Clinical Medicine, Zhengzhou University, Zhengzhou, China
| | - Di Yang
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Guiju Gao
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Yi Ding
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Chengyu Gao
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Jiang Xiao
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| | - Hongxin Zhao
- Clinical and Research Center of AIDS, Beijing Ditan Hospital, Capital Medical University, Beijing, China
| |
Collapse
|