On the use of graph models to achieve individual and group fairness
Arturo Pérez-Peralta, Sandra Benítez-Peña, Rosa E. Lillo
TL;DR
This work addresses the challenge of achieving both individual and group fairness in machine learning by introducing Fair Sheaf Diffusion (FSD), a topological framework that uses cellular sheaves and graph-based diffusion to encode and enforce fairness constraints. By shaping the diffusion dynamics through tailored sheaves and topologies (e.g., identity, vector/bottleneck, kNN, unit ball, and subset topologies), the method projects data into a bias-free space while providing closed-form SHAP explanations for interpretability. The paper develops a unified theory of fair classification with group and individual metrics, derives diffusion-based mechanisms to minimize a fairness-driven energy, and demonstrates trade-offs between accuracy and fairness on simulations and standard benchmarks (German, Compas, Adult) via Pareto fronts and grid searches. A key strength is the ability to combine multiple topologies to balance multiple fairness criteria, offering pre-/in-/post-processing flexibility and insightful explanations, with potential extensions to more complex topologies and non-linear diffusion. Overall, FSD advances responsible AI by tying algebraic-topological constructs to practical fairness objectives and interpretability, enabling principled fairness-accuracy trade-offs in real-world deployments.
Abstract
Machine Learning algorithms are ubiquitous in key decision-making contexts such as justice, healthcare and finance, which has spawned a great demand for fairness in these procedures. However, the theoretical properties of such models in relation with fairness are still poorly understood, and the intuition behind the relationship between group and individual fairness is still lacking. In this paper, we provide a theoretical framework based on Sheaf Diffusion to leverage tools based on dynamical systems and homology to model fairness. Concretely, the proposed method projects input data into a bias-free space that encodes fairness constrains, resulting in fair solutions. Furthermore, we present a collection of network topologies handling different fairness metrics, leading to a unified method capable of dealing with both individual and group bias. The resulting models have a layer of interpretability in the form of closed-form expressions for their SHAP values, consolidating their place in the responsible Artificial Intelligence landscape. Finally, these intuitions are tested on a simulation study and standard fairness benchmarks, where the proposed methods achieve satisfactory results. More concretely, the paper showcases the performance of the proposed models in terms of accuracy and fairness, studying available trade-offs on the Pareto frontier, checking the effects of changing the different hyper-parameters, and delving into the interpretation of its outputs.
