Poster
CoBo: Collaborative Learning via Bilevel Optimization
Diba Hashemi · Lie He · Martin Jaggi
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