Evolutionary neural architecture search for automated MDD diagnosis using multimodal MRI imaging

Abstract

Summary: Major depressive disorder (MDD) is a prevalent mental disorder with serious impacts on life and health. Neuroimaging offers valuable diagnostic insights. However, traditional computer-aided diagnosis methods are limited by reliance on researchers’ experience. To address this, we proposed an evolutionary neural architecture search (M-ENAS) framework for automatically diagnosing MDD using multi-modal magnetic resonance imaging (MRI). M-ENAS determines the optimal weight and network architecture through a two-stage search method. Specifically, we designed a one-shot network architecture search (NAS) strategy to train supernet weights and a self-defined evolutionary search to optimize the network structure. Finally, M-ENAS was evaluated on two datasets, demonstrating that M-ENAS outperforms existing hand-designed methods. Additionally, our findings reveal that brain regions within the somatomotor network play important roles in the diagnosis of MDD, providing additional insight into the biological mechanisms underlying the disorder.

Keywords

• Neuroscience
• Bioinformatics
• Psychology