MedGAN: An adaptive GAN approach for medical image generation
Kehua Guo,Jie Chen,Tian Qiu,Shaojun Guo,Tao Luo,Tianyu Chen,Sheng Ren
DOI: https://doi.org/10.1016/j.compbiomed.2023.107119
IF: 7.7
2023-06-13
Computers in Biology and Medicine
Abstract:Generative adversarial networks (GANs) and their variants as an effective method for generating visually appealing images have shown great potential in different medical imaging applications during past decades. However, one major issue remains insufficiently investigated: many models still suffer from model collapse, vanishing gradients, and convergence issues. Considering the fact that medical images differ from typical RGB images in terms of complexity and dimensionality, we propose an adaptive generative adversarial network, namely MedGAN, to mitigate these issues. Specifically, we first use Wasserstein loss as a convergence metric to measure the convergence degree of the generator and the discriminator. Then, we adaptively train MedGAN based on this metric. Finally, we generate medical image samples based on MedGAN and use them to build few-shot medical data learning models for disease classification and lesion localization. On demodicosis, blister, molluscum, and parakeratosis datasets, experimental results verify the advantages of MedGAN in model convergence, training speed, and visual quality of generated samples. We believe this approach can be generalized to other medical classification applications and contribute to radiologists' efforts for disease diagnosis. The source code can be downloaded at https://github.com/geyao-c/MedGAN .
engineering, biomedical,computer science, interdisciplinary applications,mathematical & computational biology,biology