论文

EngramEdit:基于条件记忆实现 LLM 知识解耦更新

EngramEdit: Decoupled Knowledge Updates in LLMs through Conditional Memory

精选理由

研究者在 DeepSeek Engram 这类条件记忆架构上做了知识编辑,改一条事实不用重训模型,CoT 推理准确率是基线的近 3 倍,方法细节都在论文里。

EngramEdit 针对类似 DeepSeek Engram 的条件记忆架构提出知识编辑方法,通过 n-gram 嵌入查找来更新事实知识而冻结 Transformer 主干。该方法先计算让模型在多种表述下都预测更新事实的目标记忆表示,再联合更新共享 n-gram 嵌入,并对频繁复用的嵌入施加更强惩罚以保护无关知识。实验显示编辑成功率接近 100%,在 CoT 提示下的多跳推理准确率约为最强基线的 3 倍,且累积更新后无关知识和通用能力基本保留。

原文 · arXiv: DeepSeek

EngramEdit: Decoupled Knowledge Updates in LLMs through Conditional Memory

Conditional memory architectures such as DeepSeek Engram use input n-grams to look up learned embeddings, expanding the capacity of large language models (LLMs) with limited additional computation. Beyond model scaling, this architecture has demonstrated the potential to decouple factual knowledge storage from general-purpose computation, offering a promising route to updating factual knowledge while keeping the Transformer backbone fixed. Realizing this potential is challenging because different expressions of a fact may activate different n-gram embeddings, while updating shared embeddings can unintentionally change the model's predictions about other facts. We propose EngramEdit for decoupled knowledge updates through conditional memory. EngramEdit first computes target memory representations that make the model predict the updated fact across multiple expressions. It then jointly updates the shared n-gram embeddings to match these targets across expressions and edits, penalizing updates to frequently reused embeddings more strongly to preserve unrelated knowledge. Experiments show that EngramEdit enables independent factual knowledge updates through conditional memory, achieving near-perfect editing success. Revised knowledge is usable across unseen expressions and in multi-hop reasoning, with nearly three times the strongest baseline's accuracy under chain-of-thought (CoT) prompting. Unrelated knowledge and general capabilities are largely preserved even as factual updates accumulate. These findings show that EngramEdit turns conditional memory into an editable knowledge interface, extending its role beyond model scaling to support decoupled knowledge updates.