Modeling Functional Brain Networks for ADHD via Spatial Preservation-based Neural Architecture Search

Modeling functional brain networks (FBNs) for attention deficit hyperactivity disorder (ADHD) has sparked significant interest since the abnormal functional connectivity is discovered in certain functional magnetic resonance imaging (fMRI)-based brain regions compared to typical developmental control (TC) individuals. However, existing models for modeling FBNs generally use dimensionality reduction techniques to process the high dimensional input data, which results in confusion and an inaccurate representation of voxel interactions between spatially close brain regions, causing misdiagnosis of the disease. To address these issues, we propose a spatial preservation-based neural architecture search (SP-NAS) for FBNs modeling in ADHD. The main work includes three-fold: 1) A spatial preservation module is designed to embed original spatial information into dimensionality reduction data, addressing the challenge of a large number of parameters in the original data and mitigating disease misdiagnosis resulting from voxel confusion between different brain regions caused by dimensionality reduction. 2) A search space using more suitable search operations is constructed to efficiently extract spatial-temporal interaction characteristics of fMRI data in ADHD while narrowing the search space. 3) Cross-regional association differences between ADHD and TC groups are explored for ADHD auxiliary diagnosis since the abnormal activation regions of ADHD relative to TC on the brain regions and the abnormal connectivity between the lesion brain regions are identified. Model validation results on the ADHD-200 dataset show that the FBNs obtained from SP-NAS not only achieve competitive results in ADHD diagnosis but also reveal abnormal connections in the lesion regions of ADHD consistent with clinical diagnosis.