arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.00363cs.CLcs.LG
收藏

修改Transformer模型中的内存

Modifying Memories in Transformer Models

University of Maryland, College Park(马里兰大学帕克分校);Google Research(谷歌研究院)

Chen Zhu, Ankit Singh Rawat, Manzil Zaheer, Srinadh Bhojanapalli, Daliang Li, Felix Yu, Sanjiv Kumar

摘要:

大型Transformer模型在许多自然语言任务中取得了令人印象深刻的性能。特别地,基于Transformer的语言模型已经被证明在以其大量参数对事实知识进行编码方面具有很强的能力。虽然提高Transformers的记忆和概括能力的任务已经得到了广泛的研究,但如何使transformers忘记特定的旧事实并记住新事实还不太清楚。在本文中,我们提出了一个新的任务\emph{explicitly modifying specific factual knowledge in Transformer models while ensuring the model performance does not degrade on the unmodified facts}。此任务在许多情况下都很有用,例如更新陈旧的知识,保护隐私,以及消除存储在模型中的意外偏见。我们对几种方法进行了基准测试,这些方法在这项任务上提供了自然的基线性能。这导致了Transformer模型的关键组件的发现,这些组件对于知识修改特别有效。这项工作还提供了不同的训练阶段(如预训练和微调)对记忆和知识修改的作用的见解。

英文摘要:

Large Transformer models have achieved impressive performance in many natural language tasks. In particular, Transformer based language models have been shown to have great capabilities in encoding factual knowledge in their vast amount of parameters. While the tasks of improving the memorization and generalization of Transformers have been widely studied, it is not well known how to make transformers forget specific old facts and memorize new ones. In this paper, we propose a new task of \emph{explicitly modifying specific factual knowledge in Transformer models while ensuring the model performance does not degrade on the unmodified facts}. This task is useful in many scenarios, such as updating stale knowledge, protecting privacy, and eliminating unintended biases stored in the models. We benchmarked several approaches that provide natural baseline performances on this task. This leads to the discovery of key components of a Transformer model that are especially effective for knowledge modifications. The work also provides insights into the role that different training phases (such as pretraining and fine-tuning) play towards memorization and knowledge modification.

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

↑