Abstract:Importance: Early identification of the likelihood of autism spectrum disorder (ASD) using minimal information is crucial for early diagnosis and intervention, which can affect developmental outcomes. Objective: To develop and validate a machine learning (ML) model for predicting ASD using a minimal set of features from background and medical information and to evaluate the predictors and the utility of the ML model. Design, setting, and participants: For this diagnostic study, a retrospective analysis of the Simons Foundation Powering Autism Research for Knowledge (SPARK) database, version 8 (released June 6, 2022), was conducted, including data from 30 660 participants after adjustments for missing values and class imbalances (15 330 with ASD and 15 330 without ASD). The SPARK database contains participants recruited from 31 university-affiliated research clinicals and online in 26 states in the US. All individuals with a professional ASD diagnosis and their families were eligible to participate. The model performance was validated on independent datasets from SPARK, version 10 (released July 21, 2023), and the Simons Simplex Collection (SSC), consisting of 14 790 participants, followed by phenotypic associations. Exposures: Twenty-eight basic medical screening and background history items present before 24 months of age. Main outcomes and measures: Generalizable ML prediction models were developed for detecting ASD using 4 algorithms (logistic regression, decision tree, random forest, and eXtreme Gradient Boosting [XGBoost]). Performance metrics included accuracy, area under the receiver operating characteristics curve (AUROC), sensitivity, specificity, positive predictive value (PPV), and F1 score, offering a comprehensive assessment of the predictive accuracy of the model. Explainable AI methods were applied to determine the effect of individual features in predicting ASD as secondary outcomes, enhancing the interpretability of the best-performing model. The secondary outcome analyses were further complemented by examining differences in various phenotypic measures using nonparametric statistical methods, providing insights into the ability of the model to differentiate between different presentations of ASD. Results: The study included 19 477 (63.5%) male and 11 183 (36.5%) female participants (mean [SD] age, 106 [62] months). The mean (SD) age was 113 (68) months for the ASD group and 100 (55) months for the non-ASD group. The XGBoost (termed AutMedAI) model demonstrated strong performance with an AUROC score of 0.895, sensitivity of 0.805, specificity of 0.829, and PPV of 0.897. Developmental milestones and eating behavior were the most important predictors. Validation on independent cohorts showed an AUROC of 0.790, indicating good generalizability. Conclusions and relevance: In this diagnostic study of ML prediction of ASD, robust model performance was observed to identify autistic individuals with more symptoms and lower cognitive levels. The robustness and ML model generalizability results are promising for further validation and use in clinical and population settings.

Prediction of attention deficit hyperactivity disorder based on explainable artificial intelligence

Assessing machine learning for fair prediction of ADHD in school pupils using a retrospective cohort study of linked education and healthcare data

Enhancing Psychologists' Understanding Through Explainable Deep Learning Framework for ADHD Diagnosis

Use of machine learning on clinical questionnaires data to support the diagnostic classification of Attention DeficitHyperactivity Disorder: a personalized medicine approach

Predicting childhood and adolescent attention-deficit/hyperactivity disorder onset: a nationwide deep learning approach

Reliable Autism Spectrum Disorder Diagnosis for Pediatrics Using Machine Learning and Explainable AI

Transparent but Powerful: Explainability, Accuracy, and Generalizability in ADHD Detection from Social Media Data

A novel approach to ADHD classification based on severity and emotional impairment: Findings from artificial intelligence analysis

Predicting Clinical Symptoms of Attention Deficit Hyperactivity Disorder Based on Temporal Patterns Between and Within Intrinsic Connectivity Networks.

Predicting individual cases of major adolescent psychiatric conditions with artificial intelligence

Automated detection of ADHD: Current trends and future perspective

Application and expansion of an algorithm predicting attention-deficit/hyperactivity disorder and impairment in a predominantly White sample

Individualized prediction models in ADHD: a systematic review and meta-regression

Generalizable prediction of childhood ADHD symptoms from neurocognitive testing and youth characteristics

A data science approach to optimize ADHD assessment with the BRIEF-2 questionnaire

Toward explainable AI (XAI) for mental health detection based on language behavior

A systematic review on the application of machine learning models in psychometric questionnaires for the diagnosis of attention deficit hyperactivity disorder

Automated diagnoses of attention deficit hyperactive disorder using magnetic resonance imaging

Multimodal neuroimaging-based prediction of adult outcomes in childhood-onset ADHD using ensemble learning techniques

Machine Learning Prediction of Autism Spectrum Disorder From a Minimal Set of Medical and Background Information

ADHD-AID: Aiding Tool for Detecting Children's Attention Deficit Hyperactivity Disorder via EEG-Based Multi-Resolution Analysis and Feature Selection