RCC-PFL: Robust Client Clustering under Noisy Labels in Personalized Federated Learning
Authors: Abdulmoneam Ali, Ahmed Arafa
AI-generated Key Points
- Authors Abdulmoneam Ali and Ahmed Arafa address the challenge of accurately estimating cluster identities in personalized federated learning (PFL)
- PFL involves training different personal models and relies on clustering users into groups with similar objectives
- Noisy labeled data can lead to misleading loss function values and ineffective clustering
- The authors propose RCC-PFL, a label-agnostic data similarity-based clustering algorithm with three key advantages:
- Independently estimates cluster identities from training labels
- Acts as a one-shot clustering method before training
- Requires fewer communication rounds and less computation compared to iterative-based methods
- Validation using diverse models and datasets shows superiority over multiple baselines in terms of average accuracy and variance reduction
- RCC-PFL offers a robust solution for accurate cluster identity estimation in PFL settings, enhancing efficiency and effectiveness of personalized federated learning processes
Abstract: We address the problem of cluster identity estimation in a personalized federated learning (PFL) setting in which users aim to learn different personal models. The backbone of effective learning in such a setting is to cluster users into groups whose objectives are similar. A typical approach in the literature is to achieve this by training users' data on different proposed personal models and assign them to groups based on which model achieves the lowest value of the users' loss functions. This process is to be done iteratively until group identities converge. A key challenge in such a setting arises when users have noisy labeled data, which may produce misleading values of their loss functions, and hence lead to ineffective clustering. To overcome this challenge, we propose a label-agnostic data similarity-based clustering algorithm, coined RCC-PFL, with three main advantages: the cluster identity estimation procedure is independent from the training labels; it is a one-shot clustering algorithm performed prior to the training; and it requires fewer communication rounds and less computation compared to iterative-based clustering methods. We validate our proposed algorithm using various models and datasets and show that it outperforms multiple baselines in terms of average accuracy and variance reduction.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.