FAIRVAR: Fair Federated Learning via Variance Regularization
About
Federated learning (FL) allows collaborative training of machine learning models across multiple parties without sharing raw data. However, heterogeneous data can cause some clients to have disproportionate influence on the global model, leading to disparities in their performance. Fairness, understood as reducing these disparities, is therefore a crucial concern in FL and has been addressed in various ways. We studied performance equitable fairness in FL, where the goal is to minimize performance disparities across clients. We evaluated several existing fairness-aware methods and introduce here a new gradient-variance-regularized method, implemented in two variants: FairGrad (approximate) and FairGrad* (exact). We theoretically characterize the connections between these methods and, empirically, on heterogeneous benchmarks, show that FairGrad and FairGrad* consistently improve fairness by reducing variance in client accuracies, while maintaining competitive or improved mean performance compared to existing fairness-aware baselines.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | TinyImageNet (test) | Accuracy63.46 | 562 | |
| Federated Image Classification | CIFAR100 (test) | Accuracy70.78 | 144 | |
| Image Classification | CIFAR-10 Dir(0.5) (test) | Accuracy72.61 | 36 | |
| Image Classification | CIFAR-10 Dir-0.1 (test) | Test Accuracy86.7 | 8 | |
| Image Classification | MNIST Dir(0.5) (test) | Test Accuracy89.1 | 8 | |
| Image Classification | CIFAR-10 Dir(0.05) (test) | Test Accuracy92.18 | 8 | |
| Image Classification | MNIST Dir(0.1) (test) | Test Accuracy94.12 | 8 | |
| Image Classification | MNIST Dir(0.05) (test) | Test Accuracy94.42 | 8 |