Knowledge Editing for Large Language Model with Knowledge Neuronal Ensemble

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Yongchang, Zhu, Yujin, Yan, Tao, Fan, Shijian, Wu, Gang, Xu, Liang
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916545753513984
author Li, Yongchang
Zhu, Yujin
Yan, Tao
Fan, Shijian
Wu, Gang
Xu, Liang
author_facet Li, Yongchang
Zhu, Yujin
Yan, Tao
Fan, Shijian
Wu, Gang
Xu, Liang
contents As real-world knowledge is constantly evolving, ensuring the timeliness and accuracy of a model's knowledge is crucial. This has made knowledge editing in large language models increasingly important. However, existing knowledge editing methods face several challenges, including parameter localization coupling, imprecise localization, and a lack of dynamic interaction across layers. In this paper, we propose a novel knowledge editing method called Knowledge Neuronal Ensemble (KNE). A knowledge neuronal ensemble represents a group of neurons encoding specific knowledge, thus mitigating the issue of frequent parameter modification caused by coupling in parameter localization. The KNE method enhances the precision and accuracy of parameter localization by computing gradient attribution scores for each parameter at each layer. During the editing process, only the gradients and losses associated with the knowledge neuronal ensemble are computed, with error backpropagation performed accordingly, ensuring dynamic interaction and collaborative updates among parameters. Experimental results on three widely used knowledge editing datasets show that the KNE method significantly improves the accuracy of knowledge editing and achieves, or even exceeds, the performance of the best baseline methods in portability and locality metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2412_20637
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Knowledge Editing for Large Language Model with Knowledge Neuronal Ensemble
Li, Yongchang
Zhu, Yujin
Yan, Tao
Fan, Shijian
Wu, Gang
Xu, Liang
Computation and Language
68T50
As real-world knowledge is constantly evolving, ensuring the timeliness and accuracy of a model's knowledge is crucial. This has made knowledge editing in large language models increasingly important. However, existing knowledge editing methods face several challenges, including parameter localization coupling, imprecise localization, and a lack of dynamic interaction across layers. In this paper, we propose a novel knowledge editing method called Knowledge Neuronal Ensemble (KNE). A knowledge neuronal ensemble represents a group of neurons encoding specific knowledge, thus mitigating the issue of frequent parameter modification caused by coupling in parameter localization. The KNE method enhances the precision and accuracy of parameter localization by computing gradient attribution scores for each parameter at each layer. During the editing process, only the gradients and losses associated with the knowledge neuronal ensemble are computed, with error backpropagation performed accordingly, ensuring dynamic interaction and collaborative updates among parameters. Experimental results on three widely used knowledge editing datasets show that the KNE method significantly improves the accuracy of knowledge editing and achieves, or even exceeds, the performance of the best baseline methods in portability and locality metrics.
title Knowledge Editing for Large Language Model with Knowledge Neuronal Ensemble
topic Computation and Language
68T50
url https://arxiv.org/abs/2412.20637