A Comparative Analysis of Machine Learning Models for Prediction of Autism Spectrum Disorder Using Screening Data

Authors

  • Ming Yue Yeap Faculty of Computer Science and Information Technology, Universiti Malaysia Sarawak (UNIMAS), 94300 Kota Samarahan, Sarawak, Malaysia
  • Stephanie Chua Faculty of Computer Science and Information Technology, Universiti Malaysia Sarawak (UNIMAS), 94300 Kota Samarahan, Sarawak, Malaysia
  • Arif Bramantoro School of Computing and Informatics, Universiti Teknologi Brunei, Jalan Tungku Link, Gadong BE1410, Brunei Darussalam

DOI:

https://doi.org/10.37934/araset.53.1.175185

Keywords:

Autism spectrum disorder, Classification, Machine learning, Prediction

Abstract

Autism spectrum disorder (ASD) is a neurological and developmental disorder that affects how people interact with others, communicate, learn, and behave. ASD prediction is difficult because the diagnostic factors may not be based solely on observation. In this research paper, an in-depth comparative analysis of various machine learning models applied to the task of classifying autism traits was presented. Our study aimed to assess the performance of these models within the context of identifying individuals with autism based on relevant features and data. The machine learning models investigated in this study encompassed Logistic Regression (LR), Random Forest (RF), Support Vector Machine (SVM), k-Nearest Neighbours (KNN), Naive Bayes (NB), and Neural Network (NN). The models were evaluated using six essential classification metrics: accuracy, precision, recall, specificity, F1 score, and AUC score. On the training dataset, our results reveal nuanced performance characteristics. SVM and RF excel in precision and recall, showing promise for accurate autism trait classification. KNN exhibits remarkable specificity, suggesting its potential for minimizing false positives. LR and NB demonstrate balanced performance across multiple metrics, while NN exhibits high precision and recall, albeit with higher computational demands. It was concluded that SVM was the best classification model for autism trait classification.

Downloads

Download data is not yet available.

Author Biographies

Ming Yue Yeap, Faculty of Computer Science and Information Technology, Universiti Malaysia Sarawak (UNIMAS), 94300 Kota Samarahan, Sarawak, Malaysia

71997@siswa.unimas.my

Stephanie Chua, Faculty of Computer Science and Information Technology, Universiti Malaysia Sarawak (UNIMAS), 94300 Kota Samarahan, Sarawak, Malaysia

chlstephanie@unimas.my

Arif Bramantoro, School of Computing and Informatics, Universiti Teknologi Brunei, Jalan Tungku Link, Gadong BE1410, Brunei Darussalam

arif.bramantoro@utb.edu.bn

Published

2024-10-04

Issue

Section

Articles