Traffic flow prediction method based on combination of multiple attention and adaptive graph convolution

The invention discloses a traffic flow prediction method based on multi-head attention combined with adaptive graph convolution, and is applied to the technical field of traffic flow prediction. Comprising the following steps: S1, constructing a model; s2, modeling the spatial correlation of the tra...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: WANG HONGYAN, KAN SUNAN, ZHANG HONG, GONG LEI, ZHU SIYU, ZHANG XIJUN, CAO JIE
Format: Patent
Sprache:chi ; eng
Schlagworte:
Online-Zugang:Volltext bestellen
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:The invention discloses a traffic flow prediction method based on multi-head attention combined with adaptive graph convolution, and is applied to the technical field of traffic flow prediction. Comprising the following steps: S1, constructing a model; s2, modeling the spatial correlation of the traffic flow by using graph convolution, capturing adjacent spatial information of each node on a graph, and extracting spatial features of the traffic flow; s3, discovering the connection of hidden nodes in the graph structure from the data through training, thereby capturing the spatial relationship of the hidden traffic flow in the data; s4, capturing time dependence information of adjacent time slices of the traffic flow; s5, extracting spatio-temporal information of different spatial dimensions from different historical moments by introducing a plurality of attention heads, and capturing dynamic spatio-temporal correlation of traffic flow to obtain high-dimensional spatio-temporal features; and S6, fusion is carr