HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Graph Classification with 2D Convolutional Neural Networks

Antoine Jean-Pierre Tixier; Giannis Nikolentzos; Polykarpos Meladianos; Michalis Vazirgiannis

Graph Classification with 2D Convolutional Neural Networks

Abstract

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.

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-collab2D CNN
Accuracy: 71.76%
graph-classification-on-imdb-b2D CNN
Accuracy: 70.40%
graph-classification-on-re-m12k2D CNN
Accuracy: 48.13%
graph-classification-on-re-m5k2D CNN
Accuracy: 52.11%

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Graph Classification with 2D Convolutional Neural Networks | Papers | HyperAI