Title: Explainable ensemble machine learning model for autism identification using EEG and optimised feature selection
Authors: Anamika Ranaut; Padmavati Khandnor; Trilok Chand
Addresses: Department of Computer Science and Engineering, Punjab Engineering College (Deemed to be University), Chandigarh, India ' Department of Computer Science and Engineering, Punjab Engineering College (Deemed to be University), Chandigarh, India ' Department of Computer Science and Engineering, Punjab Engineering College (Deemed to be University), Chandigarh, India
Abstract: Autism spectrum disorder (ASD) is a complex neurodevelopmental condition that remains challenging to diagnose due to its reliance on subjective clinical assessments. To address this limitation, an automated electroencephalography (EEG)-based ASD identification framework is proposed, aiming to enhance identification accuracy. Temporal dynamics of EEG signals analysed through three-phase feature selection approach involving independent samples t-test elimination, followed by a novel binary search-driven mutual information (BiS-MI) and recursive feature elimination (BiS-RFE). Six ensemble models trained on the selected features, and feature importance and classification predictions interpreted using explainable artificial intelligence (XAI). CatBoost achieved the highest performance, with an accuracy of 0.9923 and recall of 0.9864 using BiS-MI, and an accuracy of 0.9936 and recall of 0.9911 using BiS-RFE. SHAP analysis identified features from frontal and central EEG electrodes as the most significant contributors. These results highlight the potential for developing interpretable, non-invasive and improved diagnostic tools for ASD identification.
Keywords: autism; electroencephalography; ensemble models; feature selection; feature extraction; mutual information; recursive feature elimination; statistical test.
DOI: 10.1504/IJBET.2025.150082
International Journal of Biomedical Engineering and Technology, 2025 Vol.49 No.3, pp.202 - 232
Received: 27 Apr 2025
Accepted: 07 Jul 2025
Published online: 28 Nov 2025 *