Lung cancer diagnosis using Hessian adaptive learning optimization in generative adversarial networks

E. Thirumagal,K. Saruladha
DOI: https://doi.org/10.1007/s00500-023-07877-8
IF: 3.732
2023-02-08
Soft Computing
Abstract:Lung cancer is the most frequent cancer and the reason for cancer death, with high morbidity and mortality. Computed tomography is one of the efficient medical imaging tools for lung cancer diagnosis, which offers internal lung details. However, as there is limited availability of datasets and requires large number of images for interpretation, it is hard for radiologists to diagnose lung cancer. The generative adversarial network (GAN) is a significant generative model employed for data augmentation, which has the benefit of simulating data distribution without the explicit modeling of potential probability density functions. Despite the benefits of GAN, training process remains challenging due to high convergence time and mode collapse problems. To resolve these issues, in this paper, a Hessian Adaptive Learning (HAL) Optimization technique. The proposed HL technique uses gradient and curvature data to eliminate the mode collapse problem and to improve the dataset size via a generation of diverse images. The experiments were conducted on Vanilla GAN, Wasserstein Generative Adversarial Network (WGAN), Conditional Generative Adversarial Network (CGAN), Wasserstein, and Deep Convolutional Generative Adversarial Network (DCGAN). Each GAN is tested using stochastic Gradient descent (SGD), Gauss–Newton (GN) Second-order learning, and proposed HAL optimization techniques. The experimental outcomes prove that the GANs with HAL optimization technique yields better performance compared to SGD and GN models. The experimental results assured that the GANs converge fast and eliminate mode collapse problems using HAL optimization.
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?