arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2408.09489cs.CLcs.AI

REFINE-LM:通过强化学习缓解语言模型刻板印象

REFINE-LM: Mitigating Language Model Stereotypes via Reinforcement Learning

ADAPT Centre, Trinity College Dublin(都柏林圣三一学院ADAPT中心);CRIL CNRS, Univ Artois(阿尔图瓦大学CRIL法国国家科学研究中心);INRIA/IRISA Rennes(雷恩INRIA/IRISA研究所);U. Lorraine, CNRS, LORIA(洛林大学法国国家科学研究中心LORIA研究所);U. Lisboa, INESC-I(里斯本大学INESC-I研究所)

Rameez Qureshi, Naïm Es-Sebbani, Luis Galárraga, Yvette Graham, Miguel Couceiro, Zied Bouraoui

AI总结:

针对语言模型存在的多类刻板印象偏差问题,提出无需微调、标注少、成本低的REFINE-LM强化学习去偏方法,可在保留模型性能的同时有效降低多类偏差。

摘要:

随着(大)语言模型的推出,人们对这类模型可能从训练数据中继承的意外偏差产生了极大担忧。大量研究表明,这类模型会传播性别刻板印象,以及地理和种族偏差等多种偏差。现有研究通过数据预处理和嵌入去偏来解决该问题,但所提方法需要大量计算资源和标注工作,且仅局限于特定类型的偏差。为解决这些问题,我们提出REFINE-LM,这是一种无需任何微调即可利用强化学习处理不同类型偏差的去偏方法。通过在语言模型的词概率分布之上训练一个简单模型,我们的偏差无关强化学习方法无需人工标注或大量计算资源即可实现模型去偏。在包括多个语言模型在内的广泛模型上开展的实验表明,我们的方法(i)在保留语言模型性能的同时显著降低了刻板印象偏差;(ii)适用于不同类型的偏差,可在性别、种族、宗教和国籍类偏差等场景中泛化;(iii)训练成本不高。

英文摘要:

With the introduction of (large) language models, there has been significant concern about the unintended bias such models may inherit from their training data. A number of studies have shown that such models propagate gender stereotypes, as well as geographical and racial bias, among other biases. While existing works tackle this issue by preprocessing data and debiasing embeddings, the proposed methods require a lot of computational resources and annotation effort while being limited to certain types of biases. To address these issues, we introduce REFINE-LM, a debiasing method that uses reinforcement learning to handle different types of biases without any fine-tuning. By training a simple model on top of the word probability distribution of a LM, our bias agnostic reinforcement learning method enables model debiasing without human annotations or significant computational resources. Experiments conducted on a wide range of models, including several LMs, show that our method (i) significantly reduces stereotypical biases while preserving LMs performance; (ii) is applicable to different types of biases, generalizing across contexts such as gender, ethnicity, religion, and nationality-based biases; and (iii) it is not expensive to train.

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

↑