Skip to yearly menu bar Skip to main content


Poster

CoBo: Collaborative Learning via Bilevel Optimization

Diba Hashemi · Lie He · Martin Jaggi


Abstract:

Collaborative learning is an important tool to train multiple clients more effectively by enabling communication among agents. Identifying helpful clients, however, presents challenging and often introduces significant overhead. In this paper, we model client-selection and model-training as two interconnected optimization problems, proposing a novel bilevel optimization problem for collaborative learning.We introduce CoBo, a scalable and elastic, SGD-type alternating optimization algorithm that efficiently addresses these problem with theoretical convergence guarantees. Empirically, CoBo achieves superior performance, surpassing popular personalization algorithms by 5.8% in accuracy on a task with high heterogeneity, involving datasets distributed among 80 clients.

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