大型语言模型(LLMs)革新了现代自然语言处理(NLP),显著提升了各种任务性能。然而,现已部署的LLMs依然会受到知识截断和谬误问题的限制。例如,ChatGPT和LlaMA等LLMs仅具备截至训练最后时点的信息,也可能会因预训练数据中的偏见和差异生成不准确或误导性的输出。因此,高效更新LLMs的参数化知识进而调整特定行为,变得至关重要。
尽管微调和参数高效微调可以修改LLMs,但成本较高,还可能导致LLMs失去预训练所得能力,并且其修改也不总能泛化到相关输入。另一种方法是,使用手动编写或检索的提示影响LLMs的输出,但这类方法没有参数更新,可靠性不足。
为了解决上述问题,浙江大学和东海实验室的研究团队提出了一个易于使用的LLMs知识编辑框架——EasyEdit,该框架支持各种知识编辑方法,且可以轻松应用于众多LLMs,如T5、GPT-J和LlaMA等。
相关研究论文以“EasyEdit: An Easy-to-use Knowledge Editing Framework for Large Language Models”为题,已发表在预印本网站arXiv上。相关数据集、Google Colab教程和文档也已发布在GitHub上。为了使不相关输入的影响最小化,并迅速有效地修改LLMs的行为,一种可行的解决方案是知识编辑。
关于LLMs的知识编辑研究在各种任务和设置下取得显著进展,包括Memory based、Meta-learning和Locate-Then-Edit三类方法。然而,目前关于LLMs知识编辑的研究在实现和任务设置上的差异妨碍了知识编辑统一和综合框架的发展。值得注意的是,这种复杂性阻碍了不同方法之间有效性和可行性的直接比较,也使得创建新的知识编辑方法变得复杂。
据介绍,EasyEdit框架整合了各种编辑技术,支持在不同LLMs之间自由组合模块。通过统一的框架和接口,EasyEdit能使用户迅速理解并应用包含在该框架中的主流知识编辑方法。EasyEdit具有统一的Editor、Method和Evaluate框架,分别代表编辑场景、编辑技术和评估方法。
此外,EasyEdit还提供了五个评估编辑方法性能的关键指标,包括可靠性(Reliability)、泛化性(Generalization)、局部性(Locality)、可移植性(Portability)和效率(Efficiency)。为验证知识编辑在LLMs中的应用潜力,研究团队选用了参数庞大的LlaMA 2模型,并利用ZsRE数据集(QA数据集)来测试知识编辑将大量一般事实关联整合进模型的能力。
测试结果证明,EasyEdit在可靠性和泛化性方面超越了传统的微调方法。据论文描述,作为一个易于使用的LLMs知识编辑框架,EasyEdit支持许多尖端方法和不同种类的LLMs。通过一种可控且有针对性的方式对LLMs进行编辑,该技术可能为各种自然语言处理任务中的知识增强和适应带来新的可能。
未来,该研究团队将继续把先进的编辑技术、创新特性(如编辑个性)和新的编辑目标(如针对多模态LLMs的知识编辑)融入EasyEdit中,从而促进进一步的研究,并在自然语言处理领域激发新的思想。