Demographic parity (DP) is a widely studied fairness criterion in regression, enforcing independence between the predictions and sensitive attributes. However, constraining the entire distribution can degrade predictive accuracy and may be unnecessary for many applications, where fairness concerns are localized to specific regions of the distribution. To overcome this issue, we propose a new framework for regression under DP that focuses on the tails of target distribution across sensitive groups. Our methodology builds on optimal transport theory. By enforcing fairness constraints only over targeted regions of the distribution, our approach enables more nuanced and context-sensitive interventions. Leveraging recent advances, we develop an interpretable and flexible algorithm that leverages the geometric structure of optimal transport. We provide theoretical guarantees, including risk bounds and fairness properties, and validate the method through experiments in regression settings.
Figures & tables
Figure 1: Illustration for t↦Qs∗(t) for some s∈S in the case where α≤s∈S∑psQf∗∣s(p) (red curve).
Figure 2: Illustration for t↦Qsξ(t) on (p,1] in the case where α>s∈S∑psQf∗∣s(p) (red curve).
Figure 3: Histogram of predictions on synthetic data before (left) and after (right) enforcing (α,p) -DP-tails fairness. The constraint parameters are set (arbitrary) to α=0 , p=0.5
Figure 4: Empirical CDF of the predictions on the synthetic data before and after enforcing DP-tails fairness. Plot (b): the parameters are set (arbitrary) to α=0 , p=0.5 ; plot (c): α=0 and p minimizes ( 7 ).
Figure 5: Evolution of the tail unfairness of g^α,pξ with respect to the size N of the unlabeled dataset on the synthetic data.
Figure 6: Evolution of the MSE and KS distance of g^αξ with respect to α on the synthetic data.
Figure 7: Empirical CDF of the predictions on the Crime before and after enforcing DP-tails fairness. Plot (b): the parameters are set (arbitrary) to α=0.2 , p=0.4 ; Plot (c): α=0.2 and p minimizes ( 7 ). More runs can be seen in Fig. 11 in Appendix C .
Figure 8: Evolution of the MSE and KS distance of g^αξ with respect to α on the real data.
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 9: Optimal solution of (P′)
Figure 10: ξ− optimal solution
Figure 11: Empirical CDF of the predictions on the Crime before and after enforcing DP-tails fairness over 20 runs. The parameters are set (arbitrary) to α=0.2 , p=0.4 .
Figure 12: Empirical CDF of the predictions on the California Housing before and after enforcing DP-tails fairness over 20 runs. The parameters are set (arbitrary) to α=1.5 , p=0.4 .
Figure 13: Empirical CDF of the predictions on the Law school before and after enforcing DP-tails fairness over 20 runs. The parameters are set (arbitrary) to α=0.5 , p=0.4 .
Demographic parity (DP) is a widely used group fairness criterion requiring predictive distributions to be invariant across sensitive groups. While natural in classification, full distributional DP is often overly restrictive in regression and can lead to substantial accuracy loss. We propose a relaxation of DP tailored to regression, enforcing parity only at a finite set of quantile levels and/or score thresholds. Concretely, we introduce a novel (ℓ, Z)-fair predictor, which imposes groupwise CDF constraints of the form F f |S=s (z m ) = ℓ m for prescribed pairs (ℓ m , z m ). For this setting, we derive closed-form characterizations of the optimal fair discretized predictor via a Lagrangian dual formulation and quantify the discretization cost, showing that the risk gap to the continuous optimum vanishes as the grid is refined. We further develop a model-agnostic post-processing algorithm based on two samples (labeled for learning a base regressor and unlabeled for calibration), and establish finite-sample guarantees on constraint violation and excess penalized risk. In addition, we introduce two alternative frameworks where we match group and marginal CDF values at selected score thresholds. In both settings, we provide closed-form solutions for the optimal fair discretized predictor. Experiments on synthetic and real datasets illustrate an interpretable fairness-accuracy trade-off, enabling targeted corrections at decision-relevant quantiles or thresholds while preserving predictive performance.
Arthur Charpentier, Christophe Denis, Romuald Elie +2
UQAM · Université du Québec à Montréal, UQAM, Canada · Kyoto University, Japan +7
Exact demographic parity requires the same prediction distribution across sensitive groups. In regression, this can be too restrictive and reduce predictive accuracy. We instead bound every pairwise Wasserstein distance between group-specific prediction distributions by a chosen tolerance, allowing the groups to differ while keeping fairness tunable. On the real line, the prediction and constrained transport problems have the same infimum. If the transport problem attains this value and suitable Monge maps exist, a deterministic transport-form predictor realizes it. Under a standard one-dimensional condition, the problem becomes a bounded quantile optimization that is convex when the conditional expected loss is convex in prediction. Under squared loss with two positively weighted groups, the W1, W2, and W∞ constraints yield soft-thresholding, shrinkage, and clipping. Under the stated conditions, we establish rates for the known-cost plug-in estimator. When the cost is unavailable, we describe an observed-loss computational proxy. Synthetic and California 2018 ACS studies illustrate the tunable fairness-accuracy trade-off and distinguish calibration feasibility from held-out fairness and accuracy.
Fairness-accuracy trade-offs are a central concern in the deployment of fairness-aware machine learning methods. When sensitive attributes are unavailable at inference time-the so called unawareness setting, principled methods for obtaining accurate predictions under relaxed fairness constraints are largely missing. In this work, we address this gap by formulating regression under a demographic parity penalty as an optimal transport problem. Our framework unifies both the aware and unaware settings and characterizes optimal prediction functions via optimal transport maps, under both squared Wasserstein-2 and Total Variation penalties. These results reveal that the choice of penalty reflects fundamentally different fairness philosophies: the Wasserstein penalty induces a smooth, population-wide compromise, while Total Variation enforces exact parity for a subset of individuals. Building on these theoretical characterizations, we propose an algorithm that is simple to implement, computationally efficient, and consistently matches or outperforms state-of-the-art baselines on real-world benchmarks.
M. Generali Lince, V. Divol, R. Flamary +2
Inria, Soda team · CREST, ENSAE, IP Paris · Inria, Fairplay joint team +1