Age Classification by WGAN Brain MR Image Augmentation

Abstract

Medical image augmentation plays a crucial role in enhancing the performance of Artificial Intelligence (AI) applications in medical sciences. Augmenting medical images is important for solving data scarcity, increasing data diversity, enhancing robustness and reliability of model and improving training and test results that can be done in medical sciences. In this work we show that Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) can be used for increasing the performance of data classification. To achieve that, we have augmented healthy brain MR images by using WGAN and updated the dataset. The results give that when dataset augmented by WGAN-GP is used as input for CNN-based model to solve age classification problem, accuracy of this model increases to 98,37% from 95,14%. It can be concluded that the purposed WGAN-based brain MR image augmentation method enhances the performance of image classification.