Federated domain adaptation for healthcare

Federated domain adaptation (FDA) describes the setting where a set of source clients seek to optimize the performance of a target client. To be effective, FDA must address some of the distributional challenges of Federated learning (FL). For instance, FL systems exhibit distribution shifts across clients. Further, labeled data are not always available among the clients. To this end, we propose and compare novel approaches for FDA, combining the few labeled target samples with the source data when auxiliary labels are available to the clients. The in-distribution auxiliary information is included during local training to boost out-of-domain accuracy. Also, during fine-tuning, we devise a simple yet efficient gradient projection method (FedGP) to detect the valuable components from each source client model by comparing them with the target direction. The extensive experiments on healthcare datasets show that our proposed framework outperforms the state-of-the-art unsupervised FDA methods with limited additional time and space complexity. Additionally, we find that common techniques such as FedAvg and fine-tuning fail with a large domain shift. To better investigate the effectiveness of FedGP under various extents of domain shift, we perform extensive semi-synthetic and real-world experiments on general-purposed datasets compared with several baselines. Our results indicate a bias-variance trade-off between source and target domains when combining source and target gradients. FedGP maintains a better trade-off between source gradients' bias (the domain shift between source and target domains) and the target gradient's variance from limited labeled data. Our experiments illustrate the effectiveness of the proposed method in practice.

Name thesis:degree_name

M.S.

Level thesis:degree_level

Thesis

Discipline thesis:degree_discipline

Computer Science

Grantor

University of Illinois at Urbana-Champaign

Year dc:date

2023

Author dc:creator

• Jiang, Enyi

Contributors dc:contributor

• Koyejo, Oluwasanmi

• Federated Learning
• Domain Adaptation
• Ml For Healthcare

Statement dc:rights

• Copyright 2023 Enyi Jiang

Language dc:language

en, eng

Handle dc:identifier

https://hdl.handle.net/2142/120432

OAI record oai:identifier

https://www.ideals.illinois.edu/oai-pmh?verb=GetRecord&metadataPrefix=etdms&identifier=oai%3Awww.ideals.illinois.edu%3A2142%2F120432