HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

DEMO-Net: Degree-specific Graph Neural Networks for Node and Graph Classification

Jun Wu; Jingrui He; Jiejun Xu

DEMO-Net: Degree-specific Graph Neural Networks for Node and Graph Classification

Abstract

Graph data widely exist in many high-impact applications. Inspired by the success of deep learning in grid-structured data, graph neural network models have been proposed to learn powerful node-level or graph-level representation. However, most of the existing graph neural networks suffer from the following limitations: (1) there is limited analysis regarding the graph convolution properties, such as seed-oriented, degree-aware and order-free; (2) the node's degree-specific graph structure is not explicitly expressed in graph convolution for distinguishing structure-aware node neighborhoods; (3) the theoretical explanation regarding the graph-level pooling schemes is unclear. To address these problems, we propose a generic degree-specific graph neural network named DEMO-Net motivated by Weisfeiler-Lehman graph isomorphism test that recursively identifies 1-hop neighborhood structures. In order to explicitly capture the graph topology integrated with node attributes, we argue that graph convolution should have three properties: seed-oriented, degree-aware, order-free. To this end, we propose multi-task graph convolution where each task represents node representation learning for nodes with a specific degree value, thus leading to preserving the degree-specific graph structure. In particular, we design two multi-task learning methods: degree-specific weight and hashing functions for graph convolution. In addition, we propose a novel graph-level pooling/readout scheme for learning graph representation provably lying in a degree-specific Hilbert kernel space. The experimental results on several node and graph classification benchmark data sets demonstrate the effectiveness and efficiency of our proposed DEMO-Net over state-of-the-art graph neural network models.

Code Repositories

jwu4sml/DEMO-Net
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-enzymesDEMO-Net(weight)
Accuracy: 27.2
node-classification-on-blogcatalog-1DEMO-Net(weight)
Accuracy: 84.9
node-classification-on-brazil-air-trafficDEMO-Net(weight)
Accuracy: 0.543 ± 0.034
node-classification-on-europe-air-trafficDEMO-Net(weight)
Accuracy: 45.9
node-classification-on-facebookDEMO-Net(weight)
Accuracy: 91.9
node-classification-on-flickrDEMO-Net(weight)
Accuracy: 0.656 ± 0.000
node-classification-on-usa-air-trafficDEMO-Net(weight)
Accuracy: 64.7
node-classification-on-wiki-voteDEMO-Net(weight)
Accuracy: 99.8

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
DEMO-Net: Degree-specific Graph Neural Networks for Node and Graph Classification | Papers | HyperAI