arXiv · 1803.07294
GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal Graphs
Abstract
We propose a new network architecture, Gated Attention Networks (GaAN), for learning on graphs. Unlike the traditional multi-head attention mechanism, which equally consumes all attention heads, GaAN uses a convolutional sub-network to control each attention head's importance. We demonstrate the effectiveness of GaAN on the inductive node classification problem. Moreover, with GaAN as a building block, we construct the Graph Gated Recurrent Unit (GGRU) to address the traffic speed forecasting problem. Extensive experiments on three real-world datasets show that our GaAN framework achieves state-of-the-art results on both tasks.
Explore related subjects
Keep this discovery
Jiani Zhang, Xingjian Shi, Junyuan Xie, Hao Ma, Irwin King, Dit-Yan Yeung. 2018-03-20. GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal Graphs. https://arxiv.org/abs/1803.07294
Cite the original work for its findings. Save a collection to share your selection of sources.