Predicting Traffic Accidents and Their Injury Severities Using Machine Learning Techniques

Predicting Traffic Accidents and Their Injury Severities Using Machine Learning Techniques

H. L. Gururaj V. Janhavi U. Tanuja Fransesco Flamini B. C. Soundarya Vinayakumar Ravi

Department of Information Technology, Manipal Institute of Technology Bengaluru, Manipal Academy of Higher Education, India

2 Vidyavardhaka College of Engineering, India

IDSIA USI-SUPSI, University of Applied Sciences and Arts of Southern Switzerland, Switzerland

Center for Artificial Intelligence, Prince Mohammad Bin Fahd University, Saudi Arabia

Available online: 
| Citation



Traffic accidents are among the most censorious issues confronting the world as they cause numerous deaths, wounds and fatalities just as monetary misfortunes consistently. According to the world health organization (WHO) reports, 5,18,3626 accidents took place in India in the year 2019. Factors that contribute to these road crashes/ traffic accidents and resulting injuries include inattentive drivers, unenforced traffic laws, poor road infrastructure, driving in bad weather conditions and others. This investigation effort establishes models to select a set of influential factors and to build up a model for classifying the severity of injuries. Machine learning models can be applied to model and predict the severity of injury that occurs during road accidents. One such way is to apply unsupervised learning models such as Apriori, Apriori TID (transaction id), SFIT (set operation for frequent itemset using transaction database) and ECLAT (equivalence class clustering and bottom-up lattice traversal) which analyze the unlabeled traffic accidents dataset and determine the relationship between traffic accidents and injury. This research work is helpful for traffic departments to decrease the number of accidents and to distinguish the injury’s seriousness extensive simulations were carried out to demonstrate the unsupervised learning algorithms for predicting the injury severity of traffic accidents. Apriori algorithm predicts the patterns in 962 milliseconds, Apriori TID (transaction id) algorithm predicts the pattern in 557 milliseconds, SFIT algorithm predicts the pattern in 516 milliseconds and ECLAT algorithm predicts the pattern in 124 milliseconds. ECLAT algorithm took less time compared to all the other algorithms.


Apriori, ECLAT, machine learning, traffic accidents


[1] Zhou, Z, Attention based stack resnet for citywide traffic accident prediction. 20th IEEE International Conference on Mobile Data Management (MDM), pp. 369-370, 2019.

[2] Amundsen, Finn Harald, and Guro Ranes, Studies on traffic accidents in Norwegian road tunnels. Tunneling and Underground Space Technology, pp. 3-11, 2000.

[3] Myounghoe Kim, Seongwon Lee, Jaehyun Lim, Jongeun Choi, and Seong Gu Kang, Unexpected Collision Avoidance Driving Strategy Using Deep Reinforcement Learning. IEEE Access, pp. 17243–17252, 2020.

[4] Wang, H., Huang, Y., Khajepour, A., Rasekhipour, Y., Zhang, Y. & Cao, D., Crash Mitigation in Motion Planning for Autonomous Vehicles. IEEE Transactions on Intelligent Transportation Systems, pp. 1–11. 2020.

[5] Albatayneh, O., Moomen, M., Farid, A., & Ksaibati, K., Complementary Modeling of Gravel Road Traffic-Generated Dust Levels Using Bayesian Regularization Feedforward Neural Networks and Binary Probit Regression. International Journal of Pavement Research and Technology, 6, pp. 1-8, 2018.

[6] Fernandes, Fábio & Alves de Sousa, Ricardo & Ptak, Mariusz., Head Injury Simulation in Road Traffic Accidents. Springer, pp. 59-48, 2018.

[7] Li, G., Ma, H., Guan, T. et al., Predicting Safer Vehicle Font-End Shapes for Pedestrian Lower Limb Protection via a Numerical Optimization Framework. Int.J Automota Technol, 21, pp. 749–756, 2020.

[8] A., Fang, C., Qiu, L., Zhao and Y., Jin, Driver Risk Assessment Using Traffic Violation and Accident Data by Machine Learning Approaches. 3rd IEEE International Conference on Intelligent Transportation Engineering (ICITE), pp. 291-295, 2018.

[9] Y., He, Z., Liu, X., Zhou and B., Zhong, Analysis of Urban Traffic Accidents Features and Correlation with Traffic Congestion in Large-Scale Construction District. International Conference on Smart Grid and Electrical Automation (ICSGEA), pp. 641-644, 2017.

[10] T., Jian, L., Zhi-qiang, X., Jian-feng and G., Hong-yu, Cause Analysis and Countermeasures of fatal Traffic Accidents on Road Passenger Transportation Based on Typical Cases. 25th International Conference on Transportation Information and Safety (ICTIS), pp. 951-955, 2019.

[11] Masoumi, K., Forouzan, A., Barzegari, H., Asgari Darian, A., Rahim, F., Zohrevandi, B. & Nabi, S., Effective Factors in Severity of Traffic accident-related Traumas an Epidemiologic Study Based on the Haddon Matrix. Emergency (Tehran, Iran), pp. 78– 82, 2018.

[12] Tripathi, S., Shetty, U., Hasnain, A. & Hallikar, R., Cloud-Based Intelligent Traffic System to Implement Traffic Rules Violation Detection and Accident Detection Units. 3rd International Conference on Trends in Electronics and Informatics (ICOEI), pp. 22-26, 2019.

