1. Enhancing automated strabismus classification with limited data: Data augmentation using StyleGAN2-ADA.
- Author
-
Joo J, Kim SY, Kim D, Lee JE, Lee SM, Suh SY, Kim SJ, and Kim SC
- Subjects
- Humans, Algorithms, Deep Learning, Strabismus diagnosis, Strabismus classification
- Abstract
In this study, we propose a generative data augmentation technique to overcome the challenges of severely limited data when designing a deep learning-based automated strabismus diagnosis system. We implement a generative model based on the StyleGAN2-ADA model for system design and assess strabismus classification performance using two classifiers. We evaluate the capability of our proposed method against traditional data augmentation techniques and confirm a substantial enhancement in performance. Furthermore, we conduct experiments to explore the relationship between the diagnosis agreement among ophthalmologists and the generation performance of the generative model. Beyond FID, we validate the generative samples on the classifier to establish their practicality. Through these experiments, we demonstrate that the generative model-based data augmentation improves overall quantitative performance in scenarios of extreme data scarcity and effectively mitigates overfitting issues during deep learning model training., Competing Interests: No authors have competing interests., (Copyright: © 2024 Joo et al. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.)
- Published
- 2024
- Full Text
- View/download PDF