Application of GAN for Reducing Data Imbalance under Limited Dataset

Gaurav Adke

2022

Abstract

The paper discusses architectural and training improvements of generative adversarial network (GAN) model for stable training. The advanced GAN architecture is proposed combining these improvements and it is applied for augmentation of a tire joint nonconformity dataset used for classification applications. The dataset used is highly unbalanced with higher number of conformity images. This unbalanced and limited dataset of nonconformity identification poses challenges in developing accurate nonconformity classification models. Therefore, a research is carried out in the presented work to augment the nonconformity dataset along with increasing the balance between different nonconformity classes. The quality of generated images is improved by incorporating recent developments in GANs. The present study shows that the proposed advanced GAN model is helpful in improving the performance classification model by augmentation under a limited unbalanced dataset. Generated results of advanced GAN are evaluated using Fréchet Inception Distance (FID) score, which shows large improvement over styleGAN architecture. Further experiments for dataset augmentation using generated images show 12% improvement in classification model accuracy over the original dataset. The potency of augmentation using GAN generated images is experimentally proved using principal component analysis plots.

Paper Citation

in Harvard Style

Adke G. (2022). Application of GAN for Reducing Data Imbalance under Limited Dataset. In Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP; ISBN 978-989-758-555-5, SciTePress, pages 60-68. DOI: 10.5220/0010782800003124

in Bibtex Style

@conference{visapp22,
author={Gaurav Adke},
title={Application of GAN for Reducing Data Imbalance under Limited Dataset},
booktitle={Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP},
year={2022},
pages={60-68},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010782800003124},
isbn={978-989-758-555-5},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 17th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022) - Volume 4: VISAPP
TI - Application of GAN for Reducing Data Imbalance under Limited Dataset
SN - 978-989-758-555-5
AU - Adke G.
PY - 2022
SP - 60
EP - 68
DO - 10.5220/0010782800003124
PB - SciTePress