Learning the Hierarchical Organization in Brain Network for Brain Disorder Diagnosis
This work addresses brain disorder diagnosis by capturing cross-network interactions often missed in prior methods, offering incremental improvements in classification and interpretability.
The authors tackled the problem of diagnosing brain disorders using fMRI data by proposing BrainHO, a method that learns hierarchical brain network dependencies without predefined sub-networks, achieving state-of-the-art classification performance on ABIDE and REST-meta-MDD datasets.
Brain network analysis based on functional Magnetic Resonance Imaging (fMRI) is pivotal for diagnosing brain disorders. Existing approaches typically rely on predefined functional sub-networks to construct sub-network associations. However, we identified many cross-network interaction patterns with high Pearson correlations that this strict, prior-based organization fails to capture. To overcome this limitation, we propose the Brain Hierarchical Organization Learning (BrainHO) to learn inherently hierarchical brain network dependencies based on their intrinsic features rather than predefined sub-network labels. Specifically, we design a hierarchical attention mechanism that allows the model to aggregate nodes into a hierarchical organization, effectively capturing intricate connectivity patterns at the subgraph level. To ensure diverse, complementary, and stable organizations, we incorporate an orthogonality constraint loss, alongside a hierarchical consistency constraint strategy, to refine node-level features using high-level graph semantics. Extensive experiments on the publicly available ABIDE and REST-meta-MDD datasets demonstrate that BrainHO not only achieves state-of-the-art classification performance but also uncovers interpretable, clinically significant biomarkers by precisely localizing disease-related sub-networks.