Collaborative Fairness in Federated Learning
About
In current deep learning paradigms, local training or the Standalone framework tends to result in overfitting and thus poor generalizability. This problem can be addressed by Distributed or Federated Learning (FL) that leverages a parameter server to aggregate model updates from individual participants. However, most existing Distributed or FL frameworks have overlooked an important aspect of participation: collaborative fairness. In particular, all participants can receive the same or similar models, regardless of their contributions. To address this issue, we investigate the collaborative fairness in FL, and propose a novel Collaborative Fair Federated Learning (CFFL) framework which utilizes reputation to enforce participants to converge to different models, thus achieving fairness without compromising the predictive performance. Extensive experiments on benchmark datasets demonstrate that CFFL achieves high fairness, delivers comparable accuracy to the Distributed framework, and outperforms the Standalone framework.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| MRI prostate segmentation | Prostate MRI (test) | Client 1 Score91.01 | 34 | |
| Fundus Segmentation | Fundus (test) | Client 1 Score85.72 | 17 | |
| Medical Image Segmentation | RIF (test) | Site 1 Score0.8149 | 9 | |
| Retinal Fundus Segmentation | Retinal Fundus (test) | Client 1 Dice Score85.72 | 8 | |
| Client contribution estimation | Prostate MRI Segmentation (test) | Pearson Correlation75.44 | 7 | |
| Client contribution estimation | Retinal Fundus Segmentation (test) | Pearson Correlation0.9 | 7 | |
| Image Segmentation | Prostate MRI (test) | Pearson Correlation92.47 | 7 | |
| Image Segmentation | Retinal Fundus (test) | Pearson Correlation0.8453 | 7 |