Diffusion-Convolutional Neural Networks
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Pubmed | Accuracy76.8 | 742 | |
| Graph Classification | PROTEINS | Accuracy61.3 | 742 | |
| Node Classification | Citeseer (test) | Accuracy0.711 | 729 | |
| Graph Classification | MUTAG | Accuracy67 | 697 | |
| Node Classification | Cora (test) | Mean Accuracy86.77 | 687 | |
| Node Classification | PubMed (test) | Accuracy89.76 | 500 | |
| Graph Classification | NCI1 | Accuracy62.61 | 460 | |
| Graph Classification | COLLAB | Accuracy52.11 | 329 | |
| Graph Classification | IMDB-B | Accuracy49.06 | 322 | |
| Graph Classification | ENZYMES | Accuracy42.44 | 305 |