Skip to yearly menu bar Skip to main content


Poster
in
Workshop: Bayesian Decision-making and Uncertainty: from probabilistic and spatiotemporal modeling to sequential experiment design

Variational Inference in Similarity Spaces: A Bayesian Approach to Personalized Federated Learning

Pedro Henrique Barros · Fabricio Murai · Amir Houmansadr · Alejandro C. Frery · Heitor Filho

Keywords: [ Variational Inference ] [ Similarity spaces ] [ federated learning ]


Abstract:

Similarity space (or S-space) employs an encoder function, fed by labeled original pairwise data, to find a latent pairwise space with markers (prototypical) vector. It divides the space into regions where pairs of objects are either similar or dissimilar. This paper enhances S-space, equipping variational inference from personalized federated learning. The S-space representation aligns local representation spaces across clients, while variational inference improves generalization and reduces overfitting caused by data scarcity and client heterogeneity. Our theoretical analysis improved upper bounds on KL divergence between optimal local and optimal global variational models compared to traditional distributed Bayesian neural networks.

Chat is not available.