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

Preview

Text s44163-023-00078-0.pdf - Published Version Available under License Creative Commons Attribution. Download (3MB)

Abstract

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:	10.1007/s44163-023-00078-0
Dates:	Date Event 22 August 2023 Accepted 29 August 2023 Published 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 > College of Computing
Depositing User:	Gemma Tonks
Date Deposited:	27 Apr 2024 12:31
Last Modified:	27 Apr 2024 12:31
URI:	https://www.open-access.bcu.ac.uk/id/eprint/15452

Actions (login required)

View Item