用多目标强化学习让大模型编辑知识更可靠、通用且精准。
Towards Reliable, Generalizable, and Specific In-Context Knowledge Editing via Multi-Objective Reinforcement Learning

- 设计多目标强化学习框架,全局优化提示的结构与内容平衡。
- 在Llama-3.2上将知识编辑成功率提升至92.0%,保留率提高23.0%。
- 适合需要高精度知识更新的黑盒大模型应用场景。
大语言模型虽强大,但其参数化知识在预训练结束后无法更新。知识编辑通过无需重训练的方式调整模型对特定事实的认知。其中,上下文知识编辑因其无需训练且适用于黑盒模型而受到关注。现有基于强化学习的方法虽能自适应提示构造,但仍未能兼顾可靠性、通用性和特异性三者之间的平衡。本文提出多目标上下文知识编辑(MO-IKE),将提示构造建模为约束马尔可夫决策过程,训练动态检索器以协同优化多个目标,实现更均衡、全局一致的提示生成。在Llama-3.2上,MO-IKE将编辑成功率(可靠性)从85.0%提升至92.0%,重述一致性(通用性)从77%提升至79%,同时保留率(特异性)相比先前强化学习方法提高23.0%。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are powerful but limited by static parametric knowledge that becomes outdated once pretraining ends. Knowledge editing addresses this problem by updating model behavior on target facts without full retraining. In particular, in-context knowledge editing has gained attention because it is training-free and readily applicable to black-box LLMs. Recent reinforcement learning (RL)-based approaches improve over fixed retrieval strategies by adapting prompt construction to the quantity-quality trade-off. Despite initial success, they fail to model the prompt as a structured entity under the distinct and often competing objectives of reliability, generality, and specificity. Previous methods largely optimize a single objective and make decisions over only part of the prompt construction process, thereby overlooking both the balance of different objectives and the global organization of demonstrations. We propose Multi-Objective In-context Knowledge Editing (MO-IKE), a multi-objective RL algorithm that formulates prompt construction for in-context knowledge editing as a Constrained Markov Decision Process. MO-IKE trains a dynamic retriever to optimize competing objectives in knowledge editing, enabling more balanced and globally coherent prompt construction. On Llama-3.2, MO-IKE improves edit success (reliability) from 85.0% to 92.0%, paraphrase consistency (generality) from 77% to 79%, while increasing retention rate (specificity) by 23.0% compared to prior RL-based methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。