Vikas Verma, Kenji Kawaguchi, Alex Lamb, Juho Kannala, Arno Solin, Yoshua Bengio, David Lopez-Paz
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.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Image Classification | CIFAR-10, 4000 Labels | Percentage error | 7.29 | ICT (CNN-13) |
| Image Classification | CIFAR-10, 4000 Labels | Percentage error | 7.66 | ICT (WRN-28-2) |
| Image Classification | CIFAR-10, 2000 Labels | Accuracy | 90.74 | ICT (CNN-13) |
| Image Classification | CIFAR-10, 1000 Labels | Accuracy | 84.52 | ICT (CNN-13) |
| Image Classification | SVHN, 1000 labels | Accuracy | 96.47 | ICT (WRN-28-2) |
| Image Classification | SVHN, 1000 labels | Accuracy | 96.11 | ICT |
| Semi-Supervised Image Classification | CIFAR-10, 4000 Labels | Percentage error | 7.29 | ICT (CNN-13) |
| Semi-Supervised Image Classification | CIFAR-10, 4000 Labels | Percentage error | 7.66 | ICT (WRN-28-2) |
| Semi-Supervised Image Classification | CIFAR-10, 2000 Labels | Accuracy | 90.74 | ICT (CNN-13) |
| Semi-Supervised Image Classification | CIFAR-10, 1000 Labels | Accuracy | 84.52 | ICT (CNN-13) |
| Semi-Supervised Image Classification | SVHN, 1000 labels | Accuracy | 96.47 | ICT (WRN-28-2) |
| Semi-Supervised Image Classification | SVHN, 1000 labels | Accuracy | 96.11 | ICT |