Optimizing GAN parameters for efficient and accurate image generation: A study of WGAN-GP in brain tumor dataset

Abstract

In this era of unprecedented rapid development of artificial intelligence, researchers are moving forward to develop new neural networks. Nevertheless, few have considered optimizing the extant generative adversarial network (GAN) to achieve the most effective and accurate solutions to provide optimal results for the challenges. This study aims to investigate the critical role of parameter optimization in GAN neural networks, with a particular focus on the Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) architecture applied to the generation of medical images, especially images depicting brain tumors. Therefore, the project used Kaggles brain tumor data set as a canvas to conduct an in-depth study of the impact of these parameters on the training model and generated results. For the starter, to investigate how alterations in the learning rate impact the model, this article selects a series of values for meticulous analysis to determine the most effective configuration. Then, evaluate and find a better and more suitable choice between Adam and SGD optimizers through comparison, focusing on their impact on training dynamics. As the last one, this study examines how the Tanh activation function constrains pixel values and shapes image realism through comparative results. By dissecting and understanding the interaction of these parameters in detail, we lay the foundation for optimizing GAN neural networks, increasing their efficiency, and producing accurate solutions for accurate diagnostics and healthcare applications. The journey through the labyrinth of GAN parameter tuning ultimately provided valuable insights into seamlessly synthesizing medical images.