Functional magnetic resonance imaging (fMRI) holds great potential for diagnosing and understanding brain disorders. However, the complexity and subtlety of disease-relevant variations in fMRI present significant challenges. To address this issue, we propose a framework that combines equivariant learning and contrastive learning (ECL) to disentangle disease-relevant patterns from irrelevant patterns in fMRI. The framework uses a personalized mask to separate the functional connectivity network from fMRI into a disease-relevant subgraph and an irrelevant subgraph. The disease-relevant subgraph undergoes an equivariant learning pipeline to align the orbit of the encoded features with the orbit of the augmented views of the inputs. The disease-irrelevant subgraph undergoes a contrastive learning pipeline that pulls the encoded features to be close from augmented views of the same input. By combining these 2 learning processes, the learned encoder can be invariant to perturbations to disease-irrelevant patterns while equivariant to disease-relevant variations. The proposed approach achieved state-of-the-art classification performance across 3 benchmark datasets: ABIDE I, ABIDE II, and ADHD-200, with significant improvements in accuracy (improved by up to 5%). Interpretability experiments identified disease-related regions of interest (ROIs) of clinical relevance. These results establish our framework as a promising tool for analyzing brain networks in fMRI. The code is available at https://github.com/CXshen468/ecl .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Disentangle Disease-Relevant Patterns from Irrelevant Patterns in fMRI Analysis Using Equivariant and Contrastive Learning

  • Xin Shen,
  • Shengjie Zhang,
  • Wenbin Liu,
  • Yuan Zhou

摘要

Functional magnetic resonance imaging (fMRI) holds great potential for diagnosing and understanding brain disorders. However, the complexity and subtlety of disease-relevant variations in fMRI present significant challenges. To address this issue, we propose a framework that combines equivariant learning and contrastive learning (ECL) to disentangle disease-relevant patterns from irrelevant patterns in fMRI. The framework uses a personalized mask to separate the functional connectivity network from fMRI into a disease-relevant subgraph and an irrelevant subgraph. The disease-relevant subgraph undergoes an equivariant learning pipeline to align the orbit of the encoded features with the orbit of the augmented views of the inputs. The disease-irrelevant subgraph undergoes a contrastive learning pipeline that pulls the encoded features to be close from augmented views of the same input. By combining these 2 learning processes, the learned encoder can be invariant to perturbations to disease-irrelevant patterns while equivariant to disease-relevant variations. The proposed approach achieved state-of-the-art classification performance across 3 benchmark datasets: ABIDE I, ABIDE II, and ADHD-200, with significant improvements in accuracy (improved by up to 5%). Interpretability experiments identified disease-related regions of interest (ROIs) of clinical relevance. These results establish our framework as a promising tool for analyzing brain networks in fMRI. The code is available at https://github.com/CXshen468/ecl .