A short-term traffic flow prediction network using filter-genetic feature selection method

Abstract

Accurate selection of spatial-temporal features is key to the short-term traffic flow prediction model outputting higher quality results, which can effectively reduce the difficulty of constructing the prediction model. The spatial-temporal feature selection of most existing short-term traffic flow prediction models mainly relies on empirical knowledge methods and lacks interpretability. The proposed short-term traffic flow prediction network, named STFP-FGFS, utilizes a filter-genetic feature selection method to better explain the results of short-term traffic flow predictions. It consists of three stages: initial generation of temporal-spatial feature set, filtering, and feature optimization, as well as the predicted model. The initial spatial features are generated based on effective travel time, target time granularity, and vehicle type; that is, original spatial features are replaced by standardized spatial features. Four widely used feature selection methods for short-term traffic flow prediction are applied and compared, evaluating three experimental targets and four types of time granularity using four evaluation indexes. The results show that the STFP-FGFS proposed method has overall superior performance, good interpretability, and readability for selected spatial-temporal features. 

Keywords

Feature selection / Short-term traffic flow prediction / Artificial neural network / GRU

References

[1] Veres M, Moussa M. Deep learning for intelligent transportation systems: a survey of emerging trends. IEEE Trans Intell Transp Syst, 2020, 12(4): 1624-1639.

[2] Zhang L, Long R, Chen H, Geng J. A review of China’s road traffic carbon emissions. J Clean Prod, 2019, 207: 569-581.

[3] Zheng H, Lin F, Feng X, Chen Y. A hybrid deep learning model with attention-based Conv-LSTM networks for short-term traffic flow prediction. IEEE Trans Intell Transp Syst, 2020, 22(11): 6910-6920.

[4] Xu X, Jin X, Xiao D, Ma C, Wong SC. A hybrid autoregressive fractionally integrated moving average and nonlinear autoregressive neural network model for short-term traffic flow prediction. J Intell Transp Syst, 2021, 27(1): 1-18.

[5] Shahriari S, Ghasri M, Sisson SA, Rashidi T. Ensemble of ARIMA: combining parametric and bootstrapping technique for traffic flow prediction. Transp A Transp Sci, 2020, 16(3): 1552-1573.

[6] Duan P, Mao G, Zhang C, Wang S (2016) STARIMA-based traffic prediction with time-varying lags. in Proc. IEEE 19th International Conference Intelligent Transportation Systems 1610–1615. https://doi.org/arxiv-1701.00977

[7] Duan P, Mao G, Liang W, Zhang D. A unified spatio-temporal model for short-term traffic flow prediction. IEEE Trans Intell Transp Syst, 2019, 20(9): 3212-3223.

[8] Guo J, Huang W, Williams BM. Adaptive Kalman filter approach for stochastic short-term traffic flow rate prediction and uncertainty quantification. Transportation Research Part C: Emerging Technologies, 2014, 43(1): 50-64.

[9] Emami A, Sarvi M, Bagloee SA. Using Kalman filter algorithm for short-term traffic flow prediction in a connected vehicle environment. J Mod Transp, 2019, 27: 222-232.

[10] Shen Z, Wang W, Shen Q, Zhu S, Fardoun HM, Lou J. A novel learning method for multi-intersections aware traffic flow forecasting. Neurocomputing, 2020, 398(20): 477-484.

[11] Tang J, Chen X, Hu Z, Zong F, Han C, Li L. Traffic flow prediction based on combination of support vector machine and data denoising schemes. Physica A Stat Mech Appl, 2019, 534: 120642.

[12] Vijayaraju P, Sripathy B, Arivudainambi D, Balaji S. Hybrid memetic algorithm with two-dimensional discrete Haar wavelet transform for optimal sensor placement. IEEE Sens J, 2017, 17(7): 2267-2278.

[13] Cheng A, Jiang X, Li Y, Zhang C, Zhu H. Multiple sources and multiple measures based traffic flow prediction using the chaos theory and support vector regression method. Physica A Stat Mech Appl, 2017, 466: 422-434.

