Hierarchical feature extraction on functional brain networks for autism spectrum disorder identification with resting-state fMRI data

Abstract

Autism Spectrum Disorder (ASD) is a pervasive developmental disorder of the central nervous system, primarily manifesting in childhood. It is characterized by atypical and repetitive behaviors. Conventional diagnostic methods mainly rely on questionnaire surveys and behavioral observations, which are prone to misdiagnosis due to their subjective nature. With advancements in medical imaging, MR imaging-based diagnostics have emerged as a more objective alternative. In this paper, we propose a Hierarchical Neural Network model for ASD identification, termed ASD-HNet, which hierarchically extracts features from functional brain networks based on resting-state functional magnetic resonance imaging (rs-fMRI) data. This hierarchical approach enhances the extraction of brain representations, improving diagnostic accuracy and aiding in the identification of brain regions associated with ASD. Specifically, features are extracted at three levels, i.e., the local region of interest (ROI) scale, the community scale, and the global representation scale. At the ROI scale, graph convolution is employed to transfer features between ROIs. At the community scale, functional gradients are introduced, and a K-Means clustering algorithm is applied to group ROIs with similar functional gradients into communities. Features from ROIs within the same community are then extracted to characterize the communities. At the global representation scale, we extract global features from the whole community-scale brain networks to represent the entire brain. We validate the effectiveness of the ASD-HNet model using the publicly available Autism Brain Imaging Data Exchange I (ABIDE-I) dataset, ADHD-200,dataset and ABIDE-II dataset. Extensive experimental results demonstrate that ASD-HNet outperforms existing baseline methods. The code is available at https://github.com/LYQbyte/ASD-HNet.