Proportional Fairness in Federated Learning

Guojun Zhang, Saber Malekmohammadi, Xi Chen, Yaoliang Yu

Abstract: With the increasingly broad deployment of federated learning (FL) systems in the real world, it is critical but challenging to ensure fairness in FL, i.e. reasonably satisfactory performances for each of the numerous diverse clients. In this work, we introduce and study a new fairness notion in FL, called proportional fairness (PF), which is based on the relative change of each client's performance. From its connection with the bargaining games, we propose PropFair, a novel and easy-to-implement algorithm for finding proportionally fair solutions in FL, and study its convergence properties. Through extensive experiments on vision and language datasets, we demonstrate that PropFair can approximately find PF solutions, and it achieves a good balance between the average performances of all clients and of the worst 10% clients.

Submission Length: Long submission (more than 12 pages of main content)

Assigned Action Editor: ~Kangwook_Lee1

License: Creative Commons Attribution 4.0 International (CC BY 4.0)

Submission Number: 543