用强化学习优化生成模型的偏见,实现推理时可控的公平性调节。
DSO: Direct Steering Optimization for Bias Mitigation
- 通过强化学习寻找激活值的线性变换,直接优化偏见缓解策略。
- 在视觉语言和语言模型上均实现公平性与性能的最佳平衡。
- 支持用户在推理时灵活调节公平性与模型能力的权衡,适合实际部署。
生成模型常被用于辅助用户决策,例如视觉语言模型(VLMs)识别房间中谁是医生,以帮助视障人士。然而,这些模型的判断会受输入者外貌特征的影响,导致女性医生被忽视等偏见问题。当降低偏见会带来性能下降时,用户对公平性与模型能力的平衡需求各异,因此需要推理时可控制的偏见缓解方法。激活值调制是一种流行的推理时可控性技术,已在大语言模型(LLMs)中展现提升安全性的潜力。但现有调制方法难以有效纠正需在不同人口群体间实现等概率输出的偏见。为此,我们提出直接调制优化(DSO),利用强化学习寻找针对特定偏见缓解目标的线性调制变换,在保持性能的同时实现精确控制。实验表明,DSO 在 VLMs 与 LLMs 上均达到当前最佳的公平性-性能权衡,并为实践者提供推理时的可控调节能力。本工作强调:直接为行为控制目标优化调制策略,比依赖预设启发式方法更有效地干预模型偏见。
原文摘要 · Abstract (English)
Generative models are often deployed to make decisions on behalf of users, such as vision-language models (VLMs) identifying which person in a room is a doctor to help visually impaired individuals. Yet, VLM decisions are influenced by the perceived demographic attributes of people in the input, which can lead to biased outcomes like failing to identify women as doctors. Moreover, when reducing bias leads to performance loss, users may have varying needs for balancing bias mitigation with overall model capabilities, highlighting the demand for methods that enable controllable bias reduction during inference. Activation steering is a popular approach for inference-time controllability that has shown potential in inducing safer behavior in large language models (LLMs). However, we observe that current steering methods struggle to correct biases, where equiprobable outcomes across demographic groups are required. To address this, we propose Direct Steering Optimization (DSO) which uses reinforcement learning to find linear transformations for steering activations, tailored to mitigate bias while maintaining control over model performance. We demonstrate that DSO achieves state-of-the-art trade-off between fairness and capabilities on both VLMs and LLMs, while offering practitioners inference-time control over the trade-off. Overall, our work highlights the benefit of designing steering strategies that are directly optimized to control model behavior, providing more effective bias intervention than methods that rely on pre-defined heuristics for controllability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。