Semi-supervised generative adversarial networks with spatial coevolution for enhanced image generation and classification

被引:4
作者
Toutouh, Jamal [1 ,2 ]
Nalluru, Subhash [2 ]
Hemberg, Erik [2 ]
O'Reilly, Una-May [2 ]
机构
[1] Univ Malaga, ITIS Software, Malaga 29071, Spain
[2] MIT, Comp Sci & Artificial Intelligence Lab, Cambridge, MA 02139 USA
关键词
Generative adversarial network; Semi supervised learning; Coevolution; Spatial distribution;
D O I
10.1016/j.asoc.2023.110890
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Labeling images for classification can be expensive. Semi-Supervised Learning (SSL) Generative Adversarial Network (GAN) methods train good classifiers with a few labeled images. However, authors generally do not train SSL-GAN generators to produce new high-quality images, but as a component to train the classifier. In this article, we use a coevolutionary algorithm (CoEA) with SSL-GANs to train both the classifier and the image generative model using a few labeled images. A CoEA introduces diversity into the GAN training and mitigates training pathologies. We use a two-dimensional grid of GANs to inject diversity via distributed training that exchanges GAN components between neighboring cells based on performance and population-based hyperparameter tuning. In addition, we identify simple and efficient SSL-GAN architectures. We demonstrate the utility on three separate benchmark datasets, achieving good classification accuracy and high image quality generation while using fewer labeled data exemplars. The image quality and classification accuracy are also competitive with State-of-the-Art methods.
引用
收藏
页数:22
相关论文
共 61 条
[1]  
Al-Dujaili A., 2018, ASS ADVANCEMENT ARTI, P1
[2]  
[Anonymous], 2012, Handbook of Natural Computing, DOI DOI 10.1007/978-3-540-92910-931
[3]  
[Anonymous], 2016, P 4 INT C LEARN REPR
[4]  
[Anonymous], 2022, Conv2D-PyTorch master documentation
[5]  
[Anonymous], 2018, ICLR2018
[6]  
[Anonymous], 2022, ConvTranspose2D-PyTorch master documentation
[7]  
Arjovsky M, 2017, PR MACH LEARN RES, V70
[8]  
Arora S., 2018, INT C LEARNING REPRE
[9]  
Binkowski M.., 2018, INT C LEARN REPR
[10]   A Systematic Literature Review of Volumetric 3D Model Reconstruction Methodologies Using Generative Adversarial Networks [J].
Byrd, Riley ;
Damani, Kulin ;
Che, Hongjia ;
Calandra, Anthony ;
Kim, Dae-kyoo .
JOURNAL OF INFORMATION SCIENCE AND ENGINEERING, 2022, 38 (06) :1243-1263