Perspective Transformation and Viewpoint Attention Enhancement for Generative Adversarial Networks in Endoscopic Image Augmentation
Laimonas Janutėnas, Dmitrij Šešok · Applied Sciences · 2025
This study presents an enhanced version of the StarGAN model, with a focus on medical applications, particularly endoscopic image augmentation. Our model incorporates novel Perspective Transformation and Viewpoint Attention Modules for StarGAN that improve image classification accuracy in a multiclass classification task. The Perspective Transformation Module enables the generation of more diverse viewing angles, while the Viewpoint Attention Module helps focus on diagnostically significant regions. We evaluate the performance of our enhanced architecture using the Kvasir v2 dataset, which contains 8000 images across eight gastrointestinal disease classes, comparing it against baseline models including VGG-16, ResNet-50, DenseNet-121, InceptionNet-V3, and EfficientNet-B7. Experimental results demonstrate that our approach achieves better performance in all models for this eight-class classification problem, increasing accuracy on average by 0.7% on VGG-16 and 0.63% on EfficientNet-B7 models. The addition of perspective transformation capabilities enables more diverse examples to augment the database and provide more samples of specific illnesses. Our approach offers a promising solution for medical image generation, enabling effective training with fewer data samples, which is particularly valuable in medical model development where data are often scarce due to challenges in acquisition. These improvements demonstrate significant potential for advancing machine learning disease classification systems in gastroenterology and medical image augmentation as a whole.