Enhancing Deep Learning for Autism Spectrum Disorder Detection with Dual-Encoder GAN-based Augmentation of Electroencephalogram Data

Authors

DOI:

https://doi.org/10.56294/sctconf2024958

Keywords:

Autism Spectrum Disorder, EEG, Rest-HGCN, Class Imbalance, Data Augmentation, Differential Entropy, Dual-Encoder, Wasserstein GAN

Abstract

Autism Spectrum Disorder (ASD) is a general neurodevelopmental condition that requires early and accurate diagnosis. Electroencephalography (EEG) signals are reliable biomarkers for ASD detection and diagnosis. A recent Deep Learning (DL) model called Resting-state EEG-based Hybrid Graph Convolutional Network (Rest-HGCN) has been developed for this purpose. However, a challenge in ASD diagnosis is the limited availability of EEG data, leading to imbalanced classes and ineffective model training. To address this issue, a new approach is proposed in this paper, which involves a generative model for EEG data augmentation. A novel Dual Encoder-Balanced Conditional Wasserstein Generative Adversarial Network (DEBCWGAN) is designed to produce fine synthetic minority-class EEG examples and augment the original training dataset. This model integrates the Variational Auto-Encoder (VAE) and balanced conditional Wasserstein GAN. Initially, EEG signals for ASD in the training dataset are pre-processed as Differential Entropy (DE) features and split into different segments. Each feature segment is processed in the temporal and the spatial domain depending on the electrode place. Then, twin encoders are trained to capture both spatial and temporal information from these features, concatenate them as Latent Variables (LVs), and provide them to the decoder to produce synthetic EEG examples. Additionally, gradient penalty and L2 regularization are used to speed up convergence and prevent overfitting effectively. Further, the augmented dataset is used to train the Rest-HGCN for ASD detection, enhancing its robustness and generalizability. Finally, test outcomes demonstrate that the DEBWGAN-GP-Rest-HGCN on the EEG Dataset for ASD and ABC-CT dataset achieves 91,6 % and 88,1 % accuracy, respectively compared to the Rest-HGCN, AlexNet, K-Nearest Neighbor (KNN) and Support Vector Machine (SVM)

References

1. Hadders‐Algra M. Emerging signs of autism spectrum disorder in infancy: Putative neural substrate. Dev Med Child Neurol. 2022;64(11):1344-50.

2. Riglin L, Wootton RE, Thapar AK, Livingston LA, Langley K, Collishaw S, et al. Variable emergence of autism spectrum disorder symptoms from childhood to early adulthood. Am J Psychiatry. 2021;178(8):752-60.

3. McCarty P, Frye RE. Early detection and diagnosis of autism spectrum disorder: Why is it so difficult? Semin Pediatr Neurol. 2020;35:100831.

4. Kaufman NK. Rethinking “gold standards” and “best practices” in the assessment of autism. Appl Neuropsychol Child. 2022;11(3):529-40.

5. Iadarola S, Pellecchia M, Stahmer A, Lee HS, Hauptman L, Hassrick EM, et al. Mind the gap: An intervention to support caregivers with a new autism spectrum disorder diagnosis is feasible and acceptable. Pilot Feasibility Stud. 2020;6:1-13.

6. Lalli, K., & Senbagavalli, M. (2023). Identification of Biomarker for Autism Spectrum Disorder Using Eeg: A Review. Proceedings-2023 International Conference on Advanced Computing and Communication Technologies, ICACCTech 2023.

7. Haweel R, AbdElSabour Seada N, Ghoniemy S, ElBaz A. A review on autism spectrum disorder diagnosis using task-based functional MRI. Int J Intell Comput Inf Sci. 2021;21(2):23-40.

8. Moridian P, Ghassemi N, Jafari M, Salloum-Asfar S, Sadeghi D, Khodatars M, et al. Automatic autism spectrum disorder detection using artificial intelligence methods with MRI neuroimaging: A review. Front Mol Neurosci. 2022;15:999605.

9. Kurkin S, Smirnov N, Pitsik E, Kabir MS, Martynova O, Sysoeva O, et al. Features of the resting-state functional brain network of children with autism spectrum disorder: EEG source-level analysis. Eur Phys J Spec Top. 2023;232(5):683-93.

10. Stoyell SM, Wilmskoetter J, Dobrota MA, Chinappen DM, Bonilha L, Mintz M, et al. High-density EEG in current clinical practice and opportunities for the future. J Clin Neurophysiol. 2021;38(2):112-23.

11. Kohli M, Kar AK, Sinha S. The role of intelligent technologies in early detection of autism spectrum disorder (ASD): A scoping review. IEEE Access. 2022;10:104887-913.

