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 ]
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.