Controlling the quality of GAN-based generated images for predictions tasks - Archive ouverte HAL
Communication Dans Un Congrès Année : 2022

Controlling the quality of GAN-based generated images for predictions tasks

Résumé

Recently, Generative Adversarial Networks (GANs) have been widely applied for data augmentation given limited datasets. The state of the art is dominated by measures evaluating the quality of the generated images, that are typically all added to the training dataset. There is however no control of the generated data, in terms of the compromise between diversity and closeness to the original data, and this is our work’s focus. Our study concerns the prediction of soil moisture dissipation rates from synthetic aerial images using a CNN regressor. CNNs, however, require large datasets to successfully train them. To this end, we apply and compare two Generative Adversarial Networks (GANs) models: (1) Deep Convolutional Neural Network (DCGAN) and (2) Bidirectional Generative Adversarial Network (BiGAN), to generate fake images. We propose a novel approach that consists of studying which generated images to include into the augmented dataset. We consider a various number of images, selected for training according to their realistic character, based on the discriminator loss. The results show that, using our approach, the CNN trained on the augmented dataset generated by BiGAN and DCGAN allows a significant relative decrease of the Mean Absolute Error w.r.t the CNN trained on the original dataset. We believe that our approach can be generalized to any Generative Adversarial Network model.
Fichier principal
Vignette du fichier
Hammouch er al. (2022) - Controlling the Quality of GAN-Based Generated Images for Predictions Tasks.pdf (16.82 Mo) Télécharger le fichier
Origine Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-03944317 , version 1 (03-03-2023)

Identifiants

Citer

Hajar Hammouch, Mounim El-Yacoubi, Huafeng Qin, Hassan Berbia, Mohamed Chikhaoui. Controlling the quality of GAN-based generated images for predictions tasks. CPRAI 2022: 3rd International Conference on Pattern Recognition and Artificial Intelligence, Jun 2022, Paris, France. pp.121-133, ⟨10.1007/978-3-031-09037-0_11⟩. ⟨hal-03944317⟩
45 Consultations
26 Téléchargements

Altmetric

Partager

More