Multimodal Connectome Fusion via Cross-Attention for Autism Spectrum Disorder Classification Using Graph Learning
Organizations: Department of Medicine, Danube Private University (DPU), Krems, Austria · Department of Medical Physics and Biomedical Engineering, Medical University of Vienna, Vienna, Austria · Department of Medical Informatics, Faculty of Medicine, Gonabad University of Medical Sciences, Gonabad, Iran · Austrian Center for Medical Innovation and Technology (ACMIT), Wiener Neustadt, Austria
Abstract
Autism spectrum disorder (ASD) is a complex neurodevelopmental condition characterized by atypical functional brain connectivity and subtle structural alterations. rs-fMRI has been widely used to identify disruptions in large-scale brain networks, while structural MRI provides complementary information about morphological organization. Despite their complementary nature, effectively integrating these heterogeneous imaging modalities within a unified framework remains challenging. This study proposes a multimodal graph learning framework that preserves the dominant role of functional connectivity while integrating structural imaging and phenotypic information for ASD classification. The proposed framework is evaluated on ABIDE-I dataset. Each subject is represented as a node within a population graph. Functional and structural features are extracted as modality-specific node attributes, while inter-subject relationships are modeled using a pairwise association encoder (PAE) based on phenotypic information. Two Edge Variational GCNs are trained to learn subject-level embeddings. To enable effective multimodal integration, we introduce a novel asymmetric transformer-based cross-attention mechanism that allows functional embeddings to selectively incorporate complementary structural information while preserving functional dominance. The fused embeddings are then passed to a MLP for ASD classification. Using stratified 10-fold cross-validation, the framework achieved an AUC of 87.3% and an accuracy of 84.4%. Under leave-one-site-out cross-validation (LOSO-CV), the model achieved an average cross-site accuracy of 82.0%, outperforming existing methods by approximately 3% under 10-fold cross-validation and 7% under LOSO-CV. The proposed framework effectively integrates heterogeneous multimodal data from the multi-site ABIDE-I dataset, improving automated ASD classification across imaging sites.
Figures & tables
| Model | Modality | AUC | Acc | Sensitivity | Specificity | F1-Score |
|---|---|---|---|---|---|---|
| rs-fMRI | 79.40 | 77.40 | 76.70 | 84.41 | 80.00 | |
| Proposed | rs-fMRI* | 84.80 | 82.60 | 83.90 | 85.30 | 84.10 |
| Method | sMRI | 73.90 | 73.30 | 73.00 | 81.00 | 76.40 |
| sMRI* | 82.30 | 81.00 | 86.73 | 76.50 | 80.90 | |
| Multimodal | 87.30 | 84.40 | 85.20 | 87.00 | 85.70 | |
| Note: (*) indicates the inclusion of non-imaging data alongside the specified | ||||||
| Reference | Model | AUC | Acc |
|---|---|---|---|
| [ 11 ] | FCN | 69.60 | 66.20 |
| Proposed Method | 82.30 | 81.00 |
| Multimodal Fusion Strategy | AUC | Acc | Sens | Spec | F1-Score |
|---|---|---|---|---|---|
| Concatenation | 84.90 | 83.00 | 84.90 | 84.20 | 84.30 |
| Symmetric Cross-Attention | 82.30 | 81.30 | 84.40 | 82.30 | 82.90 |
| Asymmetric Cross-Attention | 87.30 | 84.40 | 85.20 | 87.00 | 85.70 |
| Reference | Model | Modality | AUC | ACC |
|---|---|---|---|---|
| [ 19 ] | Brain Network Transformer | rs-fMRI* | 80.2 | 71.0 |
| [ 4 ] | Com-BrainTF | rs-fMRI* | 79.6 | 72.5 |
| [ 17 ] | ASDFormer | rs-fMRI* | 81.17 | 74.60 |
| [ 33 ] | DeepGCN | rs-fMRI* | 82.59 | 77.27 |
| [ 32 ] | MMGCN | rs-fMRI* | 84.00 | 78.31 |
| [ 16 ] | EV-GCN | rs-fMRI* | 84.72 | 81.06 |
| Site | Size | ASD- | ASD- | DeepGCN | Our |
|---|---|---|---|---|---|
| DiagNet | SAENet | Proposed | |||
| CALTECH | 15 | 52.8 | 56.7 | 71.7 | 86.7 |
| CMU | 11 | 68.5 | 70.6 | 80.5 | 81.8 |
| KKI | 33 | 69.5 | 72.6 | 76.2 | 63.6 |
| LEUVEN | 56 | 61.3 | 64.6 | 74.3 | 71.4 |
| MAX_MUN | 46 | 48.6 | 47.5 | 69.7 | 87.0 |