The invention provides a time-varying graph neural network
traffic flow prediction method based on a dynamic
memory bank, and belongs to the technical field of
traffic prediction. The method adopts a layered deep
neural network architecture, and comprises a data embedding layer, a space-time coding layer, a memory enhancement layer and a prediction output layer. The data embedding layer preprocesses a
traffic flow sequence, associates a collaborative coding
time sequence mode with a road network, and synchronously constructs a dynamic graph structure; the space-time coding layer is subjected to space-time
stream decoupling extraction, a space
branch models multi-
scale space dependence through a time
delay graph
convolution module and a space Mama module, and a time
branch extracts multi-
granularity time features through a hierarchical
time sequence sensing module and a time Mama module; the memory enhancement layer performs
pattern matching and reconstruction on the space-time fusion features by means of a dynamic
memory bank; and the prediction output layer generates a prediction result by taking the GCRN as a decoder. According to the method, the space-time dependence of the traffic situation is accurately captured, the prediction curve is highly fit with the true value, and the high-precision prediction of the
traffic flow is realized.