English

Federated Martingale Posterior Samping

Machine Learning 2026-05-19 v1 Machine Learning

Abstract

Federated Bayesian neural networks require fixing a prior on the model parameters together with a likelihood. Eliciting meaningful priors on the weight space of modern overparameterized models is notoriously difficult, and misspecification of either component can severely degrade accuracy and calibration. Motivated by the rapid progress of predictive models such as large language models, the martingale posterior, also known as predictive Bayes, replaces the prior--likelihood pair with a predictive distribution and recovers parameter uncertainty by repeatedly drawing predictive samples and refitting the model. A direct federated implementation, however, would require clients to share the local data sets. This letter proposes {federated martingale posterior} (FMP) sampling, a one-shot embarrassingly parallel protocol in which each client uploads a small set of trainable data embeddings and the server runs the predictive sampler centrally. Experiments on MNIST, CIFAR-10, and CIFAR-100 show that FMP closely matches the centralized counterpart and significantly improves calibration over consensus-style baselines.

Keywords

Cite

@article{arxiv.2605.18554,
  title  = {Federated Martingale Posterior Samping},
  author = {Boning Zhang and Matteo Zecchin and Mingzhao Guo and Dongzhu Liu and Osvaldo Simeone},
  journal= {arXiv preprint arXiv:2605.18554},
  year   = {2026}
}

Comments

5 pages