HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling

Jie Chen; Tengfei Ma; Cao Xiao

FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling

Abstract

The graph convolutional networks (GCN) recently proposed by Kipf and Welling are an effective graph model for semi-supervised learning. This model, however, was originally designed to be learned with the presence of both training and test data. Moreover, the recursive neighborhood expansion across layers poses time and memory challenges for training with large, dense graphs. To relax the requirement of simultaneous availability of test data, we interpret graph convolutions as integral transforms of embedding functions under probability measures. Such an interpretation allows for the use of Monte Carlo approaches to consistently estimate the integrals, which in turn leads to a batched training scheme as we propose in this work---FastGCN. Enhanced with importance sampling, FastGCN not only is efficient for training but also generalizes well for inference. We show a comprehensive set of experiments to demonstrate its effectiveness compared with GCN and related models. In particular, training is orders of magnitude more efficient while predictions remain comparably accurate.

Code Repositories

gkunnan97/fastgcn_pytorch
pytorch
Mentioned in GitHub
gmancino/fastgcn-pytorch
pytorch
Mentioned in GitHub
jiechenjiechen/FastGCN-matlab
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-citeseer-fullFastGCN
Accuracy: 77.60%
node-classification-on-cora-full-supervisedFastGCN
Accuracy: 85.00%
node-classification-on-pubmed-full-supervisedFastGCN
Accuracy: 88.00%
node-classification-on-redditFastGCN
Accuracy: 93.70%

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
FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling | Papers | HyperAI