An Enhanced Federated Prototype Learning Method Under Domain Shift

Liang Kuang

doi:10.1007/978-981-95-5696-0_20

An Enhanced Federated Prototype Learning Method Under Domain Shift

2026-01-01

0

PRE

AI

DOI：10.1007/978-981-95-5696-0_20

Original

Original request for help

Save

Abstract

En 中文

Federated Learning (FL) allows collaborative machine learning training without sharing private data. Numerous studies have shown that one significant factor affecting the performance of federated learning models is the heterogeneity of data across different clients, especially when the data is sampled from various domains. A recent paper introduces variance-aware dual-level prototype clustering and uses a novel alpha-sparsity prototype loss, which increases intra-class similarity and reduces inter-class similarity. To ensure that the features converge within specific clusters, we introduce an improved algorithm, Federated Prototype Learning with Convergent Clusters, abbreviated as FedPLCC. To increase inter-class distances, we weight each prototype with the size of the cluster it represents. To reduce intra-class distances, considering that prototypes with larger distances might come from different domains, we select only a certain proportion of prototypes for the loss function calculation. Evaluations on the Digit-5, Office-10, and DomainNet datasets show that our method performs better than existing approaches.

Keywords:

Federated prototype learning

Domain heterogeneity

An Enhanced Federated Prototype Learning Method Under Domain Shift

Abstract

Journal

Organization