Accurate traffic flow prediction can provide sufficient information for the formation of symmetric traffic flow. To overcome the problem that the basic fruit fly optimization algorithm (FOA) is easy to fall into local optimum and the search method is single, an improved fruit fly optimization algorithm (IFOA) based on parallel search strategy and group cooperation strategy is proposed. The multi-swarm mechanism is introduced in the parallel search strategy, in which each subswarm is independent and multiple center positions are determined in the iterative process, thereby avoiding the problems of reduced diversity and premature convergence. To increase communication between fruit fly subswarms, the informative fruit flies selected from subswarms are guided by the randomly generated binary fruit fly to achieve the crossover operation in the group cooperation strategy. Then a hybrid framework model based on wavelet neural network (WNN) with IFOA (IFOA-WNN) for traffic flow prediction is designed, in which IFOA is applied to explore appropriate structure parameters for WNN to achieve better prediction performance. The simulation results verify that the IFOA can provide high-quality structural parameters for WNN, and the hybrid IFOA-WNN prediction model can achieve higher prediction accuracy and stability than the compared methods.

Accurate traffic flow forecasting is very important for urban planning and traffic management. However, this is a huge challenge due to the complex spatial–temporal relationships. Although the existing methods have researched spatial–temporal relationships, they neglect the long periodic aspects of traffic flow data, and thus cannot attain a satisfactory result. In this paper, we propose a novel model Attention-Based Spatial–Temporal Convolution Gated Recurrent Unit (ASTCG) to solve the traffic flow forecasting problem. ASTCG has two core components: the multi-input module and the STA-ConvGru module. Based on the cyclical nature of traffic flow data, the data input to the multi-input module are divided into three parts, near-neighbor data, daily-periodic data, and weekly-periodic data, thus enabling the model to better capture the time dependence. The STA-ConvGru module, formed by CNN, GRU, and attention mechanism, can capture both temporal and spatial dependencies of traffic flow. We evaluate our proposed model using real-world datasets and experiments show that the ASTCG model outperforms the state-of-the-art model.