Supporting ASD Diagnosis with EEG, ML and Swarm Intelligence: Early Detection of Autism Spectrum Disorder Based on Electroencephalography Analysis by Machine Learning and Swarm Intelligence

Deficits in social interaction and communication characterize Autism Spectrum Disorder (ASD). Although widely recognized by its symptoms, diagnosing ASD remains challenging due to its wide range of clinical presentations. Methods: In this study, we propose a method to assist in the early diagnosis of autism, which is currently primarily based on clinical assessments. Our approach aims to develop an early differential diagnosis based on electroencephalogram (EEG) signals, seeking to identify patterns associated with ASD. In this study, we used EEG data from 56 participants obtained from the Sheffield dataset, including 28 individuals diagnosed with Autism Spectrum Conditions (ASC) and 28 neurotypical controls, applying numerical techniques to handle missing data. Subsequently, after a detailed analysis of the signals, we applied three different starting approaches: one with the original database and the other two with selection of the most significant attributes using the PSO and evolutionary search methods. In each of these approaches, we applied a series of machine learning models, where relatively high performances for classification were observed. Results: We achieved accuracies of 99.13% ± 0.44 for the dataset with original signals, 99.23% ± 0.38 for the dataset after applying PSO, and 93.91% ± 1.10 for the dataset after the evolutionary search methodology. These results were obtained using classical classifiers, with SVM being the most effective among the first two approaches, while Random Forest with 500 trees proved more efficient in the third approach. Conclusions: Even with all the limitations of the base, the results of the experiments demonstrated promising findings in identifying patterns associated with Autism Spectrum Disorder through the analysis of EEG signals. Finally, we emphasize that this work is the starting point for a larger project with the objective of supporting and democratizing the diagnosis of ASD both in children early and later in adults.