Skip to main navigation menu Skip to main content Skip to site footer
×
Español (España) | English
Editorial
Home
Indexing
Original

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

By
K Lalli ,
K Lalli

Alliance College of Engineering and Design, Alliance University, Bangalore, India

Search this author on:

PubMed | Google Scholar
Senbagavalli.M ,
Senbagavalli.M

Alliance College of Engineering and Design, Alliance University, Bangalore, India

Search this author on:

PubMed | Google Scholar

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).

How to Cite

1.
Lalli K, M S. 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 Jun. 30 [cited 2024 Jul. 19];3:958. Available from: https://conferencias.saludcyt.ar/index.php/sctconf/article/view/958

The article is distributed under the Creative Commons Attribution 4.0 License. Unless otherwise stated, associated published material is distributed under the same licence.

Article metrics

Google scholar: See link

Metrics

Metrics Loading ...

The statements, opinions and data contained in the journal are solely those of the individual authors and contributors and not of the publisher and the editor(s). We stay neutral with regard to jurisdictional claims in published maps and institutional affiliations.