STM Journals Peer-reviewed research from Consortium e-Learning Network

Journal of Instrumentation Technology & Innovations Original Research

Early Autism Diagnosis: Machine Learning Models and Their Effectiveness

  1. Poonam Chaudhary Department of Computer Science and Engineering, The NorthCap University, Gurugram
  2. Anmol Bhatia Department of Computer Science and Engineering, The NorthCap University, Gurugram
  3. Vanshika Sangwan Department of Computer Science and Engineering, The NorthCap University, Gurugram
  4. Riya Saxena Department of Computer Science and Engineering, The NorthCap University, Gurugram
  5. Rahul Department of Computer Science and Engineering, The NorthCap University, Gurugram
  6. Sanyam Virmani Department of Computer Science and Engineering, The NorthCap University, Gurugram

Abstract

Diagnosis is of utmost importance for timely intervention and support. However, traditional diagnosis methods, which are based on subjective assessment, are delayed. This project explores the role that machine learning techniques might play in enhancing the accuracy and effectiveness of ASD detection. Several state-of-the-art classification algorithms were benchmarked using a dataset from Kaggle. Logistic Regression, XG Boost, Random Forest, Decision Tree, and Gradient Boosting were taken into consideration. Other performance measures, in terms of accuracy, F1-score, and precision, were considered. The results showed that XG Boost was the best model, because this one had the most precision and reliability of ASD prediction. The research signifies the potential of AI and ML technologies for the betterment of the diagnostic process and provides a robust and timely tool for early detection of ASD. Conclusions and recommendations of the study strongly emphasize the necessity of approaches that integrate multidisciplinary and ethical considerations for responsible translation into clinical practice.

Keywords

References (23)

  1. Thabtah, F. (2019). Machine Learning in Autism Spectrum Disorder Behavioral Research: A Review and Ways Forward. Informatics, 6(1), 4.
  2. Bone, D., Bishop, S., Black, M. P., Goodwin, M. S., & Lord, C. (2016). Use of machine learning to improve autism screening and diagnostic instruments: effectiveness, efficiency, and multi-instrument fusion. Journal of Child Psychology and Psychiatry, 57(8), 927-937.
  3. Heinsfeld, A. S., Franco, A. R., Craddock, R. C., Buchweitz, A., & Meneguzzi, F. (2018). Identification of autism spectrum disorder using deep learning and the ABIDE dataset. NeuroImage: Clinical, 17, 16-23.
  4. Thabtah, F., Peebles, D., & Seker, H. (2020). A Machine Learning Autism Classification based on Logistic Regression Analysis. Health Information Science and Systems, 8(1), 1-9.
  5. van ’t Hof et al., "Age at autism spectrum disorder diagnosis: A systematic review and meta-analysis from 2012 to 2019," Journal Title, vol. xx, no. xx, pp. xx-xx, Year.
  6. Duda, M., Ma, R., Haber, N., & Wall, D. P. (2016). Use of machine learning for behavioral distinction of autism and ADHD. Translational Psychiatry, 6(2), e732.
  7. L. Hyman et al., "Identification, Evaluation, and Management of Children With Autism Spectrum Disorder," Pediatrics, vol. 145, no. 1, pp. e20193447, Jan. 2020.
  8. A. Shaw et al., "Early Identification of Autism Spectrum Disorder Among Children Aged 4 Years — Early Autism and Developmental Disabilities Monitoring Network, Six Sites, United States, 2016," Surveillance Summaries, vol. 69, no. 3, pp. 1-11, Mar. 27, 2020.
  9. Li, M., Dai, Z., Wang, M., He, L., Huang, H., & Zheng, H. (2019). Predicting autism spectrum disorder based on multi-site MRI, clinical, and demographic data: a multisite machine learning study. Molecular Psychiatry, 24(10), 1599-1605.
  10. Lundberg, S. M., Erion, G. G., & Lee, S. I. (2020). Consistent Individualized Feature Attribution for Tree Ensembles. Nature Machine Intelligence, 2(5), 252-260.
  11. Chen, Q. Huang, L. Shi, Y. He, and Y. X. Li, "Associations between gut microbiota and autism spectrum disorder: a systematic review and meta-analysis," PLoS ONE, vol. 14, no. 9, pp. e0222907, 2019.
  12. P. Chen, C. P., Keown, C. L., Jahedi, A., & Muller, R. A. (2015). Deep learning in autism diagnosis and biomarker discovery. Journal of Autism and Developmental Disorders, 45(4), 1123-1136.
  13. Guang et al., "Synaptopathology Involved in Autism Spectrum Disorder," Journal Title, vol. xx, no. xx, pp. xx-xx, Year.
  14. Yechiam, O. Arshavsky, S. G. Shamay-Tsoory, S. Yaniv, and J. Aharon, "Adapted to explore: Reinforcement learning in Autistic Spectrum Conditions," Journal Title, vol. xx, no. xx, pp. xx-xx, Year.
  15. Bi, X., Wang, Z., Yang, Y., Gao, Y., & Xu, Y. (2018). A multi-feature learning model for early autism spectrum disorder diagnosis based on structural and functional MRI. Frontiers in Neuroscience, 12, 707.
  16. Chen, Q., Chen, X., Zhang, Y., & Ma, Q. (2020). Deep learning for neuroimaging-based diagnosis and rehabilitation of autism spectrum disorder: a review. Frontiers in Neuroscience, 14, 1007.
  17. L. Hyman et al., "Identification, Evaluation, and Management of Children With Autism Spectrum Disorder," Pediatrics, vol. 145, no. 1, pp. e20193447, Jan. 2020.
  18. Lord, E. H. Cook, B. L. Leventhal, and D. G. Amaral, "Autism Spectrum Disorders Review," Pediatrics, vol. xx, no. xx, pp. xx-xx, Year.
  19. Abraham, A., Milham, M. P., Di Martino, A., Craddock, R. C., Samaras, D., Thirion, B., & Varoquaux, G. (2017). Deriving reproducible biomarkers from multi-site resting-state data: An Autism-based example. NeuroImage, 147, 736-745.
  20. Hazlett, H. C., Gu, H., Munsell, B. C., Kim, S. H., Styner, M., Wolff, J. J., & Piven, J. (2017). Early brain development in infants at high risk for autism spectrum disorder. Nature, 542(7641), 348-351.
  21. Khosla, M., Jamison, K., Ngo, G. H., Kuceyeski, A., & Sabuncu, M. R. (2019). Machine learning in resting-state fMRI analysis. Magnetic Resonance Imaging, 64, 101-121.
  22. Kuwabara, H., Lu, J., Lim, L., Irie, H., & Lopez, P. T. (2016). Machine learning approaches for predicting autism spectrum disorder diagnosis using facial imaging features. Journal of Child Psychology and Psychiatry, 57(8), 927-937.
  23. Vabalas, A., Gowen, E., Poliakoff, E., & Casson, A. J. (2019). Machine learning algorithm validation with a limited sample size. PLoS One, 14(11), e0224365.
Support