Bayesian Federated Learning with Stochastic Variational Inference
2024 (English)In: 2024 2nd International Conference on Federated Learning Technologies and Applications, FLTA 2024 / [ed] Awaysheh F.M., Alawadi S., Alawadi S., Carnevale L., Lloret Mauri J., Alsmirat M., Institute of Electrical and Electronics Engineers (IEEE), 2024, p. 290-297Conference paper, Published paper (Refereed)
Abstract [en]
Federated Learning (FL) faces significant challenges, such as handling non-IID (Non-Independent and Identically Distributed) data and efficiently aggregating distributed models, which can lead to slower convergence and reduced model accuracy. This paper proposes a novel framework, Bayesian Federated Learning with Stochastic Variational Inference (BayFL-SVI), to address these issues. Stochastic Variational Inference (SVI) is a scalable approximation method for Bayesian inference that optimizes the Evidence Lower Bound (ELBO) using mini-batches of data through stochastic gradient descent. By computing the ELBO for each client update, our approach quantifies the significance of these updates, effectively managing the heterogeneity of non-IID data and improving the aggregation process. This approach results in a more accurate and robust integration of client contributions, enhancing convergence rates and overall model performance. We provide theoretical analysis with convergence guarantees. Our empirical results demonstrate significant improvements in convergence rates and model accuracy, establishing a solid foundation for future studies.
Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024. p. 290-297
Keywords [en]
Aggregation, Bayesian Federated Learning, Federated Learning, Stochastic Variational Inference, Adversarial machine learning, Contrastive Learning, Bayesian, Convergence model, Convergence rates, Distributed data, Low bound, Modeling accuracy, Stochastics, Variational inference
National Category
Artificial Intelligence Probability Theory and Statistics
Identifiers
URN: urn:nbn:se:bth-27503DOI: 10.1109/FLTA63145.2024.10840014Scopus ID: 2-s2.0-85217880651ISBN: 9798350354812 (print)OAI: oai:DiVA.org:bth-27503DiVA, id: diva2:1941320
Conference
2nd IEEE International Conference on Federated Learning Technologies and Applications, FLTA 2024, Hybrid, Valencia, Sept 17-20, 2024
2025-02-282025-02-282025-02-28Bibliographically approved