Tri-net for Semi-Supervised Deep Learning
Tri-net for Semi-Supervised Deep Learning
Dong-Dong Chen, Wei Wang, Wei Gao, Zhi-Hua Zhou
Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence
Main track. Pages 2014-2020.
https://doi.org/10.24963/ijcai.2018/278
Deep neural networks have witnessed great successes in various real applications, but it requires a large number of labeled data for training. In this paper, we propose tri-net, a deep neural network which is able to use massive unlabeled data to help learning with limited labeled data. We consider model initialization, diversity augmentation and pseudo-label editing simultaneously. In our work, we utilize output smearing to initialize modules, use fine-tuning on labeled data to augment diversity and eliminate unstable pseudo-labels to alleviate the influence of suspicious pseudo-labeled data.
Experiments show that our method achieves the best performance in comparison with state-of-the-art semi-supervised deep learning methods. In particular, it achieves 8.30% error rate on CIFAR-10 by using only 4000 labeled examples.
Keywords:
Machine Learning: Deep Learning
Machine Learning: Semi-Supervised Learning