HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Interpolation Consistency Training for Semi-Supervised Learning

Vikas Verma; Kenji Kawaguchi; Alex Lamb; Juho Kannala; Arno Solin; Yoshua Bengio; David Lopez-Paz

Interpolation Consistency Training for Semi-Supervised Learning

Abstract

We introduce Interpolation Consistency Training (ICT), a simple and computation efficient algorithm for training Deep Neural Networks in the semi-supervised learning paradigm. ICT encourages the prediction at an interpolation of unlabeled points to be consistent with the interpolation of the predictions at those points. In classification problems, ICT moves the decision boundary to low-density regions of the data distribution. Our experiments show that ICT achieves state-of-the-art performance when applied to standard neural network architectures on the CIFAR-10 and SVHN benchmark datasets. Our theoretical analysis shows that ICT corresponds to a certain type of data-adaptive regularization with unlabeled points which reduces overfitting to labeled points under high confidence values.

Code Repositories

kevinghst/ICT
pytorch
Mentioned in GitHub
vikasverma1077/ICT
Official
pytorch
Mentioned in GitHub
yanzhicong/mixup_ssl
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-image-classification-on-cifarICT (CNN-13)
Percentage error: 7.29
semi-supervised-image-classification-on-cifarICT (WRN-28-2)
Percentage error: 7.66
semi-supervised-image-classification-on-cifar-11ICT (CNN-13)
Accuracy: 84.52
semi-supervised-image-classification-on-cifar-12ICT (CNN-13)
Accuracy: 90.74
semi-supervised-image-classification-on-svhnICT (WRN-28-2)
Accuracy: 96.47
semi-supervised-image-classification-on-svhnICT
Accuracy: 96.11

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
Interpolation Consistency Training for Semi-Supervised Learning | Papers | HyperAI