HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Mutual Teaching for Graph Convolutional Networks

Kun Zhan Chaoxi Niu

Mutual Teaching for Graph Convolutional Networks

Abstract

Graph convolutional networks produce good predictions of unlabeled samples due to its transductive label propagation. Since samples have different predicted confidences, we take high-confidence predictions as pseudo labels to expand the label set so that more samples are selected for updating models. We propose a new training method named as mutual teaching, i.e., we train dual models and let them teach each other during each batch. First, each network feeds forward all samples and selects samples with high-confidence predictions. Second, each model is updated by samples selected by its peer network. We view the high-confidence predictions as useful knowledge, and the useful knowledge of one network teaches the peer network with model updating in each batch. In mutual teaching, the pseudo-label set of a network is from its peer network. Since we use the new strategy of network training, performance improves significantly. Extensive experimental results demonstrate that our method achieves superior performance over state-of-the-art methods under very low label rates.

Code Repositories

kunzhan/MTGCN
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
node-classification-on-citeseer-05MT-GCN
Accuracy: 67.7%
node-classification-on-citeseer-1MT-GCN
Accuracy: 68.9%
node-classification-on-coraMT-GCN
Accuracy: 80.9%
node-classification-on-cora-05MT-GCN
Accuracy: 66.9%
node-classification-on-cora-1MT-GCN
Accuracy: 73.1%
node-classification-on-cora-3MT-GCN
Accuracy: 78.5%
node-classification-on-pubmed-003MT-GCN
Accuracy: 65.5%
node-classification-on-pubmed-005MT-GCN
Accuracy: 69.5%
node-classification-on-pubmed-01MT-GCN
Accuracy: 73.1%

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
Mutual Teaching for Graph Convolutional Networks | Papers | HyperAI