Adaptive composing augmentations on multi-modal graph convolutional network for disease prediction

We propose an adaptive composing augmentation framework for multimodal GCN disease prediction that applies graph contrastive learning with learnable node confidence and selective edge perturbation, achieving 87.95% ACC and 90.05% AUC on ABIDE, significantly outperforming baseline and existing methods.

摘要

Graph Convolutional Networks (GCNs) have demonstrated significant success in population-based disease prediction. With the rise of multimodal technologies, multimodal GCNs integrate information from diverse data types, enhancing prediction accuracy, particularly in the fusion of imaging and non-imaging data. However, constructing a reliable population graph from limited multimodal data may result in poor generalization performance. To address this issue, we introduce graph contrastive learning as a multimodal data augmentation strategy, which reinforces the graph structure's robustness to disturbances. We propose an Adaptive Composing Augmentation framework that first employs a learnable similarity network to iteratively compute node confidence. Subsequently, the framework selectively perturbs edges of lesser importance within the graph through methods such as edge removal and edge weight permutation. Extensive experiments on three challenging medical datasets demonstrate that our method achieves state-of-the-art performance, including an accuracy (ACC) of 87.95% and area under the curve (AUC) of 90.05% on the ABIDE dataset. These results significantly outperform the baseline models, with improvements of 7.12% and 5.07%, and surpass existing methods by 6.2% and 4.83%, respectively. This confirms that contrastive learning with structured augmentations effectively enhances the generalization ability of multimodal GCNs.

主要结果

BibTeX

@article{han2025acagcn,
  author  = {Han, Longfei and Wang, Mengzhen and Zhang, Xiangsen and Li, Wenxin and Li, Haisheng and Huang, Xiankai},
  title   = {Adaptive composing augmentations on multi-modal graph convolutional network for disease prediction},
  journal = {Computers and Electrical Engineering},
  year    = {2025},
  doi     = {10.1016/j.compeleceng.2025.110277}
}