M2LG-DG: A Multi-modal Local-Global Domain Generalization Framework for Cross-site Major Depressive Disorder Classification
Not on the map. Why?
Abstract (arXiv)
Classification models based on resting-state functional magnetic resonance imaging (rs-fMRI) often show lower performance at imaging sites not included during model development, which can limit their use in clinical settings. Domain generalization (DG) addresses this issue by learning representations from source sites that remain effective for unseen target sites. However, existing DG approaches for psychiatric disorder classification commonly rely on a single imaging modality and may not fully account for site-specific acquisition effects on the learned representation space. Subjects scanned at the same site share scanner hardware, acquisition settings, and preprocessing characteristics, which can cause representations to reflect acquisition conditions rather than diagnostic information. In this work, we present M2LG-DG, a source-only multimodal local-global framework for cross-site major depressive disorder (MDD) classification. The framework employs a dual-stream rs-fMRI encoder, where the global pathway models inter-regional dependencies through self-attention and the local pathway performs graph-constrained aggregation over functional connectivity-derived brain graphs. Imaging and non-imaging representations are decomposed into shared and private components and integrated through bidirectional cross-attention with a learned modality gate. A cross-site supervised contrastive objective forms positive pairs from same-class subjects acquired at different source sites, encouraging the fused representation to preserve diagnostic information across acquisition domains. On four held-out REST-meta-MDD sites, M2LG-DG achieves an AUC of 69.48% and exceeds the closest comparison method by 2.18 percentage points. Experiments on the Autism Brain Imaging Data Exchange (ABIDE) dataset further support its applicability to other psychiatric neuroimaging classification tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 69 references that are in the map. First posted on arXiv 31 Aug 2026.
Builds on
21 of 69 references.
References · 21
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2020Supervised Contrastive Learningcited by 2,829
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2016Convolutional Neural Networks on Graphs with Fast Localized Spectral Filteringcited by 2,089
- 2019Invariant Risk Minimizationcited by 1,854
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,200
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 763
- 2021VLMo: Unified Vision-Language Pre-Training with Mixture-of-Modality-Expertscited by 425
- 2021Causality-inspired Single-source Domain Generalization for Medical Image Segmentationcited by 119
- 2023Vita-CLIP: Video and text adaptive CLIP via Multimodal Promptingcited by 91
- 2021Invariant Information Bottleneck for Domain Generalizationcited by 86
- 2018Batch-Instance Normalization for Adaptively Style-Invariant Neural Networkscited by 78
9 more
- 2023Align and Attend: Multimodal Summarization with Dual Contrastive Lossescited by 65
- 2022Interpretable Graph Neural Networks for Connectome-Based Brain Disorder Analysiscited by 59
- 2025Advances in Multimodal Adaptation and Generalization: From Traditional Approaches to Foundation Modelscited by 23
- 2022Adversarial Consistency for Single Domain Generalization in Medical Image Segmentationcited by 22
- 2024Towards Multimodal Open-Set Domain Generalization and Adaptation through Self-supervisioncited by 16
- 2022Contrastive learning for regression in multi-site brain age predictioncited by 14
- 2024MM-GTUNets: Unified Multi-Modal Graph Deep Learning for Brain Disorders Predictioncited by 9
- 2024Joint semi-supervised and contrastive learning enables domain generalization and multi-domain segmentationcited by 4
- 2024Augmentation-based Unsupervised Cross-Domain Functional MRI Adaptation for Major Depressive Disorder Identificationcited by 1
Built on by
Not cited yet.