[14] Cai P, Wang Y, Lu G, Chen P, Ding C, Sun J. A spatiotemporal correlative k-nearest neighbor model for short-term traffic multistep forecasting. Transportation Research Part C: Emerging Technologies, 2016, 62: 21-34.

[15] Tian W, Zhang Y, Zhang Y, Chen H, Liu W. A short-term traffic flow prediction method for airport group route waypoints based on the spatiotemporal features of traffic flow. Aerospace, 2024, 11(4): 248.

[16] Qu D, Chen K, Wang S, Wang Q. A two-stage decomposition-reinforcement learning optimal combined short-time traffic flow prediction model considering multiple factors. Appl Sci, 2022, 12(16): 7978.

[17] Moretti F, Pizzuti S, Panzieri S, Annunziato M. Urban traffic flow forecasting through statistical and neural network bagging ensemble hybrid modeling. Neurocomputing, 2015, 167: 3-7.

[18] Sadeghi-Niaraki A, Mirshafiei P, Shakeri M, Choi S-M. Short-term traffic flow prediction using the modified Elman recurrent neural network optimized through a genetic algorithm. IEEE Access, 2020, 8: 217526-217540.

[19] Karlaftis M, Vlahogianni E. Statistical methods versus neural networks in transportation research: differences, similarities and some insights. Transportation Research Part C: Emerging Technologies, 2011, 19(3): 387-399.

[20] Dhanasekaran S, Gopal D, Logeshwaran J, Ramya N, Salau AO. Multi-model traffic forecasting in smart cities using graph neural networks and transformer-based multi-source visual fusion for intelligent transportation management. Int J Intell Transp Syst Res, 2024, 22: 518-541.

[21] Li X, Yin X, Huang X, Liu W, Zhang S, Zhang D. Multi-dynamic residual graph convolutional network with global feature enhancement for traffic flow prediction. Int J Mach Learn Cybern, 2024, 16: 873-889.

[22] Huang X, Jiang Y, Tang J. Mapredrnn: multi-attention predictive RNN for traffic flow prediction by dynamic spatio-temporal data fusion. Appl Intell, 2023, 53: 19372-19383.

[23] Shi R, Du L. Multi-section traffic flow prediction based on MLR-LSTM neural network. Sensors, 2022, 22(19): 7517.

[24] Zhao F, Zeng G, Lu K. EnLSTM-WPEO: short-term traffic flow prediction by ensemble LSTM, NNCT weight integration, and population extremal optimization. IEEE Trans Veh Technol, 2020, 69(1): 101-113.

[25] Yang H, Jiang C, Song Y, Fan W, Deng Z, Bai X. TARGCN: temporal attention recurrent graph convolutional neural network for traffic prediction. Complex Intell Syst, 2024, 10: 8179-8196.

[26] Khajeh Hosseini M, Talebpour A. Traffic prediction using time-space diagram: a convolutional neural network approach. Transportation Research Record: Journal of the Transportation Research Board, 2019, 2673(7): 425-435.

[27] Shao Q, Piao X, Yao X, Kong Y, Hu Y, Yin B, Zhang Y. An adaptive composite time series forecasting model for short-term traffic flow. J Big Data, 2024, 11: 102.

[28] Abideen UZ, Sun H, Yang Z, Ali A. The Deep 3D Convolutional Multi-Branching Spatial-Temporal-Based Unit Predicting Citywide Traffic Flow. Appl Sci, 2021, 10(21): 7778.

[29] Wu S. Spatiotemporal dynamic forecasting and analysis of regional traffic flow in urban road networks using deep learning convolutional neural network. IEEE Trans Intell Transp Syst, 2022, 23(2): 1607-1615.

[30] Alkilane K, Alfateh M, Shen Y. TFGAN: Traffic forecasting using generative adversarial network with multi-graph convolutional network. Knowl-Based Syst, 2021, 249: 108990.

[31] Mohammed GP, Alasmari N, Alsolai H, Alotaibi SS, Alotaibi N, Mohsen H. Autonomous short-term traffic flow prediction using pelican optimization with hybrid deep belief network in smart cities. Appl Sci, 2022, 12(21): 10828.

