A Spatial-Temporal Graph Mining Algorithm Based on Spatial-Temporal Sparse Attention
XIE Yi,WANG Qiang,LI Haihong,JIN Cheng,REN Hongrun,XUE Wen,XIONG Yun
DOI: https://doi.org/10.19678/j.issn.1000-3428.0063731
2023-01-01
Abstract:Existing spatial-temporal graph mining algorithms are typically based on static graph structures,which are predefined by experts or constructed via feature augmentation.These static graph structures rely on subjective prior knowledge and are not easily adaptable to temporal dynamic changes.Thus,automatically extracting dynamic graph features from spatialtemporal graph data is challenging.Therefore,this study proposes a spatial-temporal graph mining algorithm based on a SpatialTemporal Sparse Attention Network(STSAN).First,a spatial sparse attention layer is constructed by generating a sparse graph by determining the relationship between nodes at each time slice,and the attention mechanism is used on each sparse graph structure to extract the spatial(vertical) features of the nodes.Subsequently,the temporal sparse attention layer further extracts the temporal(horizontal) features of the nodes in a similar manner.Finally,the spatial-temporal dependency modeling is completed by stacking the spatial and temporal sparse attention layers in the spatial-Temporal sparse Transformer module.Experimental results demonstrate that,compared with DCRNN,STGCN algorithms,et al,STSAN can achieve performance improvements of 2.65%-16.35% on two public traffic flow datasets.The experiments also demonstrate that directly replacing the spatial feature capturing module of existing algorithms with the spatial sparse attention layer proposed in this study can achieve an average performance improvement of 3.18%-9.14% relative to the original algorithm.