Federated learning meets Bayesian neural network: Robust and uncertainty-aware distributed variational inference

Pengfei Li, Qinghua Hu, Xiaofei Wang

Abstract: Highlights•FedUAB integrates FL and BNNs, reduces overfitting and model drift. Effective for limited, varied data.•Innovative BNN-FL fusion: Prior selection, Gaussian weight aggregation, variance control.•FedUAB boosts server and clients models. Adds uncertainty quantification.