[32] Li L, Qin L, Qu X, Zhang J, Wang Y, Ran B. Day-ahead traffic flow forecasting based on a deep belief network optimized by the multi-objective particle swarm algorithm. Knowl-Based Syst, 2019, 172: 1-14.

[33] Koesdwiady A, Soua R, Karray F. Improving traffic flow prediction with weather information in connected cars: a deep learning approach. IEEE Trans Veh Technol, 2016, 65(12): 9508-9517.

[34] Yao R, Zhang W, Long M. Dlw-net model for traffic flow prediction under adverse weather. Transportmetrica B, 2021, 10(1): 499-524.

[35] Zhang W, Yao R, Du X, Ye J. Hybrid deep spatio-temporal models for traffic flow prediction on holidays and under adverse weather. IEEE Access, 2021, 9: 157165-157181.

[36] Li T, Ma J, Lee C. Markov-based time series modeling framework for traffic-network state prediction under various external conditions. Journal of Transportation Engineering Part A Systems, 2020.

[37] Li M, Wang H, Yang L, Liang Y, Shang Z, Wan H. Fast hybrid dimensionality reduction method for classification based on feature selection and grouped feature extraction. Expert Syst Appl, 2020, 1(2): 56-70.

[38] Huang X, Wu L, Ye Y. A review on dimensionality reduction techniques. Int J Pattern Recognit Artif Intell, 2019, 33(10): 1-25.

[39] Sebban M, Nock R. A hybrid filter/wrapper approach of feature selection using information theory. Pattern Recogn, 2002, 35(4): 835-846.

[40] Peng H, Long F, Ding C. Feature selection based on mutual information criteria of max-dependency, max-relevance, and min-redundancy. IEEE Trans Pattern Anal Mach Intell, 2005, 27(8): 1226-1238.

[41] Tu Q, Chen X, Liu X. Multi-strategy ensemble grey wolf optimizer and its application to feature selection. Appl Soft Comput, 2019, 76: 16-30.

[42] Rudy S, Liu H. Neural-network feature selector. IEEE Trans Neural Networks, 1997, 8(3): 654-662.

[43] Kumar S, John B. A novel Gaussian based particle swarm optimization gravitational search algorithm for feature selection and classification. Neural Comput Appl, 2021, 33(19): 1-15.

[44] Somol P, Pudil P, Kittler J. Fast branch & bound algorithms for optimal feature selection. IEEE Trans Pattern Anal Mach Intell, 2004, 26(7): 900-912.

[45] Chen X, Yuan G, Nie F, Ming Z. Semi-supervised feature selection via sparse rescaled linear square regression. IEEE Trans Knowl Data Eng, 2020, 32(1): 165-176.

[46] Shen H, Zhu Y, Zheng W, Zhu X. Half-quadratic minimization for unsupervised feature selection on incomplete data. IEEE Trans Neural Netw Learn Syst, 2021, 32(7): 3122-3135.

[47] Jan Hauke and Kossowski Tomasz. Comparison of values of Pearson's and Spearman's correlation coefficients on the same sets of data. Quaest Geogr, 2011, 30(2): 87-93.

[48] Cho K, Merrienboer B, Gulcehre C (2014) Learning phrase representations using RNN encoder-decoder for statistical machine translation. arXiv:1406.1078. https://doi.org/10.48550/arXiv.1406.1078

[49] Ju X, Liu F. Wind farm layout optimization using self-informed genetic algorithm with information guided exploitation. Appl Energy, 2019, 248(15): 429-445.

[50] Zhou H, Zhang J, Zhou Y, Guo X, Ma Y. A feature selection algorithm of decision tree based on feature weight. Expert Syst Appl, 2021, 164: 113842.

Funding

National Natural Science Foundation of China(U2333204)

National Key R&D Program of China(No. 2021YFB1600500)

Sichuan Provincial Transportation Technology Plan Project(2021-Z2-04)

Jiangsu Provincial Transportation Technology and Achievement Transformation Project(2023Y06)

RIGHTS & PERMISSIONS

The Author(s)