Fair Data Representation for Machine Learning at the Pareto Frontier

Shizhou Xu,Thomas Strohmer
2023-11-24
Abstract:As machine learning powered decision-making becomes increasingly important in our daily lives, it is imperative to strive for fairness in the underlying data processing. We propose a pre-processing algorithm for fair data representation via which supervised learning results in estimations of the Pareto frontier between prediction error and statistical disparity. Particularly, the present work applies the optimal affine transport to approach the post-processing Wasserstein-2 barycenter characterization of the optimal fair $L^2$-objective supervised learning via a pre-processing data deformation. Furthermore, we show that the Wasserstein-2 geodesics from the conditional (on sensitive information) distributions of the learning outcome to their barycenter characterizes the Pareto frontier between $L^2$-loss and the average pairwise Wasserstein-2 distance among sensitive groups on the learning outcome. Numerical simulations underscore the advantages: (1) the pre-processing step is compositive with arbitrary conditional expectation estimation supervised learning methods and unseen data; (2) the fair representation protects the sensitive information by limiting the inference capability of the remaining data with respect to the sensitive data; (3) the optimal affine maps are computationally efficient even for high-dimensional data.
Machine Learning,Probability
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to achieve a fair data representation in machine learning. Specifically, when dealing with sensitive information (such as race, gender, etc.), how to find the optimal balance point between prediction error and statistical difference. Concretely, the paper proposes a pre - processing algorithm. Through this algorithm, a fair data representation can be generated, so that the results of supervised learning can be close to the Pareto frontier between prediction error and statistical difference. The paper pays special attention to how to use the optimal affine transport to approximate the representation of the post - processing Wasserstein centroid, which is a method of optimal fair \(L^2\) - objective supervised learning. In addition, the paper also explores the characteristics of Wasserstein geodesics from conditional distributions to their centroids. These characteristics define the Pareto frontier between the \(L^2\) loss and the average pairwise Wasserstein distance of sensitive groups in learning results. In short, the core problem of this paper is to generate a fair data representation through pre - processing steps to optimize the trade - off between prediction accuracy and fairness in machine - learning models. The solution to this problem is of great significance for preventing bias and discrimination in machine - learning models.