Welcome to Francis Academic Press

International Journal of New Developments in Engineering and Society, 2022, 6(3); doi: 10.25236/IJNDES.2022.060310.

Short-Term Traffic Flow Prediction Method Based on WT-IGWO-ELM


Shuilin Li1, Luyao Niu2

Corresponding Author:
Shuilin Li

1International Institute of Technology, Changsha University of Science & Technology, Changsha, 410114, China

2School of Traffic and Transportation, Beijing Jiaotong University, Beijing, 100044, China


Accurate prediction of short-term traffic flow provides crucial data support for the stable operation of intelligent transportation systems. For this issue, this paper proposes a short-term traffic flow prediction method based on WT-IGWO-ELM. The algorithm uses the Wavelet Transform (WT) method to denoise the traffic flow data in advance, which improves the data quality of the dataset. Then, the IGWO algorithm, which integrates the initial population based on Sine chaotic map and reverse learning strategy, the adjustment of nonlinear convergence factor and the introduction of dynamic weights, is used to avoid local optimality more effectively, speed up the convergence speed, and improve the solution accuracy. Finally, the improved grey wolf optimizer (IGWO) was used to update the optimal parameters of the ELM prediction model, and the average relative error of the prediction of the WT-IGWO-ELM model was verified by comparison experiments compared with those of ELM and WT. -ELM, GWO-ELM, WT-GWO-ELM and IGWO-ELM decreased by 96.6625%, 95.5972%, 87.9447%, 79.5021%, 72.0571%, respectively, and its prediction effect was much better than ELM, WT-ELM, GWO-ELM, WT-GWO-ELM and IGWO-ELM methods have high prediction performance and accuracy in short-term traffic flow prediction.


Short-term traffic flow prediction; Wavelet Transform; Improved Grey Wolf Optimizer; Extreme Learning Machine

Cite This Paper

Shuilin Li, Luyao Niu. Short-Term Traffic Flow Prediction Method Based on WT-IGWO-ELM. International Journal of New Developments in Engineering and Society (2022) Vol.6, Issue 3: 55-62. https://doi.org/10.25236/IJNDES.2022.060310.


[1] Daraghmiya, Yicw, Chiangtc. Negative binomial additive models for short-term traffic flow forecasting in urban areas. .IEEE Transactions on Intelligent Transportation Systems, 2014,15(2):784-793.

[2] Song Chi, Shen Guojiang. Review of short-term traffic flow forecasting model. Automation Expo, 2012, 29(06):84-87.

[3] Liu Haihong, Zhou Chunmei. A review of short-term traffic flow prediction methods. Journal of Wuhu Vocational and Technical College, 2011,13(04):31-33. 

[4] Yang Fengman. A review of traffic flow prediction methods based on artificial neural network. Highway Transportation Science and Technology, 2020,37(S1):130-135.

[5] Huang Darong, Song Jun, Wang Da, Cao Jianqiu, Li Wei. Research on Traffic Flow Prediction Model Based on ARMA and Wavelet Transform. Computer Engineering and Application, 2006(36):191-194+224. 

[6] Mirjalili S, Mirjalili S M, Lewis A . Grey Wolf Optimizer. Advances in Engineering Software, 2014, 69(3):46–61. 

[7] Zhang Yang, Zhou Xizhao. Improved Grey Wolf Algorithm for Global Optimization Problems. Journal of University of Shanghai for Science and Technology, 2021,43(1):73-82. 

[8] Ni Jing, Qin Bin, Zeng Fanlong. An Improved Grey Wolf Optimization Algorithm with Mixed Strategies . Software Guide, 2021,20(05):72-76. 

[9] CHO K,MIYANO T. Chaotic cryptography using augmented Lorenz equations aided by quantum key distribution [J/OL]. IEEE Transactions on Circuits and Systems I: Regular Papers,2015,62 ( 2 ):478-487.

[10] Tan Fafang, Zhao Junjie, Wang Qi. Research on a Gray Wolf Optimization Algorithm with Improved Nonlinear Convergence Method. Microelectronics and Computers, 2019, 36(5): 89-95. 

[11] Guo Zhenzhou, Liu Ran, Gong Changqing, et al. Improvement research based on gray wolf algorithm. Computer Application Research, 2017, 34(12): 3603-3606+3610. 

[12] Wang Qiuping, Wang Mengna, Wang Xiaofeng. Grey wolf optimization algorithm with improved convergence factor and proportional weight. Computer Engineering and Applications, 2019, 55(21): 60–65, 98. 

[13] Zhou Borong, Chen Weiguo, Xu Zhenyi, Wen Xiulan. Research on short-term traffic flow prediction model based on improved online extreme learning machine. Computer Engineering and Science, 2022, 44(05): 944-950.

[14] Rui Lanlan, Li Qinming. Short-term traffic flow prediction algorithm based on combined model. Journal of Electronics and Information, 2016, 38(05): 1227-1233.

[15] Ganlu. Research and application of extreme learning machine [D]. Xidian University, 2014.

[16] Wang Jie, Bi Haoyang. An extreme learning machine based on particle swarm optimization. Journal of Zhengzhou University (Science Edition), 2013, 45(01): 100-104.