Fast and Efficient MMD-based Fair PCA via Optimization over Stiefel Manifold
Not on the map. Why?
Abstract (arXiv)
This paper defines fair principal component analysis (PCA) as minimizing the maximum mean discrepancy (MMD) between dimensionality-reduced conditional distributions of different protected classes. The incorporation of MMD naturally leads to an exact and tractable mathematical formulation of fairness with good statistical properties. We formulate the problem of fair PCA subject to MMD constraints as a non-convex optimization over the Stiefel manifold and solve it using the Riemannian Exact Penalty Method with Smoothing (REPMS; Liu and Boumal, 2019). Importantly, we provide local optimality guarantees and explicitly show the theoretical effect of each hyperparameter in practical settings, extending previous results. Experimental comparisons based on synthetic and UCI datasets show that our approach outperforms prior work in explained variance, fairness, and runtime.
Key facts. Cited by 18 papers in the map (data through 7 Oct 2026). Builds on 19 of its 52 references that are in the map. First posted on arXiv 23 Sep 2021.
Builds on
19 of 52 references.
References · 19
- 2016Equality of Opportunity in Supervised Learningcited by 2,000
- 2019A Survey on Bias and Fairness in Machine Learningcited by 1,911
- 2014Certifying and removing disparate impactcited by 715
- 2016Fairness Beyond Disparate Treatment & Disparate Impact: Learning Classification without Disparate Mistreatmentcited by 368
- 2018Learning Adversarially Fair and Transferable Representationscited by 353
- 2018Fair Clustering Through Fairletscited by 176
- 2017Data Decisions and Theoretical Implications when Adversarially Learning Fair Representationscited by 165
- 2019Fair Algorithms for Clusteringcited by 98
- 2018The Price of Fair PCA: One Extra Dimensioncited by 67
- 2021Fair Feature Distillation for Visual Recognitioncited by 59
- 2019Toward a better trade-off between performance and fairness with kernel-based distribution matchingcited by 36
- 2019A Brief Introduction to Manifold Optimizationcited by 35
7 more
- 2018Convex Formulations for Fair Principal Component Analysiscited by 20
- 2014A Stochastic PCA and SVD Algorithm with an Exponential Convergence Ratecited by 18
- 2019Simple algorithms for optimization on Riemannian manifolds with constraintscited by 14
- 2013Stochastic Optimization of PCA with Capped MSGcited by 6
- 2016Orthogonal Sparse PCA and Covariance Estimation via Procrustes Reformulationcited by 5
- 2020Covariance-Robust Dynamic Watermarkingcited by 1
- 2021Manifold Optimization for High Accuracy Spatial Location Estimation Using Ultrasound Wavescited by 1
Built on by
Citing papers per year: 2022: 3; 2023: 5; 2024: 3; 2025: 7.
18 papers.
Sort:
- 2023Efficient fair PCA for fair representation learningcited by 23
- 2022MMD-B-Fair: Learning Fair Representations with Statistical Testingcited by 13
- 2022First-Order Algorithms for Min-Max Optimization in Geodesic Metric Spacescited by 10
- 2025Fair Representation Learning for Continuous Sensitive Attributes using Expectation of Integral Probability Metricscited by 7
- 2022A novel approach for Fair Principal Component Analysis based on eigendecompositioncited by 6
- 2024Fair CoVariance Neural Networkscited by 6
- 2023Fair Streaming Principal Component Analysis: Statistical and Algorithmic Viewpointcited by 5
- 2023Last-Iterate Convergence of Adaptive Riemannian Gradient Descent for Equilibrium Computationcited by 2
- 2024Specification Overfitting in Artificial Intelligencecited by 2
- 2024On Socially Fair Low-Rank Approximation and Column Subset Selectioncited by 2
- 2023When Collaborative Filtering is not Collaborative: Unfairness of PCA for Recommendationscited by 1
- 2023Learning Fair Representations with High-Confidence Guaranteescited by 1
6 more
- 2025Hidden Convexity of Fair PCA and Fast Solver via Eigenvalue Optimizationcited by 1
- 2025PLS-based approach for fair representation learningcited by 0
- 2025Achieving Fair PCA Using Joint Eigenvalue Decompositioncited by 0
- 2025Fair PCA, One Component at a Timecited by 0
- 2025Incorporating Fairness Constraints into Archetypal Analysiscited by 0
- 2025Fair Representation Learning with Controllable High Confidence Guarantees via Adversarial Inferencecited by 0
- 2025Fair Representation Learning with Controllable High Confidence Guarantees via Adversarial Inference23 Oct 2025
- 2025Incorporating Fairness Constraints into Archetypal Analysis16 Jul 2025
- 2025Fair Representation Learning for Continuous Sensitive Attributes using Expectation of Integral Probability Metrics9 May 2025
- 2025Fair PCA, One Component at a Time27 Mar 2025
- 2025Hidden Convexity of Fair PCA and Fast Solver via Eigenvalue Optimization1 Mar 2025
- 2025Achieving Fair PCA Using Joint Eigenvalue Decomposition24 Feb 2025
- 2025PLS-based approach for fair representation learning22 Feb 2025
- 2024On Socially Fair Low-Rank Approximation and Column Subset Selection8 Dec 2024
- 2024Fair CoVariance Neural Networks13 Sep 2024
- 2024Specification Overfitting in Artificial Intelligence13 Mar 2024
- 2023Fair Streaming Principal Component Analysis: Statistical and Algorithmic Viewpoint28 Oct 2023
- 2023Learning Fair Representations with High-Confidence Guarantees23 Oct 2023
6 more
- 2023When Collaborative Filtering is not Collaborative: Unfairness of PCA for Recommendations15 Oct 2023
- 2023Last-Iterate Convergence of Adaptive Riemannian Gradient Descent for Equilibrium Computation29 Jun 2023
- 2023Efficient fair PCA for fair representation learning26 Feb 2023
- 2022MMD-B-Fair: Learning Fair Representations with Statistical Testing15 Nov 2022
- 2022A novel approach for Fair Principal Component Analysis based on eigendecomposition24 Aug 2022
- 2022First-Order Algorithms for Min-Max Optimization in Geodesic Metric Spaces4 Jun 2022
- 2023Efficient fair PCA for fair representation learningcited by 23
- 2022MMD-B-Fair: Learning Fair Representations with Statistical Testingcited by 13
- 2022First-Order Algorithms for Min-Max Optimization in Geodesic Metric Spacescited by 10
- 2025Fair Representation Learning for Continuous Sensitive Attributes using Expectation of Integral Probability Metricscited by 7
- 2022A novel approach for Fair Principal Component Analysis based on eigendecompositioncited by 6
- 2024Fair CoVariance Neural Networkscited by 6
- 2023Fair Streaming Principal Component Analysis: Statistical and Algorithmic Viewpointcited by 5
- 2023Last-Iterate Convergence of Adaptive Riemannian Gradient Descent for Equilibrium Computationcited by 2
- 2024Specification Overfitting in Artificial Intelligencecited by 2
- 2024On Socially Fair Low-Rank Approximation and Column Subset Selectioncited by 2
- 2023When Collaborative Filtering is not Collaborative: Unfairness of PCA for Recommendationscited by 1
- 2023Learning Fair Representations with High-Confidence Guaranteescited by 1
12 most cited of 18 All 18 (JSON)
- 2025Fair Representation Learning with Controllable High Confidence Guarantees via Adversarial Inference23 Oct 2025
- 2025Incorporating Fairness Constraints into Archetypal Analysis16 Jul 2025
- 2025Fair Representation Learning for Continuous Sensitive Attributes using Expectation of Integral Probability Metrics9 May 2025
- 2025Fair PCA, One Component at a Time27 Mar 2025
- 2025Hidden Convexity of Fair PCA and Fast Solver via Eigenvalue Optimization1 Mar 2025
- 2025Achieving Fair PCA Using Joint Eigenvalue Decomposition24 Feb 2025
- 2025PLS-based approach for fair representation learning22 Feb 2025
- 2024On Socially Fair Low-Rank Approximation and Column Subset Selection8 Dec 2024
- 2024Fair CoVariance Neural Networks13 Sep 2024
- 2024Specification Overfitting in Artificial Intelligence13 Mar 2024
- 2023Fair Streaming Principal Component Analysis: Statistical and Algorithmic Viewpoint28 Oct 2023
- 2023Learning Fair Representations with High-Confidence Guarantees23 Oct 2023
12 newest of 18 All 18 (JSON)