FedGST: Federated Graph Spatio-Temporal Framework for Brain Functional Disease Prediction

Junbin Mao, Hanhe Lin, Xu Tian, Yi Pan, Jin Liu

Research output: Chapter in Book/Report/Conference proceedingConference contribution

7 Downloads (Pure)

Abstract

Currently, most medical institutions face the challenge of training a unified model using fragmented and isolated data to address disease prediction problems. Although federated learning has become the recognized paradigm for privacy-preserving model training, how to integrate federated learning with fMRI temporal characteristics to enhance predictive performance remains an open question for functional disease prediction. To address this challenging task, we propose a novel Federated Graph Spatio-Temporal (FedGST) framework for brain functional disease prediction. Specifically, anchor sampling is used to process variable-length time series data on local clients. Then dynamic functional connectivity graphs are generated via sliding windows and Pearson correlation coefficients. Next, we propose an InceptionTime model to extract temporal information from the dynamic functional connectivity graphs on the local clients. Finally, the hidden activation variables are sent to a global server. We propose a UniteGCN model on the global server to receive and process the hidden activation variables from clients. Then, the global server returns gradient information to clients for backpropagation and model parameter updating. Client models aggregate model parameters on the local server and distribute them to clients for the next round of training. We demonstrate that FedGST outperforms other federated learning methods and baselines on ABIDE-1 and ADHD200 datasets.

Original languageEnglish
Title of host publicationProceedings of 2023 IEEE International Conference on Bioinformatics and Biomedicine (BIBM)
EditorsXingpeng Jiang, Haiying Wang, Reda Alhajj, Xiaohua Hu, Felix Engel, Mufti Mahmud, Nadia Pisanti, Xuefeng Cui, Hong Song
Place of PublicationIstanbul
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1356-1361
Number of pages6
ISBN (Electronic)9798350337488
ISBN (Print)9798350337495
DOIs
Publication statusPublished - 18 Jan 2024
Event2023 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2023 - Istanbul, Turkey
Duration: 5 Dec 20238 Dec 2023
https://bidma.cpsc.ucalgary.ca/IEEE-BIBM-2023/

Conference

Conference2023 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2023
Abbreviated titleIEEE BIBM 2023
Country/TerritoryTurkey
CityIstanbul
Period5/12/238/12/23
Internet address

Keywords

  • Brain Functional Disease
  • Federated Learning
  • Graph Learing
  • Spatio-Temporal

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Science Applications
  • Computer Vision and Pattern Recognition
  • Automotive Engineering
  • Modelling and Simulation
  • Health Informatics

Fingerprint

Dive into the research topics of 'FedGST: Federated Graph Spatio-Temporal Framework for Brain Functional Disease Prediction'. Together they form a unique fingerprint.

Cite this