用上下文增强的反事实数据提升模型去偏效果,不损失语言能力。
Context-Aware Counterfactual Data Augmentation for Gender Bias Mitigation in Language Models
- 利用大模型生成更符合语境的反事实数据,提升去偏语料质量
- 通过不确定性过滤剔除低质反事实样本,优化微调数据
- 在性别偏见基准上实现去偏与语言能力双提升,适合模型去偏研究者
在微调语言模型中缓解社会偏见时,常面临语言建模能力下降的问题。反事实数据增强(CDA)虽广泛使用,但生成的合成数据可能偏离真实分布,或生成过于简单的反事实,忽略敏感属性(如性别)在预训练语料中的社会背景。为此,我们提出一种简单而有效的上下文增强型CDA方法——Context-CDA,利用大语言模型提升去偏语料的多样性和语境相关性。通过增强上下文以缩小去偏语料与预训练数据之间的差异,确保更好对齐,从而保持语言建模能力。进一步采用基于不确定性的过滤机制,剔除目标小模型认为低质量的生成反事实,进一步提升微调语料质量。在性别偏见基准上的实验表明,Context-CDA有效缓解偏见,同时不牺牲语言建模性能,并通过分析下一个词预测概率的分布变化,揭示社会偏见特征。
原文摘要 · Abstract (English)
A challenge in mitigating social bias in fine-tuned language models (LMs) is the potential reduction in language modeling capability, which can harm downstream performance. Counterfactual data augmentation (CDA), a widely used method for fine-tuning, highlights this issue by generating synthetic data that may align poorly with real-world distributions or creating overly simplistic counterfactuals that ignore the social context of altered sensitive attributes (e.g., gender) in the pretraining corpus. To address these limitations, we propose a simple yet effective context-augmented CDA method, Context-CDA, which uses large LMs to enhance the diversity and contextual relevance of the debiasing corpus. By minimizing discrepancies between the debiasing corpus and pretraining data through augmented context, this approach ensures better alignment, enhancing language modeling capability. We then employ uncertainty-based filtering to exclude generated counterfactuals considered low-quality by the target smaller LMs (i.e., LMs to be debiased), further improving the fine-tuning corpus quality. Experimental results on gender bias benchmarks demonstrate that Context-CDA effectively mitigates bias without sacrificing language modeling performance while offering insights into social biases by analyzing distribution shifts in next-token generation probabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。