
摘要
我们介绍了插值一致性训练(Interpolation Consistency Training, ICT),这是一种在半监督学习范式下训练深度神经网络的简单且计算高效的算法。ICT鼓励对未标记点的插值预测与这些点上的预测插值保持一致。在分类问题中,ICT将决策边界移动到数据分布的低密度区域。我们的实验表明,当将ICT应用于CIFAR-10和SVHN基准数据集上的标准神经网络架构时,它能够实现最先进的性能。理论分析显示,ICT相当于一种基于未标记点的数据自适应正则化方法,该方法在高置信度值下减少了对已标记点的过拟合现象。
代码仓库
kevinghst/ICT
pytorch
GitHub 中提及
vikasverma1077/ICT
官方
pytorch
GitHub 中提及
yanzhicong/mixup_ssl
pytorch
GitHub 中提及
perrying/realistic-ssl-evaluation-pytorch
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| semi-supervised-image-classification-on-cifar | ICT (CNN-13) | Percentage error: 7.29 |
| semi-supervised-image-classification-on-cifar | ICT (WRN-28-2) | Percentage error: 7.66 |
| semi-supervised-image-classification-on-cifar-11 | ICT (CNN-13) | Accuracy: 84.52 |
| semi-supervised-image-classification-on-cifar-12 | ICT (CNN-13) | Accuracy: 90.74 |
| semi-supervised-image-classification-on-svhn | ICT (WRN-28-2) | Accuracy: 96.47 |
| semi-supervised-image-classification-on-svhn | ICT | Accuracy: 96.11 |