Search papers, labs, and topics across Lattice.
This paper introduces a novel framework for federated learning that enhances client aggregation by employing Conditional Random Fields (CRFs) to refine aggregation weights based on client-specific reliability and interactions. By defining unary and pairwise potentials, the method effectively addresses the challenges posed by heterogeneous data across clients, leading to improved convergence of the global model. Experimental results demonstrate that this approach outperforms traditional federated learning baselines, particularly in scenarios with non-IID data distributions.
CRF-based aggregation in federated learning significantly boosts model performance by accurately reflecting client reliability and interactions, especially under data heterogeneity.
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.
