Knowledge Editing (KE) aims to modify specific knowledge within large models and address issues of knowledge misinformation or inaccuracies. Most previous studies were mainly based on structured facts, whereas real-world knowledge updates commonly emerge in unstructured texts. In this work, we focus on the CCKS-IJCKG 2024 Competition of Knowledge Editing, which features a Chinese dataset containing both structured and unstructured knowledge. We propose a two-stage approach for KE, using LoRA based method for model editing and incorporating the In-Context Learning Knowledge Editing (IKE) method to refine the editing process. In the first stage, we demonstrate that standard fine-tuning based on LoRA can yield superior performance with some key modifications. We align the inference process with the training process and implement template-based training, specifically without masking special tokens. To enhance portability, we also incorporate reverse masked loss during training. In the second stage, the IKE method leverages in-context learning to improve the model’s understanding and application of the edited knowledge in various contexts. Our method takes the first place in the CCKS-IJCKG 2024 Knowledge Editing Evaluation Task Competition, achieving a score of 52.6388%.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Two-Stage Approach for Knowledge Editing in LLM

  • Hao Xiong,
  • Wenbiao Shao,
  • Tailai Han,
  • Wenliang Chen

摘要

Knowledge Editing (KE) aims to modify specific knowledge within large models and address issues of knowledge misinformation or inaccuracies. Most previous studies were mainly based on structured facts, whereas real-world knowledge updates commonly emerge in unstructured texts. In this work, we focus on the CCKS-IJCKG 2024 Competition of Knowledge Editing, which features a Chinese dataset containing both structured and unstructured knowledge. We propose a two-stage approach for KE, using LoRA based method for model editing and incorporating the In-Context Learning Knowledge Editing (IKE) method to refine the editing process. In the first stage, we demonstrate that standard fine-tuning based on LoRA can yield superior performance with some key modifications. We align the inference process with the training process and implement template-based training, specifically without masking special tokens. To enhance portability, we also incorporate reverse masked loss during training. In the second stage, the IKE method leverages in-context learning to improve the model’s understanding and application of the edited knowledge in various contexts. Our method takes the first place in the CCKS-IJCKG 2024 Knowledge Editing Evaluation Task Competition, achieving a score of 52.6388%.