Back to Search Start Over

Enhancing automated strabismus classification with limited data: Data augmentation using StyleGAN2-ADA.

Authors :
Joo, Jaehan
Kim, Sang Yoon
Kim, Donghwan
Lee, Ji-Eun
Lee, Seung Min
Suh, Su Youn
Kim, Su-Jin
Kim, Suk Chan
Source :
PLoS ONE; 5/25/2024, Vol. 19 Issue 5, p1-14, 14p
Publication Year :
2024

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. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
19326203
Volume :
19
Issue :
5
Database :
Complementary Index
Journal :
PLoS ONE
Publication Type :
Academic Journal
Accession number :
177468120
Full Text :
https://doi.org/10.1371/journal.pone.0303355