12/07/2020

Graph Structure of Neural Networks

Jiaxuan You, Kaiming He, Jure Leskovec, Saining Xie

Keywords: Deep Learning - General

Abstract: Neural networks are often represented as graphs of connections between the neurons. However, despite their wide use there is currently no understanding of the relationship between the graph structure of a neural network and its predictive performance. Here we systematically investigate this relationship, via developing a novel graph-based representation of neural networks called relational graph, where computation is specified by rounds of message exchange along the graph structure. Using our novel framework we show that (1) there is a “sweet spot”, where relational graphs within certain range of average path length and clustering coefficient lead to neural networks with significant improvements in predictive performance; (2) perhaps even more surprisingly, we find that these sweet spots tend to highly correlate across different architectures and datasets; and, (3) we show that discovering top-performing relational graphs only requires a few epochs of training. Overall, our results suggest promising avenues for designing and understanding neural networks with graph representations.

 0
 0
 0
 0
This is an embedded video. Talk and the respective paper are published at ICML 2020 virtual conference. If you are one of the authors of the paper and want to manage your upload, see the question "My papertalk has been externally embedded..." in the FAQ section.

Comments

Post Comment
no comments yet
code of conduct: tbd

Similar Papers