Idecomp: imbalance-aware decomposition for class-decomposed classification using conditional GANs

Buczek, Patryk and Zidan, Usama and Gaber, Mohamed Medhat and Abdelsamea, Mohammed M. (2023) Idecomp: imbalance-aware decomposition for class-decomposed classification using conditional GANs. Discover Artificial Intelligence, 3 (1). ISSN 2731-0809

s44163-023-00078-0.pdf - Published Version
Available under License Creative Commons Attribution.

Download (3MB)


Medical image classification tasks frequently encounter challenges associated with class imbalance, resulting in biased model training and suboptimal classification performance. To address this issue, the combination of class decomposition and transfer learning has proven to be effective in classifying imbalanced medical imaging datasets. Nevertheless, in order to further augment the performance gains achieved through the utilisation of class decomposition within deep learning frameworks, we propose a novel model coined imbalance-Aware Decomposition for Class-Decomposed Classification (iDeComp) model. By incorporating a conditional Generative Adversarial Network (GAN) model, iDeComp is capable of generating additional samples specifically tailored to underrepresented decomposed subclasses. This paper investigates the application of iDeComp using two different medical imaging datasets. iDeComp selects underrepresented samples from the training set of the sublevel classes within each dataset, which are then employed to train separate conditional Deep Convolutional GAN (DCGAN) models and verification models. The conditional DCGAN model is responsible for generating additional samples, while the verification model critically evaluates the appropriateness of the synthesised images. Subsequently, the resulting augmented samples are utilized to train the classification model. To assess the effectiveness of iDeComp, we employ various evaluation metrics including accuracy, precision, recall, and F1 score. The results obtained from our experiments clearly indicate that iDeComp outperforms existing approaches in terms of classifying both imbalanced datasets.

Item Type: Article
Identification Number:
22 August 2023Accepted
29 August 2023Published Online
Uncontrolled Keywords: class imbalance, conditional GAN, transfer learning, medical image classification
Subjects: CAH11 - computing > CAH11-01 - computing > CAH11-01-01 - computer science
Divisions: Faculty of Computing, Engineering and the Built Environment > School of Computing and Digital Technology
Depositing User: Gemma Tonks
Date Deposited: 27 Apr 2024 12:31
Last Modified: 27 Apr 2024 12:31

Actions (login required)

View Item View Item


In this section...