MELO: Enhancing Model Editing with Neuron-Indexed Dynamic LoRA

Authors

  • Lang Yu School of Computer Science and Technology, East China Normal University Shanghai Institute of AI for Education, East China Normal University
  • Qin Chen School of Computer Science and Technology, East China Normal University Shanghai Institute of AI for Education, East China Normal University
  • Jie Zhou School of Computer Science and Technology, East China Normal University Shanghai Institute of AI for Education, East China Normal University
  • Liang He School of Computer Science and Technology, East China Normal University Shanghai Institute of AI for Education, East China Normal University

DOI:

https://doi.org/10.1609/aaai.v38i17.29916

Keywords:

NLP: Learning & Optimization for NLP, NLP: (Large) Language Models, NLP: Safety and Robustness

Abstract

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.

Published

2024-03-24

How to Cite

Yu, L., Chen, Q., Zhou, J., & He, L. (2024). MELO: Enhancing Model Editing with Neuron-Indexed Dynamic LoRA. Proceedings of the AAAI Conference on Artificial Intelligence, 38(17), 19449-19457. https://doi.org/10.1609/aaai.v38i17.29916

Issue

Section

AAAI Technical Track on Natural Language Processing II