Artificial intelligence applied to bed regulation in Rio Grande do Norte: Data analysis and application of machine learning on the "RegulaRN Leitos Gerais" platform

Abstract

Bed regulation within Brazil's National Health System (SUS) plays a crucial role in managing care for patients in need of hospitalization. In Rio Grande do Norte, Brazil, the RegulaRN Leitos Gerais platform was the information system developed to register requests for bed regulation for COVID-19 cases. However, the platform was expanded to cover a range of diseases that require hospitalization. This study explored different machine learning models in the RegulaRN database, from October 2021 to January 2024, totaling 47,056 regulations. From the data obtained, 12 features were selected from the 24 available. After that, blank and inconclusive data were removed, as well as the outcomes that had values other than discharge and death, rendering a binary classification. Data was also correlated, balanced, and divided into training and test portions for application in machine learning models. The results showed better accuracy (87.77%) and recall (87.77%) for the XGBoost model, and higher precision (87.85%) and F1-Score (87.56%) for the Random Forest and Gradient Boosting models, respectively. As for Specificity (82.94%) and ROC-AUC (82.13%), the Multilayer Perceptron with SGD optimizer obtained the highest scores. The results evidenced which models could adequately assist medical regulators during the decision-making process for bed regulation, enabling even more effective regulation and, consequently, greater availability of beds and a decrease in waiting time for patients.

Fig 1. Workflow defined for data processing and selection.

Fig 2. Presentation of the Phik correlation for RegulaRN Leitos Gerais data.

Fig 3. Comparison of the performance of the models used.

Fig 4. Feature importance of the machine learning models.

Fig 5. Feature importance of MLP models.

Fig 6. ROC curve and AUC value of all models.