Diffusion-Convolutional Neural Networks

NeurIPS 2016  ·  James Atwood, Don Towsley ·

We present diffusion-convolutional neural networks (DCNNs), a new model for graph-structured data. Through the introduction of a diffusion-convolution operation, we show how diffusion-based representations can be learned from graph-structured data and used as an effective basis for node classification. DCNNs have several attractive qualities, including a latent representation for graphical data that is invariant under isomorphism, as well as polynomial-time prediction and learning that can be represented as tensor operations and efficiently implemented on the GPU. Through several experiments with real structured datasets, we demonstrate that DCNNs are able to outperform probabilistic relational models and kernel-on-graph methods at relational node classification tasks.

PDF Abstract NeurIPS 2016 PDF NeurIPS 2016 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Node Classification CiteSeer (0.5%) DCNN Accuracy 53.1% # 9
Node Classification CiteSeer (1%) DCNN Accuracy 62.2% # 8
Node Classification CiteSeer with Public Split: fixed 20 nodes per class DCNN Accuracy 69.4% # 33
Node Classification Cora (0.5%) DCNN Accuracy 59.0% # 9
Node Classification Cora (1%) DCNN Accuracy 66.4% # 9
Node Classification Cora (3%) DCNN Accuracy 76.7% # 9
Node Classification Cora with Public Split: fixed 20 nodes per class DCNN Accuracy 79.7% # 29
Node Classification PubMed (0.03%) DCNN Accuracy 60.9% # 8
Node Classification PubMed (0.05%) DCNN Accuracy 66.7% # 8
Node Classification PubMed (0.1%) DCNN Accuracy 73.1% # 7
Node Classification PubMed with Public Split: fixed 20 nodes per class DCNN Accuracy 76.8% # 28

Methods