HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SPI-GCN: A Simple Permutation-Invariant Graph Convolutional Network

{Jean-Claude Crivello Nataliya Sokolovska Asma Atamna}

SPI-GCN: A Simple Permutation-Invariant Graph Convolutional Network

Abstract

A wide range of machine learning problems involve handling graph-structured data. Existing machine learning approaches for graphs, however, often imply computing expensive graph similarity measures, preprocessing input graphs, or explicitly ordering graph nodes. In this work, we present a novel and simple convolutional neural network architecture for supervised learning on graphs that is provably invariant to node permutation. The proposed architecture operates directly on arbitrary graphs and performs no node sorting. It also uses a simple multi-layer perceptron for prediction as opposed to conventional convolution layers commonly used in other deep learning approaches for graphs. Despite its simplicity, our architecture is competitive with state-of-the-art graph kernels and existing graph neural networks on benchmark graph classification data sets. Our approach clearly outperforms other deep learning algorithms for graphs on multiple multiclass classification tasks. We also evaluate our approach on a real-world original application in materials science, on which we achieve extremely reasonable results.

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-coil-ragSPI-GCN
Accuracy: 75.72
graph-classification-on-enzymesSPI-GCN
Accuracy: 50.17%
graph-classification-on-hydridesSPI-GCN
Accuracy: 82.25
graph-classification-on-imdb-bSPI-GCN
Accuracy: 60.40%
graph-classification-on-imdb-mSPI-GCN
Accuracy: 44.13%
graph-classification-on-mutagSPI-GCN
Accuracy: 84.40%
graph-classification-on-nci1SPI-GCN
Accuracy: 64.11%
graph-classification-on-proteinsSPI-GCN
Accuracy: 72.06%
graph-classification-on-ptcSPI-GCN
Accuracy: 56.41%
graph-classification-on-synthieSPI-GCN
Accuracy: 71.00

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
SPI-GCN: A Simple Permutation-Invariant Graph Convolutional Network | Papers | HyperAI