Interpretable machine learning approaches for children's ADHD detection using clinical assessment data: an online web application deployment.

Abstract

Background: Attention-deficit/hyperactivity disorder (ADHD) is a prevalent mental disorder characterized by hyperactivity, impulsivity, and inattention. This study aims to develop a verifiable and interpretable machine learning model to identify ADHD and its subtypes in children using clinical assessment scales data.

Methods: This study utilized the ADHD-200 dataset, including demographic data, Behavioral Rating Scale, and Wechsler Intelligence Scale assessments, to train and validate our models. The model's performance was evaluated using 10-fold cross-validation within the internal dataset, and the best model will be used for external validation. Seven machine learning models were evaluated. The SHapley Additive exPlanations (SHAP) method was employed for model interpretation. Finally, a web application will deploy the prediction model to provide ADHD probabilities based on user input.

Results: The Random Forest (RF) model performing best in identifying ADHD and the Support Vector Machine (SVM) model excelling in distinguishing ADHD subtypes. The RF model achieved an AUC of 0.99 in 10-fold cross-validation and an AUC of 0.99 in external validation, and the SVM model achieved a micro-average AUC of 0.96 and an accuracy of 0.83 in internal validation and a micro-average AUC of 0.96 and an accuracy of 0.85 in external validation. We used SHAP to interpret the models, revealing that higher ADHD Index pushed the model towards ADHD classification. Additionally, lower IQ scores were correlated with a higher likelihood of ADHD, consistent with previous studies. The dependency analysis found that the model can identify different behavioral scales. We deployed the final model online using a web application and showed users how the model made decisions.

Conclusions: Our findings highlight the potential of using machine learning and clinical assessment scales to support the diagnosis and subtype identification of ADHD in children, offering a practical solution for improving diagnostic accuracy and efficiency in clinical settings.