RL-GCN: Traffic flow prediction based on graph convolution and reinforcement learning for smart cities

Abstract

The traffic flow problem has become essential in urban planning and management in today’s increasingly urbanized world. Traditional traffic flow prediction models cannot fully consider urban traffic networks’ complex and dynamic characteristics. To this end, this paper proposes a traffic flow prediction method for smart cities (RL-GCN) based on graph convolution, LSTM network and reinforcement learning, aiming to solve the problem of urban traffic flow prediction. Firstly, we use the graph convolutional neural network to process the urban traffic network data features, then use the LSTM network model to learn the temporal information, and then combine the reinforcement learning algorithm to develop the optimal traffic control strategy based on which the future traffic flow is predicted. Our experiments on several datasets show that the model developed in this paper has outstanding performance for urban traffic flow prediction. Compared with the traditional traffic flow prediction methods, the method in this paper has significantly improved prediction accuracy. Our research can provide valuable references and inspiration in urban planning and traffic management.