Digitala Vetenskapliga Arkivet

Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Bayesian Federated Learning with Stochastic Variational Inference
Dublin City University, Ireland.
The University of Tartu, Estonia.
Blekinge Institute of Technology, Faculty of Computing, Department of Computer Science.ORCID iD: 0000-0002-6309-2892
Dublin City University, Ireland.
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
Available from: 2025-02-28 Created: 2025-02-28 Last updated: 2025-02-28Bibliographically approved

Open Access in DiVA

fulltext(866 kB)40 downloads
File information
File name FULLTEXT01.pdfFile size 866 kBChecksum SHA-512
bc16af6c90dff4b57d0421ddf4df6ca408b6f6c3a84359e27fbf5ceec999a06a9febe890fb305f5ea6b366da072dc6fb2c7768158742eee1ce94a9d3a5d1bb21
Type fulltextMimetype application/pdf

Other links

Publisher's full textScopus

Search in DiVA

By author/editor
Alawadi, Sadi
By organisation
Department of Computer Science
Artificial IntelligenceProbability Theory and Statistics

Search outside of DiVA

GoogleGoogle Scholar
Total: 42 downloads
The number of downloads is the sum of all downloads of full texts. It may include eg previous versions that are now no longer available

doi
isbn
urn-nbn

Altmetric score

doi
isbn
urn-nbn
Total: 1359 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf