Communications - Scientific letters of the University of Zilina X:X | DOI: 10.26552/com.C.2026.039
Optuna-Optimized Machine Learning for Traffic Accident Severity Prediction in Jordan under Class Imbalance Conditions
- Al-Zaytoonah University of Jordan, Department of Civil and Infrastructure Engineering, Amman, Jordan
A machine learning framework for predicting the traffic accident severity under class imbalance conditions is predented in Amman, Jordan. The methodology began with a preprocessing pipeline consisting of IQR-based (Interquartile range) outlier removal, label encoding, feature scaling, and Synthetic Minority Over-sampling Technique (SMOTE). Sixteen model configurations were considered, comprising four ensemble learning algorithms (XGBoost, LightGBM, Random Forest, CatBoost) under oversampling, with Optuna-based Bayesian hyperparameter optimisation. The result indicated that the class balancing using SMOTE had a greater influence on the predictive performance than hyperparameter optimisation, specifically for CatBoost model. Feature importance showed that vehicles involved, collision type, and speed are the strongest predictors of accident severity.
Keywords: hyperparameter tuning, SMOTE class imbalance, traffic accident severity, machine learning, Optuna
Grants and funding:
The authors received no financial support for the research, authorship and/or publication of this article.
Conflicts of interest:
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Received: April 19, 2026; Accepted: June 15, 2026; Prepublished online: July 3, 2026
References
- GHADI, M., TOROK, A., TANCZOS, K. Study of the economic cost of road accidents in Jordan. Periodica Polytechnica Transportation Engineering [online]. 2018, 46(3), p. 129-134. ISSN 0303-7800, eISSN 1587-3811. Available from: https://doi.org/10.3311/PPtr.10392
Go to original source... - KHAWALDAH, H., ALZBOUN, N. Socio-economic and environmental impacts of Syrian Refugees in Jordan: a Jordanians' perspective. Heliyon [online]. 2022, 8(8), e10005. eISSN 2405-8440. Available from: https://doi.org/10.1016/j.heliyon.2022.e10005
Go to original source... - GHADI, M. Investigating the asymmetric effects of economic and demographic factors on traffic accident severity - evidence from a nonlinear ARDL approach. Promet - Traffic and Transportation [online]. 2025, 37(3), p. 648-664. ISSN 0353-5320, eISSN 1848-4069. Available from: https://doi.org/10.7307/ptt.v37i3.814
Go to original source... - GHADI, M. Q. Multilevel analysis of road accident frequency: the impact of the road category. International Journal of Transport Development and Integration [online]. 2023, 7(2), p. 123-130. ISSN 2058-8305, eISSN 2058-8313. Available from: https://doi.org/10.18280/ijtdi.070207
Go to original source... - ABELLAN, J., LOPEZ, G., DE ONA, J. Analysis of traffic accident severity using Decision Rules via Decision Trees. Expert Systems with Applications [online]. 2013, 40(15), p. 6047-6054. ISSN 0957-4174, eISSN 1873-6793. Available from: https://doi.org/10.1016/j.eswa.2013.05.027
Go to original source... - GHADI, M. Q., TOROK, A. Evaluation of the impact of spatial and environmental accident factors on severity patterns of road segments. Periodica Polytechnica Transportation Engineering [online]. 2020, 49(2), p. 146-155. ISSN 0303-7800, eISSN 1587-3811. Available from: https://doi.org/10.3311/PPtr.14692
Go to original source... - JIMA, D., SIPOS, T. The impact of road geometric formation on traffic crash and its severity level. Sustainability [online]. 2022, 14(14), 8475. eISSN 2071-1050. Available from: https://doi.org/10.3390/su14148475
Go to original source... - ISLAM, M. H., TEIK HUA, L., HAMID, H., AZARKERDAR, A. Relationship of accident rates and road geometric design. IOP Conference Series: Earth and Environmental Science [online]. 2019, 357, 012040. ISSN 1755-1315. Available from: https://doi.org/10.1088/1755-1315/357/1/012040
Go to original source... - GORZELANCZYK, P., GROBELNY, P. An estimate of the number of accidents on polish highways based on the kind of road. Communications - Scientific Letters of the University of Zilina [online]. 2024, 26(1), p. F13-22. ISSN 1335-4205, eISSN 2585-7878. Available from: https://doi.org/10.26552/com.C.2024.012
Go to original source... - WU, W., JIANG, S., LIU, R., JIN, W., MA, C. Economic development, demographic characteristics, road network and traffic accidents in Zhongshan, China: gradient boosting decision tree model. Transportmetrica A: Transport Science [online]. 2020, 16(3), p. 359-387. ISSN 2324-9935, eISSN 2324-9943. Available from: https://doi.org/10.1080/23249935.2020.1711543
Go to original source... - SOHAEE, N., BOHLULI, S. Nonlinear analysis of the effects of socioeconomic, demographic, and technological factors on the number of fatal traffic accidents. Safety [online]. 2024, 10(1), 11. eISSN 2313-576X. Available from: https://doi.org/10.3390/safety10010011
Go to original source... - GHADI, M. Q. Investigating the nexus between traffic accidents and tourism in Jordan: evidence from the ARDL approach. Journal of Engineering, Project, and Production Management [online]. 2025, 15(3), 0021. ISSN 2221-6529, eISSN 2223-8379. Available from: https://doi.org/10.32738/JEPPM-2025-0021
Go to original source... - GHADI, M. Q. Investigating the impact of climate change on traffic accidents in Jordan. Sustainability [online]. 2025, 17(5), 2161. eISSN 2071-1050. Available from: https://doi.org/10.3390/su17052161
Go to original source... - AL-MASAEID, H. R., BAZLAMIT, S. M., AL-ZEDANEEN, A. E., AL-MOFLEH, H. Impact of fasting on traffic accidents. 2020.
- BHUIYAN, H., ARA, J., HASIB, K. M., SOURAV, M. I. H., KARIM, F. B., SIK-LANYI, C., GOVERNATORI, G., RAKOTONIRAINY, A., YASMIN, S. Crash severity analysis and risk factors identification based on an alternate data source: a case study of developing country. Sci Rep [online]. 2022, 12, 21243. eISSN 2045-2322. Available from: https://doi.org/10.1038/s41598-022-25361-5
Go to original source... - SANTOS, K., DIAS, J. P., AMADO, C. A literature review of machine learning algorithms for crash injury severity prediction. Journal of Safety Research [online]. 2022, 80, p. 254-269. ISSN 0022-4375, eISSN 1879-1247. Available from: https://doi.org/10.1016/j.jsr.2021.12.007
Go to original source... - STANTON, N. (Ed.). Advances in human aspects of transportation [online]. Vol. 786. Cham: Springer International Publishing, 2019. ISSN 2194-5357, eISSN 2194-5365, ISBN 978-3-319-93884-4. Available from: https://doi.org/10.1007/978-3-319-93885-1
Go to original source... - ALPALHAO, N., SARMENTO, P., JARDIM, B., DE CASTRO NETO, M. Assessing the risk of traffic accidents in Lisbon using a gradient boosting algorithm with a hybrid classification/regression approach. Transportation Research Interdisciplinary Perspectives [online]. 2025, 32, 101495. eISSN 2590-1982. Available from: https://doi.org/10.1016/j.trip.2025.101495
Go to original source... - YU, B., WANG, Y. T., YAO, J. B., WANG, J. Y. A comparison of the performance of ann and svm for the prediction of traffic accident duration. Neural Network World [online]. 2016, 26, p. 271-287. ISSN 1210-0552, eISSN 2336-4335. Available from: https://doi.org/10.14311/NNW.2016.26.015
Go to original source... - FARHART, Z., KAROUNI, A., DAYA, B., CHAUVET, P., HMADEH, N. Traffic accidents severity prediction using support vector machine models. International Journal of Innovative Technology and Exploring Engineering [online]. 2020, 9(7), p. 1345-1350. eISSN 2278-3075. Available from: https://doi.org/10.35940/ijitee.F4393.059720
Go to original source... - KHALEEL, A., FARFOURA, M. E., ALQABANI, H., BARHAM, K. Y. A Hybrid learning neuro-shallow architecture obesity detection for Syrian refugees in Jordan. In: 2025 1st International Conference on Computational Intelligence Approaches and Applications ICCIAA: proceedings [online]. IEEE. 2025. ISBN 979-8-3315-2366-4, eISBN 979-8-3315-2365-7, p. 1-6. Available from: https://doi.org/10.1109/ICCIAA65327.2025.11013717
Go to original source... - ABDULJAWAD, M. Comparative analysis of machine learning algorithms for intrusion detection in IoT networks. International Journal of Advances in Soft Computing and its Applications [online]. 2025, 17(3), p. 259-274. ISSN 2710-1274, eISSN 2074-8523. Available from: https://doi.org/10.15849/IJASCA.251130.15
Go to original source... - JAMAL, A., ZAHID, M., TAUHIDUR RAHMAN, M., AL-AHMADI, H. M., ALMOSHAOGEH, M., FAROOQ, D., AHMAD, M. Injury severity prediction of traffic crashes with ensemble machine learning techniques: a comparative study. International Journal of Injury Control and Safety Promotion [online]. 2021, 28(4), p. 408-427. ISSN 1745-7300, eISSN 1745-7319. Available from: https://doi.org/10.1080/17457300.2021.1928233
Go to original source... - LIANG, Y., YUAN, H., WANG, Z., WAN, Z., LIU, T., WU, B., CHEN, S., TANG, X. Nonlinear effects of traffic statuses and road geometries on highway traffic accident severity: a machine learning approach. PLoS One [online]. 2024, 19, e0314133. eISSN 1932-6203. Available from: https://doi.org/10.1371/journal.pone.0314133
Go to original source... - BOO, Y., CHOI, Y. Comparison of mortality prediction models for road traffic accidents: an ensemble technique for imbalanced data. BMC Public Health [online]. 2022, 22, 1476. eISSN 1471-2458. Available from: https://doi.org/10.1186/s12889-022-13719-3
Go to original source... - CHEN, F., LIU, X. Q., YANG, J. J., LIU, X. K., MA, J. H., CHEN, J., XIAO, H. Y. Traffic accident severity prediction based on an enhanced MSCPO-XGBoost hybrid model. Sci Rep [online]. 2025, 15, 25729. eISSN 2045-2322. Available from: https://doi.org/10.1038/s41598-025-00797-7
Go to original source... - SUDHESHNA, P., ALJAWAHRY, H. M. M., PHADKE, R., SIVASUBRAMANIAN, S., KANTH TATIRAJU V. R. An effective prediction modelling of traffic accidents based on categorical boosting with grid search. In: 2024 International Conference on Integrated Intelligence and Communication Systems ICIICS: proceedings [online]. IEEE. 2024. ISBN 979-8-3315-0497-7, eISBN 979-8-3315-0496-0, p. 1-5. Available from: https://doi.org/10.1109/ICIICS63763.2024.10859490
Go to original source... - EL FEROUALI, S., ELAMRANI ABOU ELASSAD, Z., QASSIMI, S., ABDALI, A. From baseline to best practice: an advanced feature selection, feature resampling and grid search techniques to improve injury severity prediction. Applied Artificial Intelligence [online]. 2025, 39(1). ISSN 0883-9514, eISSN 1087-6545. Available from: https://doi.org/10.1080/08839514.2025.2452675
Go to original source... - Traffic accidents in Jordan - Jordan Traffic Institute [online] [accessed 2024-08-02]. 2024. Available from: https://www.psd.gov.jo/ar-jo
- AWAD, M., MOUSSA, G. S., WAHABALLA, A. M., YOUNES, H. A state-of-the-art review of injury severity analysis in traffic crashes: toward a generalized modeling framework. Innovative Infrastructure Solutions [online]. 2026, 11, 19. ISSN 2364-4176, eISSN 2364-4184. Available from: https://doi.org/10.1007/s41062-025-02409-9
Go to original source... - CHAKRABORTY, M., GATES, T. J., SINHA, S. Causal analysis and classification of traffic crash injury severity using machine learning algorithms. Data Science for Transportation [online]. 2023, 5, 12. ISSN 2948-135X, eISSN 2948-1368. Available from: https://doi.org/10.1007/s42421-023-00076-9
Go to original source... - AARTS, L., COMMANDEUR, J. J. F., WELSH, R., NIESEN, S., LERNER, M., HOMAS, P., BOS, N., DAVIDSE, R. J. Study on serious road traffic injuries in the EU. 2016.
- CHAWLA, N. V., BOWYER, K. W., HALL, L. O., KEGELMEYER, W. P. SMOTE: Synthetic minority over-sampling technique. Journal of Artificial Intelligence Research [online]. 2002, 16, p. 321-357. ISSN 1076-9757. Available from: https://doi.org/10.1613/jair.953
Go to original source...
This is an open access article distributed under the terms of the Creative Commons Attribution 4.0 International License (CC BY 4.0), which permits use, distribution, and reproduction in any medium, provided the original publication is properly cited. No use, distribution or reproduction is permitted which does not comply with these terms.

ORCID...