arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2404.13752cs.LGcs.AIcs.CLcs.CRmath.OC
收藏

对抗表征工程:一种面向大语言模型的通用模型编辑框架

Adversarial Representation Engineering: A General Model Editing Framework for Large Language Models

Peking University(北京大学);Singapore Management University(新加坡管理大学)

Yihao Zhang, Zeming Wei, Jun Sun, Meng Sun

AI总结:

针对大语言模型内部机制理解与修正的需求,该研究提出对抗表征工程(ARE)框架,以表征传感器为编辑预言机,实现统一可解释的概念性模型编辑且不损基准性能,多任务实验验证其有效性。

摘要:

随着大语言模型(Large Language Models, LLMs)的快速发展并取得显著成功,理解与修正其内部复杂机制已成为亟待解决的问题。近期研究尝试通过内部表征的视角解读其行为,但开发实用高效的方法以将这些表征应用于通用灵活的模型编辑仍具挑战。本研究探索如何利用表征工程的洞见指导LLM编辑,将表征传感器用作编辑预言机。我们首先明确了编辑过程中稳健可靠传感器的重要性,随后提出对抗表征工程(Adversarial Representation Engineering, ARE)框架,为概念性模型编辑提供统一且可解释的方法,且不会损害基准性能。在多项任务上的实验证明了ARE在各类模型编辑场景中的有效性。我们的代码与数据可在https://github.com/Zhang-Yihao/Adversarial-Representation-Engineering获取。

英文摘要:

Since the rapid development of Large Language Models (LLMs) has achieved remarkable success, understanding and rectifying their internal complex mechanisms has become an urgent issue. Recent research has attempted to interpret their behaviors through the lens of inner representation. However, developing practical and efficient methods for applying these representations for general and flexible model editing remains challenging. In this work, we explore how to leverage insights from representation engineering to guide the editing of LLMs by deploying a representation sensor as an editing oracle. We first identify the importance of a robust and reliable sensor during editing, then propose an Adversarial Representation Engineering (ARE) framework to provide a unified and interpretable approach for conceptual model editing without compromising baseline performance. Experiments on multiple tasks demonstrate the effectiveness of ARE in various model editing scenarios. Our code and data are available at https://github.com/Zhang-Yihao/Adversarial-Representation-Engineering.

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

↑