Autism Spectrum Disorder Classification with Interpretability in Children Based on Structural MRI Features Extracted Using Contrastive Variational Autoencoder

📅 2023-07-03
🏛️ Big Data Mining and Analytics
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
This study addresses the challenge of early autism spectrum disorder (ASD) screening in toddlers aged 0.92–4.83 years. We propose an interpretable classification framework for structural MRI (s-MRI) data. Methodologically, we introduce a novel contrastive variational autoencoder (CVAE) to disentangle ASD-specific neuroanatomical features from general developmental patterns; integrate transfer learning to enhance robustness under limited-sample conditions; and perform cortical surface area association analysis to ensure neuroanatomically grounded interpretability. Evaluated on 78 s-MRI scans from Shenzhen Children’s Hospital, our model significantly discriminates ASD from typically developing children and identifies the frontal and temporal cortices as potential biomarkers. To our knowledge, this is the first work to apply contrastive generative modeling to low-age ASD neuroimaging classification—achieving both high discriminative performance and clinical interpretability—thereby establishing a new paradigm for targeted early intervention.
📝 Abstract
Autism spectrum disorder (ASD) is a highly disabling mental disease that brings significant impairments of social interaction ability to the patients, making early screening and intervention of ASD critical. With the development of the machine learning and neuroimaging technology, extensive research has been conducted on machine classification of ASD based on structural Magnetic Resonance Imaging (s-MRI). However, most studies involve with datasets where participants' age are above 5 and lack interpretability. In this paper, we propose a machine learning method for ASD classification in children with age range from 0.92 to 4.83 years, based on s-MRI features extracted using contrastive variational autoencoder (CVAE). 78 s-MRIs, collected from Shenzhen Children's Hospital, are used for training CVAE, which consists of both ASD-specific feature channel and common shared feature channel. The ASD participants represented by ASD-specific features can be easily discriminated from TC participants represented by the common shared features. In case of degraded predictive accuracy when data size is extremely small, a transfer learning strategy is proposed here as a potential solution. Finally, we conduct neuroanatomical interpretation based on the correlation between s-MRI features extracted from CVAE and surface area of different cortical regions, which discloses potential biomarkers that could help target treatments of ASD in the future.
Problem

Research questions and friction points this paper is trying to address.

Classify ASD in children (0.92-4.83 years) using s-MRI features
Improve interpretability via CVAE-extracted ASD-specific biomarkers
Address small data limitations with transfer learning strategy
Innovation

Methods, ideas, or system contributions that make the work stand out.

Uses contrastive variational autoencoder for feature extraction
Applies transfer learning for small dataset accuracy
Correlates s-MRI features with cortical surface areas
🔎 Similar Papers
No similar papers found.
R
Ruimin Ma
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China
Ruitao Xie
Ruitao Xie
Shenzhen University
mobile computingcloud computingwireless sensor networks
Y
Yanlin Wang
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China
J
Jintao Meng
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China
Yanjie Wei
Yanjie Wei
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China
W
Wenhui Xi
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China
Y
Yi Pan
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Center for High Performance Computing, Shenzhen 518055, Guangdong, China; Shenzhen Key Laboratory of Intelligent Bioinformatics, Shenzhen Institute of Advanced Technology, Shenzhen 518055, China