Traffic fatalities prediction based on support vector machine


  • Ting Li Dalian Maritime University, Transportation Management College, Dalian, PR China Author
  • Yunong Yang Dalian Maritime University, Transportation Management College, Dalian, PR China Author
  • Yonghui Wang Dalian Maritime University, Transportation Management College, Dalian, PR China Author
  • Chao Chen Dalian University of Technology, Automotive Engineering College, Dalian, PR China Author
  • Jinbao Yao Beijing Jiaotong University, School of Civil Engineering and Architecture, Beijing, PR China Author



traffic accident, support vector machine, SVM, particle swarm optimization (PSO), PSO, prediction model, optimal parameters


To effectively predict traffic fatalities and promote the friendly development of transportation, a prediction model of traffic fatalities is established based on support vector machine (SVM). As the prediction accuracy of SVM largely depends on the selection of parameters, Particle Swarm Optimization (PSO) is introduced to find the optimal parameters. In this paper, small sample and nonlinear data are used to predict fatalities of traffic accident. Traffic accident statistics data of China from 1981 to 2012 are chosen as experimental data. The input variables for predicting accident are highway mileage, vehicle number and population size while the output variables are traffic fatality. To verify the validity of the proposed prediction method, the back-propagation neural network (BPNN) prediction model and SVM prediction model are also used to predict the traffic fatalities. The results show that compared with BPNN prediction model and SVM model, the prediction model of traffic fatalities based on PSO-SVM has higher prediction precision and smaller errors. The model can be more effective to forecast the traffic fatalities. And the method using particle swarm optimization algorithm for parameter optimization of SVM is feasible and effective. In addition, this method avoids overcomes the problem of “over learning” in neural network training progress.


ABDEL-ATY, M., HALEEM, K., 2011. Analyzing angle crashes at unsignalized intersections using machine learning techniques. Accident Analysis & Prevention, 43(1), pp. 461-470.

ABDELWAHAB, H. T., ABDEL-ATY, M. A., 2001. Development of artificial neural network models to predict driver injury severity in traffic accidents at signalized intersections. Transportation Research Record: Journal of the Transportation Research Board, 1746(1), pp. 6-13.

CAO, C., XU, J., 2007. Short-term traffic flow predication based on PSO-SVM. Proceedings of the first international conference on transportation engineering, 246, p. 28.

CAO, L. J., FRANCIS, E. H., 2003. Support vector machine with adaptive parameters in financial time series forecasting. IEEE Transactions on Neural Networks, 14(6).

CHANG, L. Y., 2005. Analysis of freeway accident frequencies: negative binomial regression versus artificial neural network. Safety science, 43(8), pp. 541-557.

CLARKE, D. D., FORSYTH, R., WRIGHT, R., 1998. Machine learning in road accident research: decision trees describing road accidents during cross-flow turns. Ergonomics, 41(7), pp. 1060-1079.

DEUBLEIN, M., SCHUBERT, M., ADEY, B. T., KÖHLER, J., FABER M. H., 2013. Prediction of road accidents: A Bayesian hierarchical approach. Accident Analysis & Prevention, 51, pp. 274-291.

DONG, B., CAO, C., LEE, S. E., 2005. Applying support vector machines to predict building energy consumption in tropical region. Energy and Buildings, 37, pp. 545-553.

GAN, X. S.,DUANMU, J., CONG, W. 2010. Fatalness assessment of flight safety hidden danger based on support vector machine. Journal of Safety Science and Technology, 6(3), pp. 206-210.

GHASEMLOU, K., AYDI, M. M., YILDIRIM, M. S., 2015. Prediction of pedal cyclists and pedestrian fatalities from total monthly accidents and registered private car numbers. Archives of Transport, 2015, 34(2), pp. 29-35.

GUAN, Z., SONG, T, QI, L., 2008. An Application of Support Vector Machine in Foundation Settlement Prediction [J]. Transactions of Shenyang Ligong University, 2, p. 024.

KENNEDY, J., EBERHART, R. C., 1995 Particle swarm optimization, Proc. of IEEE International Conference on Neural Networks, Piscataway, NJ, pp. 1942-1948.

KUNT, M. M., AGHAYAN, I., NOII, N., 2011. Prediction for traffic accident severity: comparing the artificial neural network, genetic algorithm, combined genetic algorithm and pattern search methods. Transport, 26(4), pp. 353-366.

LEE, Y., WEI, C. H., 2010. A computerized feature selection method using genetic algorithms to forecast freeway accident duration times. Computer-Aided Civil and Infrastructure Engineering, 25(2), pp. 132-148.

LI, X., LORD, D., ZHANG, Y., XIE, Y. 2008. Predicting motor vehicle crashes using support vector machine models. Accident Analysis & Prevention, 40(4), pp. 1611-1618.

LI, Z., LIU, P., WANG, W., XIE, Y., 2012. Using support vector machine models for crash injury severity analysis. Accident Analysis & Prevention, 45, pp. 478-486.

