Skip to content

A Fair Federated Learning Method for Handling Client Participation Probability Inconsistencies in Heterogeneous Environments

2025 · Neural Information Processing Systems · pp. 166678-166707 · 1 citation · ⚡ 1 influential · 68 references
Computer Science

TL;DR

This work proposes PHP-FL, a novel model-heterogeneous FL method that explicitly addresses scenarios with varying client participation probabilities to enhance both model accuracy and performance fairness, and introduces a Dual-End Aligned ensemble Learning (DEAL) module.

Abstract

Federated learning (FL) is a distributed machine learning paradigm that enables multiple clients to collaboratively train a shared model without exposing their raw data. However, existing FL research has primarily focused on optimizing learning performance based on the assumption of uniform client participation, with few studies delving into performance fairness under inconsistent client participation, particularly in model-heterogeneous FL environments. In view of this challenge, we propose PHP-FL , a novel model-heterogeneous FL method that explicitly addresses scenarios with varying client participation probabilities to enhance both model accuracy and performance fairness. Specifically, we introduce a Dual-End Aligned ensemble Learning (DEAL) module, where small auxiliary models on clients are used for dual-end knowledge alignment and local ensemble learning, effectively tackling model heterogeneity without a public dataset. Furthermore, to mitigate update conflicts caused by inconsistent participation probabilities, we propose an Importance-driven Selective Parameter Update (ISPU) module, which accurately updates critical local parameters based on training progress. Finally, we implement PHP-FL on a lightweight FL platform with heterogeneous clients across three different client participation patterns. Extensive experiments under heterogeneous settings and diverse client participation patterns demonstrate that PHP-FL achieves state-of-the-art performance in both accuracy and fairness. Our code is available at: https://github.com/Siyuan01/PHP-FL-main .

View source

Similar papers

#machine learning Preprint Sep 2026

Similarity-Aware Personalized Federated Learning in Heterogeneous Environments

Federated Learning (FL) allows decentralized clients to train models collaboratively while preserving data privacy. However, distribution mismatch across clients often leads to poor global generalization and degraded local client-level performance. In such scenarios, some of the clients with their local models trained solely on local data may perform better than the globally learnt model, thus nullifying the benefits of collaborative federated learning. To address this, we propose SAPE-FL (Similarity-Aware Personalized Federated Learning), a novel personalization framework that anchors each client's model to both the global model and a similarity-weighted peer averaged model. By incorporating dynamic, client-specific regularization based on both model similarity and output similarity, SAPE-FL adaptively balances global knowledge transfer and peer collaboration while filtering out dissimilar clients. This dual anchoring mitigates negative transfer and enhances robustness in heterogeneous settings. We theoretically analyze our algorithm establishing its convergence guarantees and empirically show that SAPE-FL outperforms state-of-the-art methods under high statistical heterogeneity and low client data regimes.

A. Kumar, Sunil Gupta, Ngyuen Dang et al. · 0 citations
Preprint Aug 2026

Assessing the Impacts of Imperfect Datasets on Client Selections in Federated Learning

This study experimentally measures the impact of non-IID data, noisy data, and fairness in client selection on model accuracy and convergence, and proposes a privacy-preserving scoring method to assess each client's contribution in FL.

Yuan-Heng Tsai, Li-Hsing Yen, Yan-Wei Chen · 0 citations
Preprint Aug 2026

Beyond Parameter Space: NTK-Guided Personalized Aggregation for Robust Federated Learning

Local Inference Guided Aggregation for Heterogeneous Training Environments to Yield Enhancement Through Agreement and Regularization (LIGHTYEAR), a federated learning framework that performs update selection in function space using an NTK-based agreement score to characterize predictive behavior and determine a personalized aggregation set for each client.

Mirko Konstantin, S. Zachow, Anirban Mukhopadhyay · 0 citations
#machine learning Preprint Aug 2026

Beyond Non-IID: Learner--Client Distribution Mismatch in Federated Learning

This paper considers the practical setting where the learner keeps a small proxy dataset, and proposes a dynamic, influence-aware client selection framework that estimates each client's potential utility to the learner's optimization objective using proxy influence signals on a learner-specific proxy set.

Yiming Xie, Linghui Su, Ningfang Mi · 0 citations
Review Open access Jul 2026

A Fairness Perspective on Client Selection and Aggregation Methods for Non-IID Mitigation in Federated Learning: A Survey

Federated learning (FL) is a promising approach for training distributed machine learning models while preserving clients’ data privacy. However, in real-world FL systems, data are often not independent and identically distributed (non-IID). This heterogeneity can slow convergence, degrade model performance, and increase client drift. To address these challenges, numerous methods have been proposed to mitigate non-IID data effects by optimizing client selection, local training, and model aggregation strategies. Despite their effectiveness in improving performance and efficiency, these methods rarely consider fairness across clients. Improving global accuracy does not guarantee balanced participation, influence, or outcomes, which may lead to biased model behavior across clients. In this survey, we review existing non-IID mitigation methods in FL from a fairness perspective and provide a systematic analysis of their implicit impact on client participation and influence. Unlike prior surveys that treat fairness as a separate research direction, this work analyzes how these methods designed for non-IID mitigation implicitly shape fairness outcomes across clients. Our taxonomy classifies existing methods into three categories—fairness-aware, semi-fairness-aware, and fairness-unaware—based on their design strategies for client selection and model aggregation. Using this taxonomy, we analyze the advantages, trade-offs, and limitations of each category and highlight that mitigating non-IID data does not guarantee fairness across clients. Finally, we identify open challenges and outline future directions, including system-level FL design that jointly considers non-IID mitigation and fairness and the development of standardized fairness evaluation metrics. Overall, this survey aims to provide a structured perspective on the relationship between non-IID mitigation and fairness and support the development of more balanced and scalable FL systems under non-IID conditions.

Mohannad Alsofyani, Isra Al-Turaiki, H. Mathkour · 0 citations
#machine learning Preprint Aug 2026

CRAD: Class-wise Reliability-Aware Distillation for Decentralized Heterogeneous Federated Learning

Conventional federated learning (FL) relies on parameter averaging, which forces clients to be doubly homogeneous: it demands an identical architecture and degrades under non-IID data. Real-world deployments usually break both assumptions. We sidestep both by building a decentralized knowledge distillation framework in which each client evaluates its peers'model snapshots on its own local data and distills from the resulting soft predictions. Because knowledge is transferred through the shared class posterior, clients are free to run different architectures; and because every teacher is evaluated on the student's own device, raw data never leaves the client, with no central server or public dataset required. Within this setting, we identify and address an under-examined problem: how to combine the peer teacher predictions. Existing methods, like uniform averaging, ignore how knowledge reliability varies across teachers and classes. We propose Class-wise Reliability-Aware Distillation (CRAD), which, per class, first discards teachers that disagree with the peer consensus and then takes a weighted average of the rest, weighting each teacher by its per-class reliability (precision, or inverse variance). Since the variance of an accuracy from $n$ samples scales as $1/n$, support enters automatically: among the teachers that survive filtering, a teacher is trusted for a class to the degree that it is both accurate and well-evidenced for it. On three image-classification benchmarks (CIFAR-10, CIFAR-100, and PathMNIST colon pathology), across heterogeneous architectures under severe non-IID skew, CRAD consistently outperforms competing methods in global accuracy.

Baraa Bilbeisi, Mengchen Fan, Baocheng Geng et al. · 1 citation

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.