arXiv:2412.03009cs.LGcs.CY2024-12被引 1

用强化学习筛选最有价值数据,快速提升模型公平性

Data Acquisition for Improving Model Fairness using Reinforcement Learning

  • 基于数据估值与多臂赌博机,动态筛选对公平性最有益的数据子集
  • 仅需少量新增标注数据,即可显著改善模型在多个数据集上的公平性
  • 适合关注模型偏见缓解的算法工程师和数据标注团队

机器学习系统在医疗、金融和司法等关键领域应用日益广泛,其公平性问题引发关注。现有偏见缓解方法强调高质量数据的重要性,但尚未充分探索数据采集阶段对缓解模型偏见的作用。本文聚焦于通过获取额外标注数据点来快速提升下游模型的公平性。由于数据池中各数据点对公平性的贡献不均,我们提出DataSift框架,基于数据估值思想,结合分区策略与多臂赌博机机制,动态排序并选择最优数据子集进行采集。该框架在多轮迭代中选定一个分区,随机抽取一批数据,利用影响函数估算获取该批数据对模型公平性的提升效果,而无需重新训练模型,从而高效评估并更新各分区价值。我们在多个真实世界与合成数据集上进行实验,结果表明:仅需少量新增数据点,即可显著提升模型公平性。

原文摘要 · Abstract (English)

Machine learning systems are increasingly being used in critical decision making such as healthcare, finance, and criminal justice. Concerns around their fairness have resulted in several bias mitigation techniques that emphasize the need for high-quality data to ensure fairer decisions. However, the role of earlier stages of machine learning pipelines in mitigating model bias has not been explored well. In this paper, we focus on the task of acquiring additional labeled data points for training the downstream machine learning model to rapidly improve its fairness. Since not all data points in a data pool are equally beneficial to the task of fairness, we generate an ordering in which data points should be acquired. We present DataSift, a data acquisition framework based on the idea of data valuation that relies on partitioning and multi-armed bandits to determine the most valuable data points to acquire. Over several iterations, DataSift selects a partition and randomly samples a batch of data points from the selected partition, evaluates the benefit of acquiring the batch on model fairness, and updates the utility of partitions depending on the benefit. To further improve the effectiveness and efficiency of evaluating batches, we leverage influence functions that estimate the effect of acquiring a batch without retraining the model. We empirically evaluate DataSift on several real-world and synthetic datasets and show that the fairness of a machine learning model can be significantly improved even while acquiring a few data points.

数据采集模型公平性强化学习数据估值

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。