Abstract
In recent AI-driven disease diagnosis, the success of models has depended mainly on extensive data sets and advanced algorithms. However, creating traditional data sets for rare or emerging diseases presents significant challenges. To address this issue, this study introduces a direct-self-attention Wasserstein generative adversarial network (DSAWGAN) designed to improve diagnostic capabilities in infectious diseases with limited data availability. DSAWGAN enhances convergence speed, stability, and image quality by integrating attention modules and leveraging the Wasserstein distance optimization. We compared DSAWGAN-generated images with traditional data augmentation and other image generation techniques, evaluating their effectiveness using classification neural networks for diagnostic accuracy. This model integration was then applied to a mobile app, enabling rapid, portable, and cost-effective diagnostic testing across various concentration ranges. Using only half of the raw data (n = 1500), DSAWGAN achieves an accuracy increase from 98.00 to 99.33%. Even with just 10% of the original data (n = 300), a neural network trained with the augmented data set maintains an accuracy of 92.67%, demonstrating the approach's effectiveness in resource-limited settings.