Heterogeneity-Aware Client Sampling for Optimal and Efficient Federated Learning

heterogeneity-aware-client-sampling-for-optimal-and-efficient-federated-learning

Written by

in

, ,

[Submitted on 16 May 2025 (v1), last revised 9 Oct 2026 (this version, v4)]

View PDF HTML (experimental)

Abstract:Federated learning (FL) commonly involves clients with diverse communication and computational capabilities. Such heterogeneity can significantly distort the optimization dynamics and lead to objective inconsistency, where the global model converges to an incorrect stationary point potentially far from the pursued optimum. Despite its critical impact, the joint effect of communication and computation heterogeneity has remained largely unexplored, due to the intrinsic complexity of their interaction. In this paper, we reveal the fundamentally distinct mechanisms through which heterogeneous communication and computation drive inconsistency in FL. To the best of our knowledge, this is the first unified theoretical analysis of general heterogeneous FL, offering a principled understanding of how these two forms of heterogeneity jointly distort the optimization trajectory under arbitrary choices of local solvers. Motivated by these insights, we propose Federated Heterogeneity-Aware Client Sampling, FedACS, a universal method to eliminate all types of objective inconsistency. We theoretically prove that FedACS converges to the correct optimum at a rate of $O(1/sqrt{R})$, even in dynamic heterogeneous environments. Extensive experiments across multiple datasets show that FedACS outperforms state-of-the-art and category-specific baselines by 4.3%-36%, while reducing communication costs by 22%-89% and computation loads by 14%-105%, respectively.

Submission history

From: Shudi Weng [view email]
[v1] Fri, 16 May 2025 14:31:36 UTC (1,124 KB)
[v2] Mon, 16 Feb 2026 06:46:40 UTC (585 KB)
[v3] Sun, 22 Feb 2026 05:58:59 UTC (580 KB)
[v4] Fri, 9 Oct 2026 01:27:04 UTC (1,377 KB)