[13] Wadhahi, N.T.S.A., Hussain, S.M., Yosof, K.M., Hussain, S.A., & Singh, A.V.,Accidents Detection and Prevention System to reduce Traffic Hazards using IR Sensors. 2018 7th International Conference on Reliability, Infocom Technologies and Optimization (Trends and Future Directions) (ICRITO), pp. 254-257, 2018.

[14] Zhang, J., Li, Z., Pu, Z., & Xu, C., Comparing Prediction Performance for Crash Injury Severity among Various Machine Learning and Statistical Methods. IEEE Access, pp. 1–1, 2018.

[15] Guang Yu, Shauo lu, Speed and lane changing control optimization research on accident and congestion concentrated section of guang-shen freeway, IEEE Access, pp. 1-8, 2019.

[16] Galatioto, F., Catalano, M., Shaikh, N., McCormick, E., & Johnston, R., Advanced accident prediction models, and impacts assessment. IET Intelligent Transport Systems, 12(9), pp. 1131–1141, 2018.

[17] M., Zheng, et al., Traffic Accident’s Severity Prediction: A Deep-Learning ApproachBased CNN Network. IEEE Access, pp. 39897-39910, 2019.

[18] Leijdesdorff, H. A., Gillissen, S., Schipper, I. B., & Krijnen, P., Injury Pattern and Injury Severity of In-Hospital Deceased Road Traffic Accident Victims in The Netherlands: Dutch Road Traffic Accidents Fatalities. World Journal of Surgery, 6, pp. 268-272, 2020.

[19] Zong, F., Chen, X., Tang, J., Yu, P. & Wu, T., Analyzing Traffic Crash Severity With Combination of Information Entropy and Bayesian Network. IEEE Access, pp. 63288– 63302, 2019.

[20] Gao, Z., Pan, R., Yu, R. & Wang, X., Research on Automated Modeling Algorithm Using Association Rules for Traffic Accidents. IEEE International Conference on Big Data and Smart Computing (BigComp), pp. 1-6, 2019.

[21] Song, J., Wen, R. & Yan, W., Identification of Traffic Accident Clusters using Kulldorff’s Space-Time Scan Statistics. 2018 IEEE International Conference on Big Data (Big Data), pp. 1-6, 2018.

[22] Tai, W.-K., Wang, H.-C., Chiang, C.-Y., Chien, C.-Y., Lai, K. & Huang, T.-C., RTAIS: Road Traffic Accident Information System. 2018 IEEE 20th International Conference on High-Performance Computing and Communications, pp. 1-8, 2017.

[23] Wenqi, L., Dongyu, L. & Menghua, Y., A model of traffic accident prediction based on convolutional neural network. 2017 2nd IEEE International Conference on Intelligent Transportation Engineering (ICITE), pp. 67-72, 2017.

[24] Xia, X.-L., Nan, B. & Xu, C., Real-Time Traffic Accident Severity Prediction Using Data Mining Technologies. 2017 International Conference on Network and Information Systems for Computers (ICNISC), pp. 1-8, 2017.

[25] Xu, Q., Zheng, J. & Sun, C., Digital modeling analysis of urban road traffic capacity under the condition of traffic accidents. 2017 29th Chinese Control And Decision Conference (CCDC), pp. 10-12, 2017.

[26] Singh, Sanjay, Road Traffic Accidents in India: Issues and Challenges. Transportation Research Procedia. 25, pp. 4712-4723, 2017.

[27] Sanjana, K.R., S., Lavanya, and Y., Bevish Jinila, An approach on automated rescue system with intelligent traffic lights for emergency services. 2015 International Conference on Innovations in Information, Embedded, and Communication Systems (ICIIECS), pp. 1-4, 2015.

[28] E., Abdullah and A., Emam, Traffic Accidents Analyzer Using Big Data. 2015 International Conference on Computational Science and Computational Intelligence (CSCI), pp. 392-397, 2015.

[29] Alkandari, Abdulrahman Abdullah; Aljandal, Meshari, Theory of Dynamic Fuzzy Logic Traffic Light Integrated System with Accident Detection and Action. IEEE 2015 Second International Conference on Computing Technology and Information Management (ICCTIM) - Johor, pp. 62–68, 2015.

[30] Fu Huilin, Zhou Yucai, The Traffic Accident Prediction Based on Neural Network. IEEE Access, pp. 1-6, 2011.

[31] Hui, Chen; Xianghui, Wang; Xiqiang, Zhang; Shaoli, Zhang, The Evaluation of Chinese Urban Traffic Management System Application Based on Intelligent Traffic Control Technology. 7th International Conference on Intelligent Computation Technology and Automation, pp. 791–795. 2014.

[32] Li, Jikuang Yang, Otte, D., Zhou Xuenong, Zhang Chao, Yu Xiaoping, Li Weiqiang, A Study on Motor-Scooter Accidents in China and Germany. IEEE 2013 Fifth International Conference on Measuring Technology and Mechatronics Automation (ICMTMA 2013), pp. 110–113, 2013.

[33] Sherif, Hossam M.; Shedid, M. Amer; Senbel, Samah A., Real-time traffic accident detection system using a wireless sensor network. IEEE 2014 6th International Conference of Soft Computing and Pattern Recognition (SoCPaR), pp. 59–64, 2014.

[34] Xie, Fang, Feng, Qingdong, Research of effects of the accident on traffic flow characteristics. IEEE 2013 International Conference on Mechatronic Sciences, Electric Engineering and Computer (MEC) - Shengyang, China Proceedings 2013 International Conference on Mechatronic Sciences, pp. 1670–1674, 2013.