A GAN-Based Data Augmentation Method for Mitigating Class Imbalance Problem in Histopathological Image Classification
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.
Authors
- Hongzan Sun (ORCID: https://orcid.org/0000-0002-4724-5828)
- Marcin Grzegorzek (ORCID: https://orcid.org/0000-0003-4877-8287)
- Lingling Yuan
- Md Mamunur Rahaman (ORCID: https://orcid.org/0000-0003-2268-2092)
- Chen Li (ORCID: https://orcid.org/0000-0002-0260-2182)
- Xiaoyan Li (ORCID: https://orcid.org/0000-0001-5851-2912)
- Tao Jiang (ORCID: https://orcid.org/0000-0001-8296-3027)
- Yutong Gu
Institutions
- UNSW Sydney (AU)
- Chengdu University of Traditional Chinese Medicine (CN)
- China Medical University (CN)
- Northeastern University (CN)
- University of Lübeck (DE)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10822517
- Citations
- 1
- Primary Topic
- Imbalanced Data Classification Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.35
Funders
- National Natural Science Foundation of China
- China Medical University