Multi-atlas classification of autism spectrum disorder with hinge loss trained deep architectures: ABIDE I results

Thomas Martial Epalle,Yuqing Song,Zhe Liu,Hu Lu
DOI: https://doi.org/10.1016/j.asoc.2021.107375
IF: 8.7
2021-08-01
Applied Soft Computing
Abstract:<p>The Autism Brain Imaging Data Exchange (ABIDE) database has emerged as a benchmark cohort for training and testing machine learning models for classifying autism. Considering recent successes using deep learning, increasing the overall classification accuracy on this dataset is a challenging endeavor. In this paper, we propose a multi-input deep neural network model for classifying autism. Our model's architecture is designed to incorporate neuroimaging data preprocessed using three different reference atlases. For each training example, the proposed deep neural network simultaneously receives connectome data originating from three different parcellation strategies and automatically learns discriminative features from the three input sets. This process results in learned features that are more general and less dependent on a single brain parcellation method. Moreover, the proposed neural network is trained using the hinge loss function, which strongly penalizes misclassifications. We validate our model under cross-validation frameworks, using a set of 1,038 real participants and an augmented set of 10,038 samples. We achieve a classification accuracy of 78.07% on real data and 79.13% on augmented data, which is about 9% higher than previously reported results. Our results reinforce the idea that discordance in resting-state network connectivity constitutes a major discriminatory feature between patients with autism and healthy individuals. Our proposed multi-atlas deep neural network classification framework is potentially applicable to additional brain disorders.</p>
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?