Identifying autism using EEG: unleashing the power of feature selection and machine learning

Abstract

Autism Spectrum Disorder (ASD) is a neurodevelopmental condition that is characterized by communication barriers, societal disengagement, and monotonous actions. Currently, the diagnosis of ASD is made by experts through a subjective and time-consuming qualitative behavioural examination using internationally recognized descriptive standards. In this paper, we present an EEG-based three-phase novel approach comprising 29 autistic subjects and 30 neurotypical people. In the first phase, preprocessing of data is performed from which we derived one continuous dataset and four condition-based datasets to determine the role of each dataset in the identification of autism from neurotypical people. In the second phase, time-domain and morphological features were extracted and four different feature selection techniques were applied. In the last phase, five-fold cross-validation is used to evaluate six different machine learning models based on the performance metrics and computational efficiency. The neural network outperformed when trained with maximum relevance and minimum redundancy (MRMR) algorithm on the continuous dataset with 98.10% validation accuracy and 0.9994 area under the curve (AUC) value for model validation, and 98.43% testing accuracy and AUC test value of 0.9998. The decision tree overall performed the second best in terms of computational efficiency and performance accuracy. The results indicate that EEG-based machine learning models have the potential for ASD identification from neurotypical people with a more objective and reliable method.