Heterogeneity-aware Fair Federated Learning

Xiaoli Li,Siran Zhao,Chuan Chen,Zibin Zheng
DOI: https://doi.org/10.1016/j.ins.2022.11.031
IF: 8.1
2023-01-01
Information Sciences
Abstract:The fairness of federated learning means that the global model cannot discriminate against any group. Due data heterogeneity, the update direction of some clients will hinder other clients, so the global model is difficult to treat each user fairly. The current fair federated learning methods usually use the variance of model performance to measure the fairness, which can not quantify the fairness of federated learning process; And, they all intervene from the beginning, which will reduce the convergence speed and model performance; Moreover, the weights they applied to clients are designed based on experience, so it is difficult to trade-off between model performance and fairness. To address the above problems, firstly, we use Gini coefficient to quantify the fairness of federated learning, which can reflect the fairness of federated learning process in each round. Secondly, we divide federated learning into two different training stages: label fitting and data fitting, and propose fairness intervention in the data fitting stage. Thirdly, we propose a fairness federated optimization algorithm, which introduces fairness penalty term into the objective function, and obtains the relationship between clients’ gradients and fairness through gradient descent. Experimental results show the effectiveness and fairness of the proposed method.
What problem does this paper attempt to address?