Federated Domain Adaptation via Transformer for Multi-Site Alzheimer's Disease Diagnosis

IEEE Trans Med Imaging. 2023 Dec;42(12):3651-3664. doi: 10.1109/TMI.2023.3300725. Epub 2023 Nov 30.

Abstract

In multi-site studies of Alzheimer's disease (AD), the difference of data in multi-site datasets leads to the degraded performance of models in the target sites. The traditional domain adaptation method requires sharing data from both source and target domains, which will lead to data privacy issue. To solve it, federated learning is adopted as it can allow models to be trained with multi-site data in a privacy-protected manner. In this paper, we propose a multi-site federated domain adaptation framework via Transformer (FedDAvT), which not only protects data privacy, but also eliminates data heterogeneity. The Transformer network is used as the backbone network to extract the correlation between the multi-template region of interest features, which can capture the brain abundant information. The self-attention maps in the source and target domains are aligned by applying mean squared error for subdomain adaptation. Finally, we evaluate our method on the multi-site databases based on three AD datasets. The experimental results show that the proposed FedDAvT is quite effective, achieving accuracy rates of 88.75%, 69.51%, and 69.88% on the AD vs. NC, MCI vs. NC, and AD vs. MCI two-way classification tasks, respectively.

MeSH terms

  • Alzheimer Disease* / diagnostic imaging
  • Humans
  • Image Interpretation, Computer-Assisted / methods
  • Machine Learning
  • Magnetic Resonance Imaging* / methods
  • Neuroimaging / methods