Contrastive voxel clustering for multiscale modeling of brain network

Abstract

Resting-state functional magnetic resonance imaging (fMRI) provides an efficient way to analyze the functional connectivity between brain regions. A comprehensive understanding of brain functionality requires a unified description of multi-scale layers of neural structure. However, existing brain network modeling methods often simplify this property by averaging Blood oxygen level dependent (BOLD) signals at the brain region level for fMRI-based analysis with the assumption that BOLD signals are homogeneous within each brain region, which ignores the heterogeneity of voxels within each Region of Interest (ROI). This study introduces a novel multi-stage self-supervised learning framework for multiscale brain network analysis, which effectively delineates brain functionality from voxel to ROIs and up to sample level. A Contrastive Voxel Clustering (CVC) module is proposed to simultaneously learn the voxel-level features and clustering assignments, which ensures the retention of informative clustering features at the finest voxel-level and concurrently preserves functional connectivity characteristics. Additionally, based on the extracted features and clustering assignments at the voxel level by CVC, a Brain ROI-based Graph Neural Network (BR-GNN) is built to extract functional connectivity features at the brain ROI-level and used for sample-level prediction, which integrates the functional clustering maps with the pre-established structural ROI maps and creates a more comprehensive and effective analytical tool. Experiments are performed on two datasets, which illustrate the effectiveness and generalization ability of the proposed method by analyzing voxel-level clustering results and brain ROIs-level functional characteristics. The proposed method provides a multiscale modeling framework for brain functional connectivity analysis, which will be further used for other brain disease identification. Code is available at https://github.com/yanliugroup/fmri-cvc.

Keywords

• Contrastive clustering
• Graph neural network (GNN)
• Functional connectivity
• Functional MRI (fMRI)