arXiv · 2606.30161
Federated Learning with Energy-Based Structured Probabilistic Inference
Abstract
Federated learning typically aggregates client updates using fixed or heuristic weighting rules, which can be suboptimal when clients have heterogeneous data and varying contributions to the global model. We propose a framework that refines client aggregation weights using Conditional Random Fields (CRFs). Our method defines unary potentials for individual clients and pairwise potentials for all client pairs, allowing the server to model both client-specific reliability and interactions between clients. The resulting CRF inference produces aggregation weights that enable better convergence of the global training objective. Experiments show that, under non-IID heterogeneity, our approach consistently improves performance over well-established federated learning baselines.
Explore related subjects
Keep this discovery
Dario Fenoglio, Daniil Kirilenko, Martin Gjoreski, Marc Langheinrich. 2026-06-29. Federated Learning with Energy-Based Structured Probabilistic Inference. https://arxiv.org/abs/2606.30161
Cite the original work for its findings. Save a collection to share your selection of sources.