论文标题

与条件生成的对抗网络共同监督学习范式,用于样品分类

Co-supervised learning paradigm with conditional generative adversarial networks for sample-efficient classification

论文作者

Zhen, Hao, Shi, Yucheng, Yang, Jidong J., Vehni, Javad Mohammadpour

论文摘要

使用监督学习的分类需要注释大量的类平衡数据,以进行模型培训和测试。这实际上限制了有监督的学习,尤其是深度学习的应用程序范围。为了解决与有限和不平衡数据相关的问题,本文介绍了样本效率高,共同监督的学习范式(SEC-CGAN),在该学习范式(SEC-CGAN)中,在该范式中,在该培训过程中,与分类器和补充语调的综合典型培训了一个有条件的生成对抗网络(CGAN),并在培训过程中培训了与分类器和补充语调的合成典型。在这种情况下,CGAN不仅是一个共同的人,而且还提供了互补的质量示例,可以以端到端的方式帮助分类器培训。实验表明,提出的SEC-CAN优于外部分类器GAN(EC-GAN)和基线RESNET-18分类器。为了进行比较,上述方法中的所有分类器都采用RESNET-18体系结构作为骨干。特别是,对于Street View House数字数据集,使用5%的培训数据,SEC-CAN实现了90.26%的测试准确性,而EC-GAN的测试准确性为88.59%,基线分类器的测试准确性为87.17%。对于高速公路图像数据集,使用10%的培训数据,SEC-CAN实现了98.27%的测试准确性,而EC-GAN为97.84%,基线分类器的测试准确性为97.84%。

Classification using supervised learning requires annotating a large amount of classes-balanced data for model training and testing. This has practically limited the scope of applications with supervised learning, in particular deep learning. To address the issues associated with limited and imbalanced data, this paper introduces a sample-efficient co-supervised learning paradigm (SEC-CGAN), in which a conditional generative adversarial network (CGAN) is trained alongside the classifier and supplements semantics-conditioned, confidence-aware synthesized examples to the annotated data during the training process. In this setting, the CGAN not only serves as a co-supervisor but also provides complementary quality examples to aid the classifier training in an end-to-end fashion. Experiments demonstrate that the proposed SEC-CGAN outperforms the external classifier GAN (EC-GAN) and a baseline ResNet-18 classifier. For the comparison, all classifiers in above methods adopt the ResNet-18 architecture as the backbone. Particularly, for the Street View House Numbers dataset, using the 5% of training data, a test accuracy of 90.26% is achieved by SEC-CGAN as opposed to 88.59% by EC-GAN and 87.17% by the baseline classifier; for the highway image dataset, using the 10% of training data, a test accuracy of 98.27% is achieved by SEC-CGAN, compared to 97.84% by EC-GAN and 95.52% by the baseline classifier.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源