Pith. sign in

REVIEW 4 cited by

FedBE: Making Bayesian Model Ensemble Applicable to Federated Learning

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2009.01974 v4 pith:EQOWFL3S submitted 2020-09-04 cs.LG stat.ML

FedBE: Making Bayesian Model Ensemble Applicable to Federated Learning

classification cs.LG stat.ML
keywords modelfedbemodelsusersaggregationbayesiandatafederated
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved
0 comments
Share X Bluesky LinkedIn Reddit HN
read the original abstract

Federated learning aims to collaboratively train a strong global model by accessing users' locally trained models but not their own data. A crucial step is therefore to aggregate local models into a global model, which has been shown challenging when users have non-i.i.d. data. In this paper, we propose a novel aggregation algorithm named FedBE, which takes a Bayesian inference perspective by sampling higher-quality global models and combining them via Bayesian model Ensemble, leading to much robust aggregation. We show that an effective model distribution can be constructed by simply fitting a Gaussian or Dirichlet distribution to the local models. Our empirical studies validate FedBE's superior performance, especially when users' data are not i.i.d. and when the neural networks go deeper. Moreover, FedBE is compatible with recent efforts in regularizing users' model training, making it an easily applicable module: you only need to replace the aggregation method but leave other parts of your federated learning algorithm intact. Our code is publicly available at https://github.com/hongyouc/FedBE.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 4 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. FedQHD: Closed-Form Function-Space Federated Reinforcement Learning

    cs.LG 2026-05 unverdicted novelty 7.0

    FedQHD achieves closed-form federated Q-learning via hyperdimensional encoders with linear readouts, formalizes the federation gap under heterogeneous encoders, and reports competitive performance on continuous-state ...

  2. Towards Uncertainty-Aware Federated Granger Causal Learning

    cs.LG 2026-02 conditional novelty 7.0

    Federated Granger causality uncertainty reaches a steady state determined solely by aleatoric client data statistics, independent of epistemic priors, supporting reliable hypothesis testing for cross-client interactions.

  3. Q-LocalAdam: Memory-Efficient Client-Side Adaptive Optimization for Edge Federated Learning

    cs.LG 2026-05 unverdicted novelty 6.0

    Q-LocalAdam reduces optimizer memory by 3.37x via tailored 8-bit quantization for Adam states while maintaining or improving accuracy under data heterogeneity in edge federated learning.

  4. Knowledge Distillation in Federated Learning: a Survey on Long Lasting Challenges and New Solutions

    cs.LG 2024-06 unverdicted novelty 2.0

    A survey organizing knowledge distillation techniques for addressing privacy, heterogeneity, communication, and personalization challenges in federated learning.