用可控合成数据精调有偏模型,提升公平性且不损失性能
AIM-Fair: Advancing Algorithmic Fairness via Selectively Fine-Tuning Biased Models with Contextual Synthetic Data
- 用上下文感知的LLM生成提示,驱动扩散模型生成高质量合成数据
- 仅更新对偏见敏感、对领域漂移不敏感的模型参数,有效缓解偏差
- 无需真实标签即可改进公平性,适合缺乏标注数据的场景
近期生成模型的发展推动了利用AI生成数据改善模型公平性的研究。然而,现有方法在合成数据的多样性和质量上常受限,导致公平性和整体性能下降。此外,许多方法依赖于成本高昂的人工标注群体标签。本文提出AIM-Fair,旨在克服这些局限,充分发挥先进生成模型在促进算法公平性方面的潜力。我们采用从无群体标签的真实数据训练的有偏模型出发,通过状态领先扩散模型生成的无偏合成数据进行微调。针对该范式中的两个关键挑战——合成数据质量低、真实与合成数据间存在领域和偏见差距——我们提出上下文合成数据生成(CSDG),利用文本到图像扩散模型(T2I)结合上下文感知大模型生成的提示,确保合成数据的多样性并控制偏见。为解决领域和偏见偏移问题,我们设计了一种新型选择性微调方案,仅更新对偏见更敏感、对领域变化更不敏感的模型参数。在CelebA和UTKFace数据集上的实验表明,AIM-Fair在保持模型效用的同时提升了公平性,优于全量和部分微调方法。
原文摘要 · Abstract (English)
Recent advances in generative models have sparked research on improving model fairness with AI-generated data. However, existing methods often face limitations in the diversity and quality of synthetic data, leading to compromised fairness and overall model accuracy. Moreover, many approaches rely on the availability of demographic group labels, which are often costly to annotate. This paper proposes AIM-Fair, aiming to overcome these limitations and harness the potential of cutting-edge generative models in promoting algorithmic fairness. We investigate a fine-tuning paradigm starting from a biased model initially trained on real-world data without demographic annotations. This model is then fine-tuned using unbiased synthetic data generated by a state-of-the-art diffusion model to improve its fairness. Two key challenges are identified in this fine-tuning paradigm, 1) the low quality of synthetic data, which can still happen even with advanced generative models, and 2) the domain and bias gap between real and synthetic data. To address the limitation of synthetic data quality, we propose Contextual Synthetic Data Generation (CSDG) to generate data using a text-to-image diffusion model (T2I) with prompts generated by a context-aware LLM, ensuring both data diversity and control of bias in synthetic data. To resolve domain and bias shifts, we introduce a novel selective fine-tuning scheme in which only model parameters more sensitive to bias and less sensitive to domain shift are updated. Experiments on CelebA and UTKFace datasets show that our AIM-Fair improves model fairness while maintaining utility, outperforming both fully and partially fine-tuned approaches to model fairness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。