arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.03300cs.LGcs.CY
收藏

在分布偏移下追求公平性:一种模型权重扰动方法

Chasing Fairness Under Distribution Shift: A Model Weight Perturbation Approach

Texas A&M University(德克萨斯农工大学);Rice University(莱斯大学);Samsung Electronics America(三星电子美国公司)

Zhimeng Jiang, Xiaotian Han, Hongye Jin, Guanchu Wang, Rui Chen, Na Zou, Xia Hu

AI总结:

针对分布偏移下机器学习公平性下降的问题,通过理论分析分布偏移与模型权重扰动的联系,提出鲁棒公平性正则化(RFR)方法,在权重扰动最坏情况下优化模型,实验证明其在公平性与准确性权衡上优于基线方法。

摘要:

机器学习中的公平性近年来引起了广泛关注。针对分布内数据提升算法公平性的方法,在分布偏移下可能表现不佳。本文首先从理论上证明了分布偏移、数据扰动与模型权重扰动之间的内在联系。随后,我们分析了保证目标数据集公平性(即人口统计平价)的充分条件,包括源数据集的公平性,以及对于每个敏感属性,源数据集与目标数据集之间的低预测差异。受这些充分条件启发,我们提出了鲁棒公平性正则化(RFR),通过考虑每个敏感属性组在模型权重扰动球内的最坏情况来优化模型。我们在多个数据集上评估了所提出的RFR算法在合成和真实分布偏移下的有效性。实验结果表明,与多个基线方法相比,RFR实现了更好的公平性与准确性的权衡性能。源代码可在 https://github.com/zhimengj0326/RFR_NeurIPS 获取。

英文摘要:

Fairness in machine learning has attracted increasing attention in recent years. The fairness methods improving algorithmic fairness for in-distribution data may not perform well under distribution shifts. In this paper, we first theoretically demonstrate the inherent connection between distribution shift, data perturbation, and model weight perturbation. Subsequently, we analyze the sufficient conditions to guarantee fairness (i.e., low demographic parity) for the target dataset, including fairness for the source dataset, and low prediction difference between the source and target datasets for each sensitive attribute group. Motivated by these sufficient conditions, we propose robust fairness regularization (RFR) by considering the worst case within the model weight perturbation ball for each sensitive attribute group. We evaluate the effectiveness of our proposed RFR algorithm on synthetic and real distribution shifts across various datasets. Experimental results demonstrate that RFR achieves better fairness-accuracy trade-off performance compared with several baselines. The source code is available at \url{https://github.com/zhimengj0326/RFR_NeurIPS23}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