Optimized Machine Learning Techniques for Accurate Autism Spectrum Disorder Diagnosis
Keywords:
Autism Spectrum Disorder, machine learning, predictive modelling, data analysis, feature engineering, classification, neural networks, explainable AI, early diagnosis, autism screening.Abstract
Autism Spectrum Disorder (ASD) is a neurodevelopmental disorder characterized by deficits in communication, social interaction, and behavior. Early diagnosis and prompt intervention can help to maximize developmental outcomes. This work proposes a refined predictive ASD model using other advanced techniques of machine learning. We leverage two datasets—“Autism Spectrum Disorder Screening Data for Toddlers in Saudi Arabia” and the “Autism Prediction Dataset”—to develop a robust predictive model. The methodology includes extensive data preprocessing, feature engineering, and algorithm selection to improve classification performance. Various machine learning models, including Random Forest, Support Vector Machine (SVM), Gradient Boosting, and Neural Networks, are employed and evaluated based on accuracy, precision, recall, and F1-score. The study also integrates explainable AI (XAI) techniques to interpret model predictions, enhancing transparency in decision-making. Experimental results demonstrate that the proposed approach significantly improves the predictive accuracy of ASD detection compared to traditional methods. The findings suggest that AI-driven diagnostic tools can assist healthcare professionals in making more informed decisions, ultimately aiding in early intervention strategies. Future work includes expanding the dataset and refining models for real-world applications.
References
[1] S. S. Rajagopalan, Y. Zhang, A. Yahia, and K. Tammimies, “Machine Learning Prediction of Autism Spectrum Disorder From a Minimal Set of Medical and Background Information,” JAMA Netw Open, vol. 7, no. 8, pp. e2429229–e2429229, Aug. 2024, doi: 10.1001/JAMANETWORKOPEN.2024.29229.
[2] U. J. Ganai, A. Ratne, B. Bhushan, and K. S. Venkatesh, “Early detection of autism spectrum disorder: gait deviations and machine learning,” Scientific Reports 2025 15:1, vol. 15, no. 1, pp. 1–16, Jan. 2025, doi: 10.1038/s41598-025-85348-w.
[3] Y. Ding, H. Zhang, and T. Qiu, “Deep learning approach to predict autism spectrum disorder: a systematic review and meta-analysis,” BMC Psychiatry, vol. 24, no. 1, p. 739, Dec. 2024, doi: 10.1186/S12888-024-06116-0.
[4] J. Song et al., “Combining Radiomics and Machine Learning Approaches for Objective ASD Diagnosis: Verifying White Matter Associations with ASD”.
[5] S. Rubio-Martín, M. T. García-Ordás, M. Bayón-Gutiérrez, N. Prieto-Fernández, and J. A. Benítez-Andrades, “Enhancing ASD detection accuracy: a combined approach of machine learning and deep learning models with natural language processing,” Health Inf Sci Syst, vol. 12, no. 1, p. 20, Mar. 2024, doi: 10.1007/s13755-024-00281-y.
[6] V. Ramesh and R. Assaf, “Detecting Autism Spectrum Disorders with Machine Learning Models Using Speech Transcripts,” Oct. 2021, Accessed: Feb. 07, 2025. [Online]. Available: https://arxiv.org/abs/2110.03281v1
[7] R. A. Bahathiq, H. Banjar, A. K. Bamaga, and S. K. Jarraya, “Machine learning for autism spectrum disorder diagnosis using structural magnetic resonance imaging: Promising but challenging,” Front Neuroinform, vol. 16, p. 949926, Sep. 2022, doi: 10.3389/FNINF.2022.949926/BIBTEX.
[8] S. S. Rajagopalan, Y. Zhang, A. Yahia, and K. Tammimies, “Machine Learning Prediction of Autism Spectrum Disorder From a Minimal Set of Medical and Background Information,” JAMA Netw Open, vol. 7, no. 8, pp. e2429229–e2429229, Aug. 2024, doi: 10.1001/JAMANETWORKOPEN.2024.29229.
[9] Z. Yin et al., “Early Autism Diagnosis based on Path Signature and Siamese Unsupervised Feature Compressor,” Cerebral Cortex, vol. 34, no. 13, pp. 72–83, Jul. 2023, doi: 10.1093/cercor/bhae069.
[10] S. S. Rajagopalan, Y. Zhang, A. Yahia, and K. Tammimies, “Machine Learning Prediction of Autism Spectrum Disorder From a Minimal Set of Medical and Background Information,” JAMA Netw Open, vol. 7, no. 8, pp. e2429229–e2429229, Aug. 2024, doi: 10.1001/JAMANETWORKOPEN.2024.29229.
Downloads
Published
Issue
Section
License

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Articles in this journal are licensed under the Creative Commons Attribution-NonCommercial 4.0 International License. This license permits others to copy, distribute, and adapt the work, provided it is for non-commercial purposes, and the original author and source are properly credited.