MITAS, A. W., KONIOR, W., KONIOR A., 2013. VMS parameters impact on safety and reliability in Road traffic management. Archives of Transport, 25-26(1-2), pp. 69-83.

NASSIRI, H., NAJAF, P., AMIRI, A. M., 2014. Prediction of Roadway Accident Frequencies: Count Regressions versus Machine Learning Models. Scientia Iranica, 21(2), pp. 263-275.

NOWAKOWSKA, M., 2012. Road Traffic Accident Patterns: A Conceptual Grouping Approach to Evaluate Crash Clusters. Archives of Transport, 2012, 24(1), pp. 73-98.

PAMUŁA, T., 2012. Classification and Prediction of Traffic Flow Based on Real Data Using Neural Networks. Archives of Transport, 2012, 24(4), pp. 519-529.

POCH, M., 1996. Mannering, F. Negative binomial analysis of intersection-accident frequencies. Journal of Transportation Engineering, 122(2), pp. 105-113.

QUEK, C., PASQUIER, M., LIM, B. B. S., 2006. POP-TRAFFIC: A novel fuzzy neural approach to road traffic analysis and prediction. IEEE Transactions on Intelligent Transportation Systems, 7(2), pp. 133-146.

RAMANI, R. G., SELVARAJ, S. A., 2014. Pragmatic Approach for Refined Feature Selection for the Prediction of Road Accident Severity. Studies in Informatics and Control, 23(1), pp. 41-52.

STELMACH, A., 2012. Neural Model of the Aircraft Landing Phase. Archives of Transport, 24(2), pp. 249-258.

TESEMA, T. B., ABRAHAM, A., GROSAN, C., 2005. Rule mining and classification of road traffic accidents using adaptive regression trees. International Journal of Simulation, 6(10), pp. 80-94.

VAPNIK, V. N., 1999. An overview of statistical learning theory. IEEE Transactions on Neural Networks, 10(5), pp. 988-999.

XIE, Y., LORD, D., ZHANG, Y., 2007. Predicting motor vehicle collisions using Bayesian neural network models: An empirical analysis. Accident Analysis & Prevention, 39(5), pp. 922-933.

XU, C., WANG, W., LIU, P., 2013. A genetic programming model for real-time crash prediction on freeways. IEEE Transactions on

Intelligent Transportation Systems, 14(2), pp. 574-586.

YANG, J., ZHAO, J., 2013. Road Traffic Safety Prediction Based on Improved SVM. ICTE 2013, pp. 107-114.

YAO, B. Z., HU, P., ZHANG, M. H., JIN, M. Q., 2014. A Support Vector Machine with the Tabu Search Algorithm For Freeway Incident Detection. International Journal of Applied Mathematics and Computer Science, 24(2), pp. 397-404.

YAO, B. Z., YAO, J. B., ZHANG, M. H., YU, L., 2014. Improved support vector machine regression in multi-step-ahead prediction for rock displacement surrounding a tunnel. Scientia iranica, 21(4), pp. 1309-1316.

YASDI, R., 1991. Prediction of road traffic using a neural network approach. Neural computing & applications, 8(2), pp. 135-142.

YU, B., YANG, Z. Z., SUN, X. S., YAO, B. Z., ZENG, Q. C., JEPPESEN, E., 2011. Parallel Genetic Algorithm in Bus Route Headway Optimization. Applied Soft Computing, 11(8), pp. 5081-5091.

YU, B., YANG, Z. Z., YAO, J. B., 2010. Genetic Algorithm For Bus Frequency Optimization. Journal of Transportation Engineering, 136(6), pp. 576-583.

YU, B., ZHU, H. B.,CAI, W. J., MA, N., YAO, B. Z., 2013. Two-phase Optimization Approach to Transit Hub Location--the Case of Dalian. Journal of Transport Geography, 33, pp. 62-71.

ZHANG, C., YAN, X., MA, L.,AN, M., 2014. Crash Prediction and Risk Evaluation Based on Traffic Analysis Zones. Mathematical Problems in Engineering, 2014(2014), pp. 9.

ZONG, F., JIA, H. F., PAN, X., WU, Y., 2013b. Prediction of Commuter’s Daily Time Allocation. PROMET - Traffic&Transportation, 25(5), pp. 445-455.

ZONG, F., XU, H. G., ZHANG, H. Y., 2013a. Prediction for Traffic Accident Severity: Comparing the Bayesian Network and Regression Models. Mathematical Problems in Engineering, 2013(2013), pp. 9.






Original articles

How to Cite

Li, T., Yang, Y., Wang, Y., Chen, C., & Yao, J. (2016). Traffic fatalities prediction based on support vector machine. Archives of Transport, 39(3), 21-30.


Most read articles by the same author(s)

1 2 3 4 5 6 7 8 9 10 > >> 

Similar Articles

1-10 of 196

You may also start an advanced similarity search for this article.