HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs

Martin Simonovsky; Nikos Komodakis

Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs

Abstract

A number of problems can be formulated as prediction on graph-structured data. In this work, we generalize the convolution operator from regular grids to arbitrary graphs while avoiding the spectral domain, which allows us to handle graphs of varying size and connectivity. To move beyond a simple diffusion, filter weights are conditioned on the specific edge labels in the neighborhood of a vertex. Together with the proper choice of graph coarsening, we explore constructing deep neural networks for graph classification. In particular, we demonstrate the generality of our formulation in point cloud classification, where we set the new state of the art, and on a graph classification dataset, where we outperform other deep learning approaches. The source code is available at https://github.com/mys007/ecc

Code Repositories

mys007/ecc
Official
pytorch
Mentioned in GitHub
rusty1s/pytorch_cluster
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-object-classification-on-modelnet10ECC (12 votes)
Accuracy: 90
3d-object-classification-on-modelnet40ECC (12 votes)
Classification Accuracy: 83.2
3d-point-cloud-classification-on-modelnet40ECC
Mean Accuracy: 83.2
Overall Accuracy: 87.4
3d-point-cloud-classification-on-sydney-urbanECC
F1: 78.4
graph-classification-on-ddECC (5 scores)
Accuracy: 74.1%
graph-classification-on-enzymesECC (5 scores)
Accuracy: 52.67%
graph-classification-on-mutagECC (5 scores)
Accuracy: 88.33%
graph-classification-on-nci1ECC (5 scores)
Accuracy: 83.8%
graph-classification-on-nci109ECC (5 scores)
Accuracy: 82.14

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
Dynamic Edge-Conditioned Filters in Convolutional Neural Networks on Graphs | Papers | HyperAI