Predicting Patient Triage at the Emergency Department using Machine Learning Classification: The Case Study of UNS Hospital, Indonesia
Received: 16 March 2025 | Revised: 16 April 2025 | Accepted: 19 April 2025 | Online: 4 June 2025
Corresponding author: Pringgo Widyo Laksono
Abstract
The triage of patients in the Emergency Department (ED) plays a critical role in determining the urgency and type of treatment that shall be administered. Therefore, an accurate prediction system for patient triage can be very helpful. This study aims to develop a Machine Learning (ML) classification model to predict triage decisions for patients admitted to the ED at UNS Hospital. Several classification models were evaluated, including Naïve Bayes (NB), Decision Trees (DT), Random Forest (RF), Support Vector Machines (SVM), and k-Nearest Neighbors (k-NN). To assess model performance, multiple metrics were employed: accuracy, precision, recall, F1-score, and the confusion matrix. Following an initial evaluation, hyperparameter tuning was conducted on the selected best-performing model to identify the optimal combination of parameters for improved predictive performance. RF emerged as the best-performing model for triage prediction. The parameters tuned included the number of estimators, criterion, maximum features, and maximum depth, using a 5-fold cross-validation strategy. The optimal parameter values were found to be 900 estimators, 'entropy' for the criterion, 'log2' for max features, and a maximum depth of 40. The results of this study demonstrate that hyperparameter tuning can significantly enhance model performance, reducing recall errors, improving the F1-score, and decreasing the number of mispredictions.
Keywords:
emergency department, triage, machine learning, classificationDownloads
References
I. D. A. Indonesia, "Mengenal Instalasi Gawat Darurat (IGD) dan Pediatric Intensive Care Unit (PICU) di Rumah Sakit." [Online]. Available: https://www.idai.or.id/artikel/seputar-kesehatan-anak/mengenal-instalasi-gawat-darurat-igd-dan-pediatric-intensive-care-unit-picu-di-rumah-sakit.
E. Foley and A. T. Reisner, "Triage," in Ciottone’s Disaster Medicine, Elsevier, 2016, pp. 337–343.
K. B. Ahsan, M. A. Karim, G. J. FitzGerald, D. G. Morel, and J. A. Burke, "Development of Relationship between Triaging of Patients and Emergency Department Performance," Procedia Manufacturing, vol. 30, pp. 200–207, 2019.
G. Büchi, M. Cugno, and R. Castagnoli, "Smart factory performance and Industry 4.0," Technological Forecasting and Social Change, vol. 150, Jan. 2020, Art. no. 119790.
E. Arnaud, M. Elbattah, C. Ammirati, G. Dequen, and D. A. Ghazali, "Use of Artificial Intelligence to Manage Patient Flow in Emergency Department during the COVID-19 Pandemic: A Prospective, Single-Center Study," International Journal of Environmental Research and Public Health, vol. 19, no. 15, Aug. 2022, Art. no. 9667.
S. Boussen et al., "Triage and monitoring of COVID-19 patients in intensive care using unsupervised machine learning," Computers in Biology and Medicine, vol. 142, Mar. 2022, Art. no. 105192.
H. Jiang et al., "Machine learning-based models to support decision-making in emergency department triage for patients with suspected cardiovascular disease," International Journal of Medical Informatics, vol. 145, Jan. 2021, Art. no. 104326.
B. H. Bhavani and N. C. Naveen, "An Approach to Determine and Categorize Mental Health Condition using Machine Learning and Deep Learning Models," Engineering, Technology & Applied Science Research, vol. 14, no. 2, pp. 13780–13786, Apr. 2024.
B. P. Roquette, H. Nagano, E. C. Marujo, and A. C. Maiorano, "Prediction of admission in pediatric emergency department with deep neural networks and triage textual data," Neural Networks, vol. 126, pp. 170–177, Jun. 2020.
S. Morreel et al., "Triaging and referring in adjacent general and emergency departments (the TRIAGE trial): A cluster randomised controlled trial," PLOS ONE, vol. 16, no. 11, Nov. 2021, Art. no. e0258561.
F. Gao, B. Boukebous, M. Pozzar, E. Alaoui, B. Sano, and S. Bayat-Makoei, "Predictive Models for Emergency Department Triage using Machine Learning: A Systematic Review," Obstetrics and Gynecology Research, vol. 05, no. 02, 2022.
O. H. Salman, Z. Taha, M. Q. Alsabah, Y. S. Hussein, A. S. Mohammed, and M. Aal-Nouman, "A review on utilizing machine learning technology in the fields of electronic emergency triage and patient priority systems in telemedicine: Coherent taxonomy, motivations, open research challenges and recommendations for intelligent future work," Computer Methods and Programs in Biomedicine, vol. 209, Sep. 2021, Art. no. 106357.
V. Kalra and R. Aggarwal, "Importance of Text Data Preprocessing & Implementation in RapidMiner," presented at the The First International Conference on Information Technology and Knowledge Management, Jan. 2018, pp. 71–75.
S. Walfish, "A review of statistical outlier methods," Pharmaceutical Technology, vol. 30, no. 11, pp. 82–86, Nov. 2006.
R. J. May, H. R. Maier, and G. C. Dandy, "Data splitting for artificial neural networks using SOM-based stratified sampling," Neural Networks, vol. 23, no. 2, pp. 283–294, Mar. 2010.
S. Mishra, "Handling Imbalanced Data : SMOTE vs. Random Undersampling," International Research Journal of Engineering and Technology (IRJET), vol. 4, no. 8, pp. 317–320, Aug. 2017.
M. Schuld, R. Sweke, and J. J. Meyer, "Effect of data encoding on the expressive power of variational quantum-machine-learning models," Physical Review A, vol. 103, no. 3, Mar. 2021, Art. no. 032430.
R. G. Mantovani, A. L. D. Rossi, J. Vanschoren, B. Bischl, and A. C. P. L. F. Carvalho, "To tune or not to tune: Recommending when to adjust SVM hyper-parameters via meta-learning," in 2015 International Joint Conference on Neural Networks (IJCNN), Killarney, Ireland, Jul. 2015, pp. 1–8.
G. S. K. Ranjan, A. Kumar Verma, and S. Radhika, "K-Nearest Neighbors and Grid Search CV Based Real Time Fault Monitoring System for Industries," in 2019 IEEE 5th International Conference for Convergence in Technology (I2CT), Bombay, India, Mar. 2019, pp. 1–5.
Downloads
How to Cite
License
Copyright (c) 2025 Pringgo Widyo Laksono, Rizki Ananda Putra Nur Rohmat, Retno Wulan Damayanti, Eko Pujiyanto, Cucuk Nur Rosyidi
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.
