Synthetic Electroretinogram Signal Generation Using Conditional Generative Adversarial Network for Enhancing Classification of Autism Spectrum Disorder

Mikhail Kulyabin,Paul A. Constable,Aleksei Zhdanov,Irene O. Lee,David H. Skuse,Dorothy A. Thompson,Andreas Maier
2024-07-11
Abstract:The electroretinogram (ERG) is a clinical test that records the retina's electrical response to light. The ERG is a promising way to study different neurodevelopmental and neurodegenerative disorders, including autism spectrum disorder (ASD) - a neurodevelopmental condition that impacts language, communication, and reciprocal social interactions. However, in heterogeneous populations, such as ASD, where the ability to collect large datasets is limited, the application of artificial intelligence (AI) is complicated. Synthetic ERG signals generated from real ERG recordings carry similar information as natural ERGs and, therefore, could be used as an extension for natural data to increase datasets so that AI applications can be fully utilized. As proof of principle, this study presents a Generative Adversarial Network capable of generating synthetic ERG signals of children with ASD and typically developing control individuals. We applied a Time Series Transformer and Visual Transformer with Continuous Wavelet Transform to enhance classification results on the extended synthetic signals dataset. This approach may support classification models in related psychiatric conditions where the ERG may help classify disorders.
Machine Learning,Artificial Intelligence,Signal Processing
What problem does this paper attempt to address?
This paper mainly explores how to use Conditional Generative Adversarial Network (CGAN) to generate synthetic Electroretinogram (ERG) signals for enhancing the classification of Autism Spectrum Disorder (ASD). ERG is a clinical test that records the electrical response of the retina to light and has the potential for studying neurodevelopmental and neurodegenerative diseases, including ASD. However, the application of artificial intelligence is limited in heterogeneous populations like ASD due to difficulties in collecting large amounts of data. The paper proposes a method to generate synthetic signals similar to real ERG signals using CGAN, which can expand the natural data set and improve the performance of the classification model. The research team applies Time Series Transformer (TST) and Visual Transformer (ViT), combined with Continuous Wavelet Transform (CWT), to classify these synthetic signals. Experimental results show that this method improves the accuracy of ASD and control group classification. In addition, the paper discusses various factors that affect ERG signals, such as patient age, sex, and pupil diameter, and points out that signal analysis advantages in other biological signals, such as ECG, have not been widely used in the field of ophthalmology. Lastly, the paper suggests that synthetic ERG signals may contribute to ophthalmic research, including balancing case and control groups in clinical trials, and promoting the application of ERG in the early identification and classification of neurological diseases.