Shan, Md. Abdul Kahhar SiddikiQuaiyum, Md. AbdulSaha, SugataAnik, S. M. Navin NayerRafin, Nafiz ImtiazAlam, Md. Golam Rabiul2026-10-052026-10-052025-01-01M. A. K. S. Shan, M. A. Quaiyum, S. Saha, S. M. N. N. Anik, N. I. Rafin and M. G. R. Alam, "Optimal Transport Theory-Based GAN for Medical Image Augmentation and Classification," 2025 28th International Conference on Computer and Information Technology (ICCIT), Cox's Bazar, Bangladesh, 2025, pp. 389-394, doi: 10.1109/ICCIT68739.2025.11491184.97983315786712-s2.0-105041610907https://hdl.handle.net/10361/30419Generative Adversarial Networks (GANs) can create realistic images without relying on original data, which is particularly useful in addressing class imbalance in medical imaging due to disparities in illness occurrence. GAN-based methods have been proposed to tackle this issue, but they often struggle with small-scale disorders due to the challenge of extracting significant information from few pixels. To combat data scarcity and imbalance, data augmentation techniques, like image modifications, are employed to maintain performance. Optimal Transport (OT) theory aids in defining distance in function spaces, interpolating functions, and establishing barycenters of weighted functions. However, using OT in generative models poses challenges such as lack of smoothness, computational burden, and difficulty in estimating high-dimensional gradients. To overcome these, we propose using Sinkhorn divergence to create a loss family that combines Wasserstein (OT) and Maximum Mean Discrepancy (MMD) losses, leveraging OT's geometry for generating highdimensional spaces from low-dimensional manifolds. This approach is suitable for image augmentation and classification, ensuring the generated images are physically and aesthetically convincing. Our Optimal Transport Theory-based GAN model excels in detecting and differentiating chest X-ray images, achieving 96.94% accuracy in classifying normal, COVID-19-affected, lung opacity, and viral pneumonia images. This performance surpasses CNN models used in our research, demonstrating superior accuracy, precision, recall, and F1 score on the available datasets. The integration of OT and GANs represents a significant advancement in medical imaging, addressing critical challenges and improving diagnostic capabilities.6 Pagesen-USCircuitsMillimeter wave integrated circuitsMonolithic integrated circuitsCircuits and systemsOptimal Transport (OT) theoryGenerative Adversarial Networks (GANs)Unsupervised neural networksWasserstein distanceMaximum Mean Discrepancy(MMD)Chest X-RayCovid-19Viral pneumoniaLung opacityChest--Diseases--Diagnosis.Chest--Radiography.Generative adversarial networks (Computer networks).Optimal transport theory-based GAN for medical image augmentation and classificationConference Proceeding10.1109/ICCIT68739.2025.11491184