arXiv · 2603.11307
Client-Conditional Federated Learning via Local Training Data Statistics
Abstract
Federated learning (FL) under data heterogeneity remains challenging: existing methods either ignore client differences (FedAvg), require costly cluster discovery (IFCA), or maintain per-client models (Ditto). All degrade when data is sparse or heterogeneity is multi-dimensional. We propose conditioning a single global model on locally-computed PCA statistics of each client's training data, requiring zero additional communication. Evaluating across 97~configurations spanning four heterogeneity types (label shift, covariate shift, concept shift, and combined heterogeneity), four datasets (MNIST, Fashion-MNIST, CIFAR-10, CIFAR-100), and seven FL baseline methods, we find that our method matches the Oracle baseline -- which knows true cluster assignments -- across all settings, surpasses it by 1--6% on combined heterogeneity where continuous statistics are richer than discrete cluster identifiers, and is uniquely sparsity-robust among all tested methods.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Rickard Brännvall. 2026-03-11. Client-Conditional Federated Learning via Local Training Data Statistics. https://arxiv.org/abs/2603.11307
Cite the original work for its findings. Save a collection to share your selection of sources.