Skip to main navigation Skip to search Skip to main content

A GAN-based data augmentation method for mitigating class imbalance problem in histopathological image classification

Lingling Yuan, Md Rahaman, Hongzan Sun, Chen Li*, Yutong Gu, Tao Jiang, Marcin Grzegorzek, Xiaoyan Li

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceeding › Conference proceeding contribution › peer-review

Abstract

This study proposes a Generative Adversarial Network (GAN)-based data augmentation method to tackle the class imbalance issue in histopathological image classification. The proposed GAN generates high-quality images for minority classes, improving data diversity and classification performance. By integrating Global Context Attention (GCA) and UpBlock-CompRes modules, the GAN generates images that closely resemble real data, achieving superior results in FID, IS, PSNR, and SSIM compared to FastGAN. Experimental results demonstrate that the proposed GAN outperforms both the baseline without data augmentation and FastGAN, with noticeable improvements across key metrics. These enhancements highlight the effectiveness of our GAN in generating high-quality images and achieving more balanced classification results.

Original languageEnglish
Title of host publicationProceedings
Subtitle of host publication2024 IEEE International Conference on Bioinformatics and Biomedicine (BIBM)
EditorsMario Cannataro, Huiru (Jane) Zheng, Lin Gao, Jianlin (Jack) Cheng, Joao Luis de Miranda, Ester Zumpano, Xiaohua Hu, Young-Rae Cho, Taesung Park
Place of PublicationLisbon
PublisherInstitute of Electrical and Electronics Engineers (IEEE)
Pages5327-5334
Number of pages8
ISBN (Electronic)9798350386226
ISBN (Print)9798350386233
DOIs
Publication statusPublished - 2024
Externally publishedYes
Event2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024 - Lisbon, Portugal
Duration: 3 Dec 2024 → 6 Dec 2024

Publication series

NameIEEE International Conference on Bioinformatics and Biomedicine
PublisherIEEE
ISSN (Print)2156-1125
ISSN (Electronic)2156-1133

Conference

Conference2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
Country/TerritoryPortugal
CityLisbon
Period3/12/24 → 6/12/24

Keywords

  • class imbalance problem
  • data augmentation
  • GAN
  • histopathological image classification

Fingerprint

Dive into the research topics of 'A GAN-based data augmentation method for mitigating class imbalance problem in histopathological image classification'. Together they form a unique fingerprint.

Cite this