Proceedings

SetKE: Knowledge Editing for Knowledge Elements Overlap

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 8295-8303.

https://doi.org/10.24963/ijcai.2025/922

Abstract

Large Language Models (LLMs) excel in tasks such as retrieval and question answering but require updates to incorporate new knowledge and reduce inaccuracies and hallucinations.

Traditional updating methods, like fine-tuning and incremental learning, face challenges such as overfitting and high computational costs.

Knowledge Editing (KE) provides a promising alternative but often overlooks the Knowledge Element Overlap (KEO) phenomenon, where multiple triplets share common elements, leading to editing conflicts.

We identify the prevalence of KEO in existing KE datasets and show its significant impact on current KE methods, causing performance degradation in handling such triplets.

To address this, we propose a new formulation, Knowledge Set Editing (KSE), and introduce SetKE, a method that edits sets of triplets simultaneously.

Experimental results demonstrate that SetKE outperforms existing methods in KEO scenarios on mainstream LLMs. Additionally, we introduce EditSet, a dataset containing KEO triplets, providing a comprehensive benchmark.

  1. [Cao et al., 2021] Nicola De Cao, Wilker Aziz, and Ivan Titov. Editing factual knowledge in language models. In Proceedings of the 2021 Conference on Empirical Meth- ods in Natural Language Processing, pages 6491-6506, 2021.
  2. [Cheng et al., 2024] Siyuan Cheng, Ningyu Zhang, Bozhong Tian, Xi Chen, Qingbin Liu, and Huajun Chen. Editing language model-based knowledge graph embeddings. In Proceedings of the AAAI Conference on Artificial Intelli- gence, volume 38, pages 17835-17843, 2024.
  3. [Cohen et al., 2024] Roi Cohen, Eden Biran, Ori Yoran, Amir Globerson, and Mor Geva. Evaluating the ripple effects of knowledge editing in language models. Trans- actions of the Association for Computational Linguistics, 11:283-298, 2024.
  4. [Dai et al., 2022] Damai Dai, Li Dong, Yaru Hao, Zhifang Sui, Baobao Chang, and Furu Wei. Knowledge neurons in pretrained transformers. In Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 8493-8502, 2022.
  5. [Dong et al., 2022] Qingxiu Dong, Damai Dai, Yifan Song, Jingjing Xu, Zhifang Sui, and Lei Li. Calibrating factual knowledge in pretrained language models. In Findings of the Association for Computational Linguistics: EMNLP 2022, pages 5937-5947, 2022.
  6. [Elazar et al., 2021] Yanai Elazar, Nora Kassner, Shauli Ravfogel, Abhilasha Ravichander, Eduard Hovy, Hinrich Schütze, and Yoav Goldberg. Measuring and improving consistency in pretrained language models. Transactions of the Association for Computational Linguistics, 9:1012- 1031, 2021.
  7. [Elsahar et al., 2018] Hady Elsahar, Pavlos Vougiouklis, Ar- slen Remaci, Christophe Gravier, Jonathon Hare, Fred- erique Laforest, and Elena Simperl. T-rex: A large scale alignment of natural language with knowledge base triples. In Proceedings of the Eleventh International Conference on Language Resources and Evaluation (LREC 2018), 2018.
  8. [Geva et al., 2020] Mor Geva, Roei Schuster, Jonathan Be- rant, and Omer Levy. Transformer feed-forward layers are key-value memories. arXiv preprint arXiv:2012.14913, 2020.
  9. [Geva et al., 2021] Mor Geva, Roei Schuster, Jonathan Be- rant, and Omer Levy. Transformer feed-forward layers are key-value memories. In Proceedings of the 2021 Confer- ence on Empirical Methods in Natural Language Process- ing, pages 5484-5495, 2021.
  10. [Geva et al., 2022] Mor Geva, Avi Caciularu, Kevin Wang, and Yoav Goldberg. Transformer feed-forward layers build predictions by promoting concepts in the vocabulary space. In Proceedings of the 2022 Conference on Empir- ical Methods in Natural Language Processing, pages 30- 45, 2022.
  11. [Gupta et al., 2023] Anshita Gupta, Debanjan Mondal, Ak- shay Sheshadri, Wenlong Zhao, Xiang Li, Sarah Wiegr- effe, and Niket Tandon. Editing common sense in trans- formers. In Proceedings of the 2023 Conference on Em- pirical Methods in Natural Language Processing, pages 8214-8232, 2023.
  12. [Hu et al., 2024] Chenhui Hu, Pengfei Cao, Yubo Chen, Kang Liu, and Jun Zhao. Wilke: Wise-layer knowl- edge editor for lifelong knowledge editing. arXiv preprint arXiv:2402.10987, 2024.
  13. [Huang et al., 2022] Zeyu Huang, Yikang Shen, Xiaofeng Zhang, Jie Zhou, Wenge Rong, and Zhang Xiong. Transformer-patcher: One mistake worth one neuron. In The Eleventh International Conference on Learning Rep- resentations, 2022.
  14. [Huang et al., 2024a] Xiusheng Huang, Jiaxiang Liu, Yequan Wang, and Kang Liu. Reasons and solutions for the decline in model performance after editing. In The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024.
  15. [Huang et al., 2024b] Xiusheng Huang, Yequan Wang, Jun Zhao, and Kang Liu. Commonsense knowledge editing based on free-text in llms. In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, pages 14870-14880, 2024.
  16. [Kuhn, 1955] Harold W Kuhn. The hungarian method for the assignment problem. Naval research logistics quarterly, 2(1-2):83-97, 1955.
  17. [Levy et al., 2017] Omer Levy, Minjoon Seo, Eunsol Choi, and Luke Zettlemoyer. Zero-shot relation extraction via reading comprehension. In 21st Conference on Com- putational Natural Language Learning, CoNLL 2017, pages 333-342. Association for Computational Linguis- tics (ACL), 2017.
  18. [Li et al., 2024] Xiaopeng Li, Shasha Li, Shezheng Song, Jing Yang, Jun Ma, and Jie Yu. Pmet: Precise model edit- ing in a transformer. In Proceedings of the AAAI Confer- ence on Artificial Intelligence, volume 38, pages 18564- 18572, 2024.
  19. [Ma et al., 2024] Jun-Yu Ma, Zhen-Hua Ling, Ningyu Zhang, and Jia-Chen Gu. Neighboring perturbations of knowledge editing on large language models. In Forty-first International Conference on Machine Learning, 2024.
  20. [Meng et al., 2022a] Kevin Meng, David Bau, Alex Ando- nian, and Yonatan Belinkov. Locating and editing factual associations in gpt. Advances in Neural Information Pro- cessing Systems, 35:17359-17372, 2022.
  21. [Meng et al., 2022b] Kevin Meng, Arnab Sen Sharma, Alex J Andonian, Yonatan Belinkov, and David Bau. Mass-editing memory in a transformer. In The Eleventh International Conference on Learning Representations, 2022.
  22. [Mitchell et al., 2021] Eric Mitchell, Charles Lin, Antoine Bosselut, Chelsea Finn, and Christopher D Manning. Fast model editing at scale. In International Conference on Learning Representations, 2021.
  23. [Mitchell et al., 2022] Eric Mitchell, Charles Lin, Antoine Bosselut, Christopher D Manning, and Chelsea Finn. Memory-based model editing at scale. In International Conference on Machine Learning, pages 15817-15831.
  24. PMLR, 2022.
  25. [Petroni et al., 2019] Fabio Petroni, Tim Rocktäschel, Se- bastian Riedel, Patrick Lewis, Anton Bakhtin, Yuxiang Wu, and Alexander Miller. Language models as knowl- edge bases? In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Lan- guage Processing (EMNLP-IJCNLP), pages 2463-2473, 2019.
  26. [Qi et al., 2024] Siyuan Qi, Bangcheng Yang, Kailin Jiang, Xiaobo Wang, Jiaqi Li, Yifan Zhong, Yaodong Yang, and Zilong Zheng. In-context editing: Learning knowl- edge from self-induced distributions. arXiv preprint arXiv:2406.11194, 2024.
  27. [Radford et al., 2019] Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. Language models are unsupervised multitask learners. OpenAI blog, 1(8):9, 2019.
  28. [Rawat et al., 2021] Ankit Singh Rawat, Chen Zhu, Daliang Li, Felix Yu, Manzil Zaheer, Sanjiv Kumar, and Srinadh Bhojanapalli. Modifying memories in transformer mod- els. In International Conference on Machine Learning (ICML), volume 2020, 2021.
  29. [Sinitsin et al., 2019] Anton Sinitsin, Vsevolod Plokhot- nyuk, Dmitry Pyrkin, Sergei Popov, and Artem Babenko. Editable neural networks. In International Conference on Learning Representations, 2019.
  30. [Song et al., 2024] Ran Song, Shizhu He, Shuting Jiang, Yantuan Xian, Shengxiang Gao, Kang Liu, and Zheng- tao Yu. Does large language model contain task-specific neurons? In Proceedings of the 2024 Conference on Em- pirical Methods in Natural Language Processing, pages 7101-7113, 2024.
  31. [Sui et al., 2023] Dianbo Sui, Xiangrong Zeng, Yubo Chen, Kang Liu, and Jun Zhao. Joint entity and relation extrac- tion with set prediction networks. IEEE Transactions on Neural Networks and Learning Systems, 2023.
  32. [Sun et al., 2021] Zhiqing Sun, Shengcao Cao, Yiming Yang, and Kris M Kitani. Rethinking transformer-based set prediction for object detection. In Proceedings of the IEEE/CVF international conference on computer vision, pages 3611-3620, 2021.
  33. [Tan et al., 2023] Chenmien Tan, Ge Zhang, and Jie Fu. Massive editing for large language model via meta learn- ing. In The Twelfth International Conference on Learning Representations, 2023.
  34. [Tan et al., 2024] Chenmien Tan, Ge Zhang, and Jie Fu. Massive editing for large language models via meta learn- ing. In International Conference on Learning Representa- tions, 2024.
  35. [Wang et al., 2024a] Peng Wang, Zexi Li, Ningyu Zhang, Zi- wen Xu, Yunzhi Yao, Yong Jiang, Pengjun Xie, Fei Huang, and Huajun Chen. Wise: Rethinking the knowledge mem- ory for lifelong model editing of large language models. arXiv preprint arXiv:2405.14768, 2024.
  36. [Wang et al., 2024b] Xiaohan Wang, Shengyu Mao, Shumin Deng, Yunzhi Yao, Yue Shen, Lei Liang, Jinjie Gu, Huajun Chen, and Ningyu Zhang. Editing conceptual knowledge for large language models. In Findings of the Association for Computational Linguistics: EMNLP 2024, pages 706- 724, Miami, Florida, USA, 2024. Association for Compu- tational Linguistics.
  37. [Wei et al., 2023] Yifan Wei, Xiaoyan Yu, Huanhuan Ma, Fangyu Lei, Yixuan Weng, Ran Song, and Kang Liu. As- sessing knowledge editing in language models via relation perspective. arXiv preprint arXiv:2311.09053, 2023.
  38. [Wei et al., 2024] Yifan Wei, Xiaoyan Yu, Yixuan Weng, Huanhuan Ma, Yuanzhe Zhang, Jun Zhao, and Kang Liu. Does knowledge localization hold true? surprising differ- ences between entity and relation perspectives in language models. In Proceedings of the 33rd ACM International Conference on Information and Knowledge Management, pages 4118-4122, 2024.
  39. [Yao et al., 2024] Yunzhi Yao, Ningyu Zhang, Zekun Xi, Mengru Wang, Ziwen Xu, Shumin Deng, and Huajun Chen. Knowledge circuits in pretrained transformers. In The Thirty-eighth Annual Conference on Neural Informa- tion Processing Systems, 2024.
  40. [Yin et al., 2024] Xunjian Yin, Jin Jiang, Liming Yang, and Xiaojun Wan. History matters: Temporal knowledge edit- ing in large language model. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 19413-19421, 2024.
  41. [Yu et al., 2024] Lang Yu, Qin Chen, Jie Zhou, and Liang He. Melo: Enhancing model editing with neuron-indexed dynamic lora. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 19449-19457, 2024.
  42. [Zheng et al., 2023] Ce Zheng, Lei Li, Qingxiu Dong, Yux- uan Fan, Zhiyong Wu, Jingjing Xu, and Baobao Chang. Can we edit factual knowledge by in-context learning? In Proceedings of the 2023 Conference on Empirical Meth- ods in Natural Language Processing, pages 4862-4876, 2023.
  43. [Zhong et al., 2023] Zexuan Zhong, Zhengxuan Wu, Christopher D Manning, Christopher Potts, and Danqi Chen. Mquake: Assessing knowledge editing in language models via multi-hop questions. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, pages 15686-15702, 2023.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards