Self-Promoted Clustering-based Contrastive Learning for Brain Networks Pretraining

Self-Promoted Clustering-based Contrastive Learning for Brain Networks Pretraining

Junbo Ma, Caixuan Luo, Jia Hou, Kai Zhao

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence
Main Track. Pages 1164-1172. https://doi.org/10.24963/ijcai.2024/129

Rapid advancements in neuroimaging techniques, such as magnetic resonance imaging (MRI), have facilitated the acquisition of the structural and functional characteristics of the brain. Brain network analysis is one of the essential tools for exploring brain mechanisms from MRI, providing valuable insights into the brain's organization, and stimulating the understanding of brain cognition and pathology of neurodegenerative diseases. Graph Neural Networks (GNNs) are commonly used for brain network analysis, but they are limited by the scarcity of medical data. Although Graph Contrastive Learning methods have been developed to address this, they often involve graph augmentations that distort the anatomical brain structures. To address these challenges, an augmentation-free contrastive learning method, named Self-Promoted Clustering-based Contrastive Learning(SPCCL), is proposed in this paper. Specifically, by introducing a clustering-based contrastive Learning loss and a self-promoted contrastive pairs creation scheme, the proposed SPCCL can be pre-trained from additional healthy subjects' data that are relatively easier to acquire than disorder ones. The proposed SPCCL leverages these additional data with respect to the integrity of the original brain structure, making it a promising approach for effective brain network analysis. Comprehensive experiments are conducted on an open-access schizophrenic dataset, demonstrating the effectiveness of the proposed method.
Keywords:
Computer Vision: CV: Biomedical image analysis
Humans and AI: HAI: Brain sciences
Machine Learning: ML: Deep learning architectures
Machine Learning: ML: Multi-modal learning