FedSCAl: Leveraging Server and Client Alignment for Unsupervised Federated Source-Free Domain Adaptation
About
We address the Federated source-Free Domain Adaptation (FFreeDA) problem, with clients holding unlabeled data with significant inter-client domain gaps. The FFreeDA setup constrains the FL frameworks to employ only a pre-trained server model as the setup restricts access to the source dataset during the training rounds. Often, this source domain dataset has a distinct distribution to the clients' domains. To address the challenges posed by the FFreeDA setup, adaptation of the Source-Free Domain Adaptation (SFDA) methods to FL struggles with client-drift in real-world scenarios due to extreme data heterogeneity caused by the aforementioned domain gaps, resulting in unreliable pseudo-labels. In this paper, we introduce FedSCAl, an FL framework leveraging our proposed Server-Client Alignment (SCAl) mechanism to regularize client updates by aligning the clients' and server model's predictions. We observe an improvement in the clients' pseudo-labeling accuracy post alignment, as the SCAl mechanism helps to mitigate the client-drift. Further, we present extensive experiments on benchmark vision datasets showcasing how FedSCAl consistently outperforms state-of-the-art FL methods in the FFreeDA setup for classification tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | Office-31 | Average Accuracy96.46 | 261 | |
| Image Classification | DomainNet (test) | Average Accuracy88.65 | 209 | |
| Image Classification | Office-Home (test) | Mean Accuracy86.01 | 199 | |
| Image Classification | Office-Home | Average Accuracy75.56 | 142 | |
| Image Classification | Office-31 (test) | Avg Accuracy89.25 | 93 | |
| Image Classification | DomainNet | Average Accuracy78.08 | 58 | |
| Image Classification | Office-31 Amazon domain (test) | Accuracy82.87 | 20 | |
| Image Classification | DomainNet | Accuracy (Q->C)65.37 | 13 | |
| Image Classification | Office-Home Art-Source (sub-table a) | Accuracy (A->C)61.22 | 8 | |
| Image Classification | Office-31 DSLR domain (test) | Accuracy98.66 | 8 |