12. Jui SJJ, Deo RC, Barua PD, Devi A, Soar J, Acharya UR. Application of entropy for automated detection of neurological disorders with electroencephalogram signals: A review of the last decade (2012-2022). IEEE Access. 2023;11:71905-24.

13. Khodatars M, Shoeibi A, Sadeghi D, Ghaasemi N, Jafari M, Moridian P, et al. Deep learning for neuroimaging-based diagnosis and rehabilitation of autism spectrum disorder: A review. Comput Biol Med. 2021;139:104949.

14. Ahmedt-Aristizabal D, Armin MA, Denman S, Fookes C, Petersson L. Graph-based deep learning for medical diagnosis and analysis: Past, present and future. Sensors. 2021;21(14):4758.

15. Tang T, Li C, Zhang S, Chen Z, Yang L, Mu Y, et al. A hybrid graph network model for ASD diagnosis based on resting-state EEG signals. Brain Res Bull. 2024;206:110826.

16. Kang J, Han X, Song J, Niu Z, Li X. The identification of children with autism spectrum disorder by SVM approach on EEG and eye-tracking data. Comput Biol Med. 2020;120:103722.

17. Oh SL, Jahmunah V, Arunkumar N, Abdulhay EW, Gururajan R, Adib N, et al. A novel automated autism spectrum disorder detection system. Complex Intell Syst. 2021;7(5):2399-413.

18. Alturki FA, Aljalal M, Abdurraqeeb AM, Alsharabi K, Al-Shamma’a AA. Common spatial pattern technique with EEG signals for diagnosis of autism and epilepsy disorders. IEEE Access. 2021;9:24334-49.

19. Baygin M, Dogan S, Tuncer T, Barua PD, Faust O, Arunkumar N, et al. Automated ASD detection using hybrid deep lightweight features extracted from EEG signals. Comput Biol Med. 2021;134:104548.

20. Tawhid MNA, Siuly S, Wang H, Whittaker F, Wang K, Zhang Y. A spectrogram image based intelligent technique for automatic detection of autism spectrum disorder from EEG. PLoS One. 2021;16(6).

21. Chavez-Cano AM. Artificial Intelligence Applied to Telemedicine: opportunities for healthcare delivery in rural areas. LatIA 2023;1:3-3. https://doi.org/10.62486/latia20233.

22. Sinha T, Munot MV, Sreemathy R. An efficient approach for detection of autism spectrum disorder using electroencephalography signal. IETE J Res. 2022;68(2):824-32.

23. Lalli, K., Shrivastava, V. K., & Shekhar, R. (2023, April). Detecting Copy Move Image Forgery using a Deep Learning Model: A Review. In 2023 International Conference on Artificial Intelligence and Applications (ICAIA) Alliance Technology Conference (ATCON-1) (pp. 1-7). IEEE.

24. Liao M, Duan H, Wang G. Application of machine learning techniques to detect the children with autism spectrum disorder. J Healthc Eng. 2022;2022:1-10.

25. Menaka R, Karthik R, Saranya S, Niranjan M, Kabilan S. An improved AlexNet model and cepstral coefficient-based classification of autism using EEG. Clin EEG Neurosci. 2024;55(1):43-51.

26. Dickinson A, Jeste S, Milne E. Electrophysiological signatures of brain aging in autism spectrum disorder. Cortex. 2022;148:139-51.

27. McPartland JC, Bernier RA, Jeste SS, Dawson G, Nelson CA, Chawarska K, et al. The autism biomarkers consortium for clinical trials (ABC-CT): Scientific context, study design, and progress toward biomarker qualification. Front Integr Neurosci. 2020;14:16.

28. Suárez YS, Laguardia NS. Trends in research on the implementation of artificial intelligence in supply chain management. LatIA 2023;1:6-6. https://doi.org/10.62486/latia20236.

29. Senbagavalli, M. (2022). Artificial Intelligence and Medical Information Modeling. In Handbook of Research on Mathematical Modeling for Smart Healthcare Systems (pp. 1-11). IGI Global.

30. Larsen ABL, Sønderby SK, Larochelle H, Winther O. Autoencoding beyond pixels using a learned similarity metric. In: International Conference on Machine Learning; 2016. p. 1558-66.

Downloads

Published

2024-01-01

How to Cite

1.
Lalli K, Senbagavalli M. Enhancing Deep Learning for Autism Spectrum Disorder Detection with Dual-Encoder GAN-based Augmentation of Electroencephalogram Data. Salud, Ciencia y Tecnología - Serie de Conferencias [Internet]. 2024 Jan. 1 [cited 2024 Dec. 2];3:958. Available from: https://conferencias.ageditor.ar/index.php/sctconf/article/view/831