An Incremental Self-Labeling Strategy for Semi-Supervised Deep Learning Based on Generative Adversarial Networks

被引:8
作者
Wei, Xiaotao [1 ]
Wei, Xiang [1 ]
Xing, Weiwei [1 ]
Lu, Siyang [2 ]
Lu, Wei [1 ]
机构
[1] Beijing Jiaotong Univ, Sch Software Engn, Beijing 100044, Peoples R China
[2] Beijing Jiaotong Univ, Beijing Key Lab Secur & Privacy Intelligent Trans, Beijing 100044, Peoples R China
关键词
Deep learning; semi-supervised learning; generative adversarial networks; self-labeling;
D O I
10.1109/ACCESS.2020.2964315
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The recent success of deep neural networks is attributed in part to large-scale well-labeled training data. However, with the ever-increasing size of modern datasets, combined with the difficulty of obtaining label information, semi-supervised learning (SSL) has become one of the most remarkable issues in data analysis. In this paper, we propose an Incremental Self-Labeling strategy for SSL based on Generative Adversarial Nets (ISL-GAN), which functions by constantly assigning unlabeled data with virtual labels for promoting the training process. Specifically, during the virtual labeling process, we introduce a temporal-based self-labeling strategy for safe and stable data labeling. Then, to dynamically assign more virtual labels to data during the training, we conduct a phased incremental label screening and updating strategy. Finally, to balance the contribution of samples with different loss during the training process, we further introduce the Balance factor Term (BT). The experimental results show that the proposed method gives rise to state-of-the-art semi-supervised learning results for the MNIST, CIFAR-10, and SVHN datasets. Particularly, our model performs well with fewer labeled conditions. With a dataset of only 1,000 labeled CIFAR-10 images with CONV-Large Net, a test error of 11.2; can be achieved, and nearly the same performance with a 3.5; test error can be achieved with both 500 and 1,000 image-labeled SVHN datasets.
引用
收藏
页码:8913 / 8921
页数:9
相关论文
共 35 条
  • [11] Ghosh A, 2017, AAAI CONF ARTIF INTE, P1919
  • [12] Goodfellow I.J., 2014, ADV NEUR IN, p1406.2661, DOI DOI 10.1145/3422622
  • [13] Label Propagation for Deep Semi-supervised Learning
    Iscen, Ahmet
    Tolias, Giorgos
    Avrithis, Yannis
    Chum, Ondrej
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 5065 - 5074
  • [14] Krizhevsky A., 2009, Learning multiple layers of features from tiny images
  • [15] Gradient-based learning applied to document recognition
    Lecun, Y
    Bottou, L
    Bengio, Y
    Haffner, P
    [J]. PROCEEDINGS OF THE IEEE, 1998, 86 (11) : 2278 - 2324
  • [16] LeCun Y., 2015, Nature, V521, P436, DOI [DOI 10.1038/NATURE14539, 10.1038/nature14539]
  • [17] Li CX, 2017, ADV NEUR IN, V30
  • [18] Smooth Neighbors on Teacher Graphs for Semi-supervised Learning
    Luo, Yucen
    Zhu, Jun
    Li, Mengxi
    Ren, Yong
    Zhang, Bo
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 8896 - 8905
  • [19] Miyato T, 2019, IEEE T PATTERN ANAL, V41, P1979, DOI 10.1109/TPAMI.2018.2858821
  • [20] Netzer Y., 2011, P NEURIPSW