Search papers, labs, and topics across Lattice.
This paper introduces a novel approach to fair representation learning by evaluating independence through a joint discrepancy between the joint law of the representation and sensitive attribute, rather than relying on conditional laws. This method, termed FRHSIC, utilizes the Hilbert-Schmidt independence criterion (HSIC) to achieve statistical efficiency, converging at a rate of \(O(n^{-1/2})\), which is superior to the \(O(n^{-2/5})\) rate of traditional conditional estimators. The results demonstrate that FRHSIC maintains comparable fairness-accuracy tradeoffs while significantly reducing training time per epoch.
Achieving fair representation without the need for conditional laws could revolutionize how we approach fairness in machine learning models.
Fair representation learning with a continuous sensitive attribute $S$ requires a representation $Z$ that is statistically independent of $S$. Existing criteria, including generalized demographic parity, the expectation of integral probability metrics (EIPM), and mutual information, enforce this independence by averaging a per-value discrepancy between the conditional law $P_{Z \mid S=s}$ and the marginal $P_Z$ over the law of $S$. This approach requires a nonparametric surrogate for the conditional law at each sensitive value. We propose evaluating independence through a single joint discrepancy $d\left(P_{Z, S}, P_Z \otimes P_S\right)$ between the joint law and the product of its marginals. We establish a disintegration identity; on decomposable witness classes it equals the conditional-integral functional that EIPM and generalized demographic parity instantiate. By reaching the same target without the conditional law, this discrepancy can be estimated directly from samples via a dependence statistic rather than conditional smoothing. We take the Hilbert-Schmidt independence criterion (HSIC) as an instance of the joint discrepancy $d$ to investigate the statistical efficiency of replacing the conditional formulation. The HSIC estimator is a closed-form $O\left(n^2\right)$ statistic that converges at the $O\left(n^{-1 / 2}\right)$ rate, in contrast to the nonparametric $O\left(n^{-2 / 5}\right)$ rate of the conditional-route estimators. We prove this instance is equivalent to the conditional maximum mean discrepancy (MMD) integral up to an explicit spectral tail. The corresponding algorithmic implementation, i.e., FRHSIC, attains fairness-accuracy tradeoffs comparable to conditional-route basel es while reducing per-epoch training time.
