A Hybrid Framework Model Based on Wavelet Neural Network with Improved Fruit Fly Optimization Algorithm for Traffic Flow Prediction

Abstract

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.