mlsanityData to 14 Oct 2026

Certified Uncertainty Propagation in One-Shot Federated Bayesian Models via Posterior Event Transport

arXiv 2609.16373 ↗First posted 14 Sep 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Probabilistic certification of Bayesian neural networks lower-bounds the posterior probability that a model satisfies a verifier-defined safety property. In one-shot federated Bayesian learning, however, the deployed model is obtained by aggregating parameters drawn from client-specific posterior distributions, so local certificates do not directly guarantee safety of the aggregated model. This paper develops a deployment-consistent certification framework by propagating local posterior events through the deployment aggregation rule, with an exact geometric characterization for Federated Averaging (FedAvg). Each client constructs disjoint hyper-rectangular regions in parameter space and computes their probability masses. The server forms Cartesian products of these regions, maps them through the deployment rule, and retains a product event only when its aggregation image is verified to satisfy the safety property. Under independent client posteriors, each product-event probability factorizes into local masses, and summing verified disjoint events yields a lower bound on safety probability of the deployed model. For FedAvg with nonnegative aggregation coefficients, the image of a Cartesian product of axis-aligned hyper-rectangles is exactly a weighted hyper-rectangle, introducing no set over-approximation. We distinguish the proposed transported-event certificate from direct certification under posterior distributions induced by FedAvg and Product-of-Gaussians aggregation. Experiments on MNIST and Fashion-MNIST under label-Dirichlet heterogeneity show that the transported FedAvg certificate ranges from 22.51% to 46.89%, while direct global certificates range from 72.05% to 91.39%. Results show that predictive accuracy and certifiable safety do not necessarily follow the same trend, and that global posterior constructions can exhibit distinct certification behavior across architectures.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 26 references that are in the map. First posted on arXiv 14 Sep 2026.

Builds on

20 of 26 references.

References · 20

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
  3. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  4. 2019Advances and Open Problems in Federated Learningcited by 3,046
  5. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
  6. 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
  7. 2019Bayesian Nonparametric Federated Learning of Neural Networkscited by 387
  8. 2018On the Effectiveness of Interval Bound Propagation for Training Verifiably Robust Modelscited by 328
  9. 2020FedBE: Making Bayesian Model Ensemble Applicable to Federated Learningcited by 176
  10. 2019One-Shot Federated Learningcited by 125
  11. 2020Federated Learning via Posterior Averaging: A New Perspective and Practical Algorithmscited by 78
  12. 2022Personalized Federated Learning via Variational Bayesian Inferencecited by 68
8 more
  1. 2020Probabilistic Safety for Bayesian Neural Networkscited by 42
  2. 2019Statistical Guarantees for the Robustness of Bayesian Neural Networkscited by 35
  3. 2023FedLPA: One-shot Federated Learning with Layer-Wise Posterior Aggregationcited by 22
  4. 2023Calibrated One Round Federated Learning with Bayesian Inference in the Predictive Spacecited by 10
  5. 2023BNN-DP: Robustness Certification of Bayesian Neural Networks via Dynamic Programmingcited by 8
  6. 2023Adversarial Robustness Certification for Bayesian Neural Networkscited by 7
  7. 2024Tight Verification of Probabilistic Robustness in Bayesian Neural Networkscited by 5
  8. 2024Engineering Trustworthy AI: A Developer Guide for Empirical Risk Minimizationcited by 3

Built on by

Not cited yet.