<abstract xmlns="http://www.w3.org/1999/xhtml">

<p>In practical applications of machine learning, the class distribution of the collected training set is usually imbalanced, i.e., there is a large difference among the sizes of different classes. The class imbalance problem often hinders the achievable generalization performance of most classifier learning algorithms to a large extent. To ameliorate the learning performance, some effective approaches have been proposed in the literature, where the recently presented GAN-based oversampling methods are very representative. However, their generated minority class examples have the risk of high similarity and duplication degree. To further ameliorate the quality of the generated minority class examples, i.e., to make the generated examples effectively expand the minority class region, a novel oversampling approach named the GWGAN-GP is proposed, which is based on the Gaussian distribution label within the framework of aWasserstein generative adversarial network with gradient penalty (WGAN-GP). Our GWGAN-GP approach incorporates the Gaussian distribution as an input label, thereby making the generated examples more diverse and dispersive. The examples are then combined with the original dataset to form a balanced dataset, which is subsequently utilized to evaluate the classification performance of three selected classification algorithms. Experimental results on 16 imbalanced datasets demonstrate that the GWGAN-GP not only generates examples that better conform to the distribution of the original dataset, but also achieves superior classification performance. Specifically, when combined with the KNN classifier, the GWGAN-GP significantly outperforms other oversampling approaches considered in the study.</p>
</abstract>

In practical applications of machine learning, the class distribution of the collected training set is usually imbalanced, i.e., there is a large difference among the sizes of different classes. The class imbalance problem often hinders the achievable generalization performance of most classifier learning algorithms to a large extent. To ameliorate the learning performance, some effective approaches have been proposed in the literature, where the recently presented GAN-based oversampling methods are very representative. However, their generated minority class examples have the risk of high similarity and duplication degree. To further ameliorate the quality of the generated minority class examples, i.e., to make the generated examples effectively expand the minority class region, a novel oversampling approach named the GWGAN-GP is proposed, which is based on the Gaussian distribution label within the framework of aWasserstein generative adversarial network with gradient penalty (WGAN-GP). Our GWGAN-GP approach incorporates the Gaussian distribution as an input label, thereby making the generated examples more diverse and dispersive. The examples are then combined with the original dataset to form a balanced dataset, which is subsequently utilized to evaluate the classification performance of three selected classification algorithms. Experimental results on 16 imbalanced datasets demonstrate that the GWGAN-GP not only generates examples that better conform to the distribution of the original dataset, but also achieves superior classification performance. Specifically, when combined with the KNN classifier, the GWGAN-GP significantly outperforms other oversampling approaches considered in the study.

A Gaussian–Based WGAN–GP Oversampling Approach for Solving the Class Imbalance Problem

Department of Computer Science and Technology

International Journal of Applied Mathematics and Computer Science

This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 3.0 License.

In practical applications of machine learning, the class distribution of the collected training set is usually imbalanced, i.e., there is a large difference...

A Gaussian–Based WGAN–GP Oversampling Approach for Solving the Class Imbalance Problem

Publié en ligne: 25 juin 2024

Pages: 291 - 307

Reçu: 31 juil. 2023

Accepté: 21 févr. 2024

DOI: https://doi.org/10.61822/amcs-2024-0021

Mots clés
machine learning, class imbalance, generative adversarial networks, oversampling, data duplication

© 2024 Qian Zhou et al., published by Sciendo

This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 3.0 License.

A Gaussian–Based WGAN–GP Oversampling Approach for Solving the Class Imbalance Problem

Publié en ligne: 25 juin 2024

Pages: 291 - 307

Reçu: 31 juil. 2023

Accepté: 21 févr. 2024

DOI: https://doi.org/10.61822/amcs-2024-0021

Mots clésmachine learning, class imbalance, generative adversarial networks, oversampling, data duplication

© 2024 Qian Zhou et al., published by Sciendo

This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 3.0 License.

Mots clés
machine learning, class imbalance, generative adversarial networks, oversampling, data duplication