arrow
Return

Task-Aware Effective Connectivity Modeling for Cognitive Function Prediction

delete2025-12-15
delete0
PRE
AI
W
Wantong Zou
Y
Yu Li
X
Xiang Hu
X
Xun Chen
刘爱萍 cover
刘爱萍 (Aiping Liu)
DOI:10.1109/jbhi.2025.3644481delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Effective connectivity (EC) derived from resting-state functional magnetic resonance imaging (rs-fMRI) has emerged as a critical tool for deepening our understanding of brain function in both health and disease. However, most studies estimate EC on an individual basis, treating it as a hidden parameter within the model and requiring retraining the model for each subject. They often overlook the valuable population-level information and limit their generalizability. Additionally, EC is typically obtained independently of downstream tasks, reducing its capacity to effectively capture task-specific variations. To address these limitations, we propose a flexible Task-Aware Effective Connectivity (TAEC) model, designed to construct individualized, task-aware, and nonlinear causal brain networks without requiring subject-specific retraining. In this framework, a Causal Discovery Module (CDM) is introduced to capture the implicit neural representation of EC by a spatial-temporal attention mechanism, producing the estimation of an individual EC. Subsequently, we propose a Task-Aware Graph Neural Network (GNN) Predictor, which incorporates a task-aware penalty to enable end-to-end prediction, enhancing task performance and the identification of task-dependent EC patterns. Extensive experiments on twelve cognitive tasks from the Human Connectome Project (HCP) dataset demonstrate that the proposed method achieves state-of-the-art performance, validating its effectiveness in task-aware effective connectivity modeling. Furthermore, the framework discovers discriminative and task-specific EC patterns, which offer additional insights into cognitive functions.
Keywords:
Task-aware
effective connectivity
cognitive functions
Granger causality

Journal

IEEE Journal of Biomedical and Health Informatics cover
IEEE Journal of Biomedical and Health Informatics
IF:
6.8
Papers:
4.5K
Citations:
2.0W

Organization

S
sichuan mental health center
Scholars:
60
Papers: 16
Citations: 0
U
University of Science and Technology of China
Scholars:
1.6W
Papers: 5.5K
Citations: 11.3W