中文

模型编辑的统一框架

机器学习 2024-10-10 v5 人工智能 计算与语言

摘要

ROME 和 MEMIT 在很大程度上被认为是两种不同的模型编辑算法,它们之间的主要区别在于执行批量编辑的能力。在本文中,我们将这两种算法统一在一个概念框架下,优化相同的目标,我们称之为保留-记忆目标。ROME 使用等式约束来优化此目标,一次执行一次编辑,而 MEMIT 则采用更灵活的最小二乘约束,允许批量编辑。我们推广了 ROME,并以 EMMET(一种用于 Transformer 的等式约束质量模型编辑算法,一种新的批量记忆编辑算法)的形式实现了带等式约束的批量编辑。EMMET 可以执行批量大小高达 10,000 的批量编辑,在多个维度上的性能与 MEMIT 非常相似。随着 EMMET 的引入,我们真正统一了 ROME 和 MEMIT,并表明这两种算法在优化目标、能力(单次和批量编辑)、模型编辑性能及其局限性方面是等效的。

关键词

引用

@article{arxiv.2403.14236,
  title  = {A Unified Framework for Model Editing},
  author = {Akshat Gupta and Dev Sajnani and Gopala Anumanchipalli},
  journal= {arXiv preprint arXiv:2403.14236},
  year   = {2024}
}

备注

EMNLP 2024 Findings