Addressing Algorithmic Disparity and Performance Inconsistency in Federated Learning

Federated learning (FL) has gain growing interests for its capability of learning from distributed data sources collectively without the need of accessing the raw data samples across different sources. So far FL research has mostly focused on improving the performance, how the algorithmic disparity will be impacted for the model learned from FL and the impact of algorithmic disparity on the utility inconsistency are largely unexplored. In this paper, we propose an FL framework to jointly consider performance consistency and algorithmic fairness across different local clients (data sources). We derive our framework from a constrained multiobjective optimization perspective, in which we learn a model satisfying fairness constraints on all clients with consistent performance. Specifically, we treat the algorithm prediction loss at each local client as an objective and maximize the worst-performing client with fairness constraints through optimizing a surrogate maximum function with all objectives involved. A gradient-based procedure is employed to achieve the Pareto optimality of this optimization problem. Theoretical analysis is provided to prove that our method can converge to a Pareto solution that achieves the min-max performance with fairness constraints on all clients. Comprehensive experiments on synthetic and real-world datasets demonstrate the superiority that our approach over baselines and its effectiveness in achieving both fairness and consistency across all local clients.

[1]  Johannes O. Royset,et al.  Algorithms with Adaptive Smoothing for Finite Minimax Problems , 2003 .

[2]  Christopher T. Lowenkamp,et al.  False Positives, False Negatives, and False Analyses: A Rejoinder to "Machine Bias: There's Software Used across the Country to Predict Future Criminals. and It's Biased against Blacks" , 2016 .

[3]  Krishna P. Gummadi,et al.  Fairness Constraints: Mechanisms for Fair Classification , 2015, AISTATS.

[4]  Hanghang Tong,et al.  Fairness-aware Agnostic Federated Learning , 2020, SDM.

[5]  Mehryar Mohri,et al.  Agnostic Federated Learning , 2019, ICML.

[6]  Alistair E. W. Johnson,et al.  Generalizability of predictive models for intensive care unit patients , 2018, ArXiv.

[7]  Allison Woodruff,et al.  Putting Fairness Principles into Practice: Challenges, Metrics, and Improvements , 2019, AIES.

[8]  Yang Liu,et al.  Federated Learning , 2019, Synthesis Lectures on Artificial Intelligence and Machine Learning.

[9]  Max Welling,et al.  The Variational Fair Autoencoder , 2015, ICLR.

[10]  Nathan Srebro,et al.  Equality of Opportunity in Supervised Learning , 2016, NIPS.

[11]  Blaise Agüera y Arcas,et al.  Communication-Efficient Learning of Deep Networks from Decentralized Data , 2016, AISTATS.

[12]  Nathan Kallus,et al.  The Fairness of Risk Scores Beyond Classification: Bipartite Ranking and the xAUC Metric , 2019, NeurIPS.

[13]  Mikhail Posypkin,et al.  A deterministic algorithm for global multi-objective optimization , 2014, Optim. Methods Softw..

[14]  C. A. Coello Coello,et al.  Evolutionary multi-objective optimization: a historical view of the field , 2006, IEEE Computational Intelligence Magazine.

[15]  Guillermo Sapiro,et al.  Minimax Pareto Fairness: A Multi Objective Perspective , 2020, ICML.

[16]  Toniann Pitassi,et al.  Learning Fair Representations , 2013, ICML.

[17]  Blake Lemoine,et al.  Mitigating Unwanted Biases with Adversarial Learning , 2018, AIES.

[18]  Zhe Zhao,et al.  Data Decisions and Theoretical Implications when Adversarially Learning Fair Representations , 2017, ArXiv.

[19]  J. Désidéri Multiple-gradient descent algorithm (MGDA) for multiobjective optimization , 2012 .

[20]  Jun Sakuma,et al.  Fairness-aware Learning through Regularization Approach , 2011, 2011 IEEE 11th International Conference on Data Mining Workshops.

[21]  Toniann Pitassi,et al.  Learning Adversarially Fair and Transferable Representations , 2018, ICML.

[22]  Riccardo Miotto,et al.  Federated Learning of Electronic Health Records Improves Mortality Prediction in Patients Hospitalized with COVID-19 , 2020, medRxiv.

[23]  Zhao Chen,et al.  GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks , 2017, ICML.

[24]  Qingfu Zhang,et al.  Pareto Multi-Task Learning , 2019, NeurIPS.

[25]  Ron Kohavi,et al.  Scaling Up the Accuracy of Naive-Bayes Classifiers: A Decision-Tree Hybrid , 1996, KDD.

[26]  Fei Wang,et al.  Towards Model-Agnostic Post-Hoc Adjustment for Balancing Ranking Fairness and Algorithm Utility , 2020, KDD.

[27]  Jie Xu,et al.  Federated Learning for Healthcare Informatics , 2019, ArXiv.

[28]  Jean-Antoine Désidéri,et al.  Comparison between MGDA and PAES for Multi-Objective Optimization , 2011 .

[29]  Vaibhav Rajan,et al.  Multi-Task Learning with User Preferences: Gradient Descent with Controlled Ascent in Pareto Optimization , 2020, ICML.

[30]  Tian Li,et al.  Fair Resource Allocation in Federated Learning , 2019, ICLR.

[31]  Fei Wang,et al.  Explaining Algorithmic Fairness Through Fairness-Aware Causal Path Decomposition , 2021, KDD.

[32]  Alistair E. W. Johnson,et al.  The eICU Collaborative Research Database, a freely available multi-center database for critical care research , 2018, Scientific Data.

[33]  Zhao Song,et al.  Solving Linear Programs in the Current Matrix Multiplication Time , 2021, J. ACM.

[34]  Toniann Pitassi,et al.  Fairness through awareness , 2011, ITCS '12.