MELO: Enhancing Model Editing with Neuron-Indexed Dynamic LoRA

التفاصيل البيبلوغرافية
العنوان: MELO: Enhancing Model Editing with Neuron-Indexed Dynamic LoRA
المؤلفون: Yu, Lang, Chen, Qin, Zhou, Jie, He, Liang
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computation and Language
الوصف: Large language models (LLMs) have shown great success in various Natural Language Processing (NLP) tasks, whist they still need updates after deployment to fix errors or keep pace with the changing knowledge in the world. Researchers formulate such problem as Model Editing and have developed various editors focusing on different axes of editing properties. However, current editors can hardly support all properties and rely on heavy computational resources. In this paper, we propose a plug-in Model Editing method based on neuron-indexed dynamic LoRA (MELO), which alters the behavior of language models by dynamically activating certain LoRA blocks according to the index built in an inner vector database. Our method satisfies various editing properties with high efficiency and can be easily integrated into multiple LLM backbones. Experimental results show that our proposed MELO achieves state-of-the-art editing performance on three sequential editing tasks (document classification, question answering and hallucination correction), while requires the least trainable parameters and computational cost.
Comment: In Proceedings of The 38th Annual AAAI Conference on Artificial Intelligence
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2312.11795
رقم الانضمام: edsarx.2312.11795
قاعدة البيانات: arXiv