Pipeline from rs-fMRI raw data to the functional graph. The resting-state fMRI raw data are preprocessed and then parcellated into ROIs according to the AAL atlas. The FC matrices are calculated using Pearson’s correlation between ROIs. From the FC, we construct the brain functional graph G = (A, X), where A is the graph adjacency matrix characterizing the graph structure (A ∈ {0, 1} n×n ) and X is node feature matrix. Specifically, A is a binarized FC matrix, where only the top 20-percentile absolute values of the correlations of the matrix are transformed into ones, while the rest are transformed into zeros. For node feature X, Xk for node k can be defined as Xk = [ρk1, . . . , ρkn ] T , where ρkl is the Pearson’s correlation coefficient for node k and node l.

Blog

BrainIB: Interpretable brain network-based psychiatric diagnosis with graph information bottleneck

October 25, 2024

Publication

BrainIB: Interpretable brain network-based psychiatric diagnosis with graph information bottleneck

September 13, 2024

Zheng, Kaizhong; Yu, Shujian; Li, Baojuan; Jenssen, Robert; Chen, Badong.

Paper abstract

Developing new diagnostic models based on the underlying biological mechanisms rather than subjective symptoms for psychiatric disorders is an emerging consensus. Recently, machine learning (ML)-based classifiers using functional connectivity (FC) for psychiatric disorders and healthy controls (HCs) are developed to identify brain markers. However, existing ML-based diagnostic models are prone to overfitting (due to insufficient training samples) and perform poorly in new test environments. Furthermore, it is difficult to obtain explainable and reliable brain biomarkers elucidating the underlying diagnostic decisions. These issues hinder their possible clinical applications. In this work, we propose BrainIB, a new graph neural network (GNN) framework to analyze functional magnetic resonance images (fMRI), by leveraging the famed information bottleneck (IB) principle. BrainIB is able to identify the most informative edges in the brain (i.e., subgraph) and generalizes well to unseen data. We evaluate the performance of BrainIB against three baselines and seven state-of-the-art (SOTA) brain network classification methods on three psychiatric datasets and observe that our BrainIB always achieves the highest diagnosis accuracy. It also discovers the subgraph biomarkers that are consistent with clinical and neuroimaging findings. The source code and implementation details of BrainIB are freely available at the GitHub repository (https://github.com/SJYuCNEL/brain-and-Information-Bottleneck).