Autism Spectrum Disorder Identification using Machine Learning with Hybrid Feature Selection and Optuna Hyperparameter Optimization with SHAP Explainability

Abstract

Autism Spectrum Disorder (ASD) is a neurodevelopmental disorder which necessitates early and precise diagnosis to facilitate prompt intervention and enhance long-term results. Using behavioural and demographic data, this research offers a machine learningbased method for identifying ASD. The proposed system incorporates comprehensive data preprocessing techniques, such as addressing missing values, encoding categorical features, and feature scaling, to increase model efficiency. Additionally, the most pertinent characteristics that contribute to the classification of ASD are found using feature selection techniques. The predictive power of four machine learning (ML) algorithms: Support Vector Machine (SVM), Random Forest (RF), Cat-Boost (CB), and Light Gradient Boosting Machine (LGBM) is assessed through implementation. Following training and validation using appropriate data partitioning methodologies, the model's performance is evaluated using different performance metrics. A comparative study is undertaken to find out how well every model represents complex trends in the dataset. According to the experimental results, boosting-based methods in particular, Cat-Boost and LGBM perform better because they can effectively handle categorical variables and minimise overfitting. Cat-Boost achieved best results with an accuracy of 85.14%. This study shows how cutting-edge machine learning methods can improve ASD screening and assist with data-driven clinical decision-making.