Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Graph Classification with 2D Convolutional Neural Networks

About

Graph learning is currently dominated by graph kernels, which, while powerful, suffer some significant limitations. Convolutional Neural Networks (CNNs) offer a very appealing alternative, but processing graphs with CNNs is not trivial. To address this challenge, many sophisticated extensions of CNNs have recently been introduced. In this paper, we reverse the problem: rather than proposing yet another graph CNN model, we introduce a novel way to represent graphs as multi-channel image-like structures that allows them to be handled by vanilla 2D CNNs. Experiments reveal that our method is more accurate than state-of-the-art graph kernels and graph CNNs on 4 out of 6 real-world datasets (with and without continuous node attributes), and close elsewhere. Our approach is also preferable to graph kernels in terms of time complexity. Code and data are publicly available.

Antoine Jean-Pierre Tixier, Giannis Nikolentzos, Polykarpos Meladianos, Michalis Vazirgiannis• 2017

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS (10-fold cross-validation)
Accuracy77.12
197
Graph ClassificationIMDB-B (test)
Accuracy70.4
134
Graph ClassificationCOLLAB (test)
Accuracy71.33
96
Graph ClassificationREDDIT-B--
71
Graph ClassificationREDDIT-B (test)
Accuracy89.12
32
Graph ClassificationREDDIT-5K (test)
Accuracy52.11
23
Graph ClassificationREDDIT-12K (test)
Accuracy48.13
11
Graph ClassificationCOLLAB
Running Time (s)5
8
Graph ClassificationREDDIT-5K
Runtime (s)16
3
Graph ClassificationREDDIT-12K
Runtime (s)52
3
Showing 10 of 12 rows

Other info

Code

Follow for update