Exploring foci of:
arXiv (Cornell University)
GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings
June 2021 • Matthias Fey, Jan Eric Lenssen, Frank Weichert, Jure Leskovec
We present GNNAutoScale (GAS), a framework for scaling arbitrary message-passing GNNs to large graphs. GAS prunes entire sub-trees of the computation graph by utilizing historical embeddings from prior training iterations, leading to constant GPU memory consumption in respect to input node size without dropping any data. While existing solutions weaken the expressive power of message passing due to sub-sampling of edges or non-trainable propagations, our approach is provably able to maintain the expressive power o…
Computer Science
Theoretical Computer Science
Algorithm
Mathematics
Database
Engineering
Structural Engineering
Geometry