Skip to yearly menu bar Skip to main content


Poster

Connecting Federated ADMM to Bayes

Siddharth Swaroop · Mohammad Emtiyaz Khan · Finale Doshi-Velez

Hall 3 + Hall 2B #439
[ ]
Fri 25 Apr midnight PDT — 2:30 a.m. PDT

Abstract:

We provide new connections between two distinct federated learning approaches based on (i) ADMM and (ii) Variational Bayes (VB), and propose new variants by combining their complementary strengths. Specifically, we show that the dual variables in ADMM naturally emerge through the "site" parameters used in VB with isotropic Gaussian covariances. Using this, we derive two versions of ADMM from VB that use flexible covariances and functional regularisation, respectively. Through numerical experiments, we validate the improvements obtained in performance. The work shows connection between two fields that are believed to be fundamentally different and combines them to improve federated learning.

Live content is unavailable. Log in and register to view live content