arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2202.09662cs.CLcs.AI
收藏

基于transformer的语言模型中减轻毒性的奖励建模

Reward Modeling for Mitigating Toxicity in Transformer-based Language Models

Concordia University(康考迪亚大学)

Farshid Faal, Ketra Schmitt, Jia Yuan Yu

摘要:

基于转换器的语言模型能够生成流畅的文本,并且能够有效地适应各种自然语言生成任务。然而,在大型未标记的Web文本语料库上预训练的语言模型已被证明患有退化的有毒内容和社会偏见行为,从而阻碍了它们的安全部署。人们提出了各种解毒方法来减轻语言模型的毒性;然而,当提示包含与性别、种族或宗教相关的特定社会身份时,这些方法难以解毒语言模型。在这项研究中,我们提出了一种基于强化学习的方法,用于减轻语言模型中的毒性。我们解决了语言模型的安全性的挑战,并提出了一个新的奖励模型,能够检测有毒的内容,并减轻对社会身份的毒性预测无意中的偏见。实验结果表明,基于Reinforce-Defenefy的语言模型去毒方法在自动评价指标上优于现有的去毒方法,表明了该方法在语言模型去毒方面的能力,并且在生成的内容中不容易产生对社会身份的非故意偏见。

英文摘要:

Transformer-based language models are able to generate fluent text and be efficiently adapted across various natural language generation tasks. However, language models that are pretrained on large unlabeled web text corpora have been shown to suffer from degenerating toxic content and social bias behaviors, consequently hindering their safe deployment. Various detoxification methods were proposed to mitigate the language model's toxicity; however, these methods struggled to detoxify language models when conditioned on prompts that contain specific social identities related to gender, race, or religion. In this study, we propose Reinforce-Detoxify; A reinforcement learning-based method for mitigating toxicity in language models. We address the challenge of safety in language models and propose a new reward model that is able to detect toxic content and mitigate unintended bias towards social identities in toxicity prediction. The experiments demonstrate that the Reinforce-Detoxify method for language model detoxification outperforms existing detoxification approaches in automatic evaluation metrics, indicating the ability of our approach in language model detoxification and less prone to unintended bias toward social identities in generated content.

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

↑