Predicting the length-of-stay of pediatric patients using machine learning algorithms
研究用随机森林、支持向量回归等机器学习算法预测儿科患者住院时长，帮助医院优化资源分配。随机森林模型R²为65.67%，平均绝对误差3.51天。
The management of hospitals’ resource capacity has a strong impact on the quality of care, and the length-of-stay (LOS) of patients is an indicator that reflects its efficiency and effectiveness. This study aims at predicting the LOS of pediatric patients (LOS-P) in hospitals to assist in decision-making regarding resource utilisation. LOS-P forecasting presents additional challenges to the analyst compared to other medical specialties since Pediatrics comprises several other subspecialties (e.g. pediatric oncology and traumatology). Pediatric patients within subspecialties compete for the same hospital resources, and aggregate LOS-P predictions are more useful for resource planning. However, aggregate pediatric LOS datasets are harder to model and result in lower forecasting accuracy. To address that problem, we propose a forecasting model based on Machine Learning algorithms. The method for LOS-P forecasting comprises five steps (data visualisation, data pre-processing, sample partitioning, model testing, and model definition through parameter setting and variable selection) and is tested using a dataset of hospitalisations of pediatric patients from a large Brazilian University hospital. Multiple linear regression, random forest, support vector regression, ridge regression, and partial least squares algorithms are applied and compared to determine the best forecasting model. Results indicate that all forecasting models yield satisfactory accuracy, with the best algorithms being random forest and support vector regressor. After refining the model through variable selection and using a Grid Search to find the best parameters, the random forest algorithm yielded an R2 of 65.67%, with an average absolute error of 3.51 days.Highlights Prediction of the length of stay of pediatric patients (LOS-P) in hospitals based on Machine Learning algorithmsMultiple linear regression, random forest, support vector regression, ridge regression, and partial least squares algorithms were applied and comparedRandom forest algorithm yielded an R2 of 65.67%, with an average absolute error of 3.51 days