LoRAMoE: Alleviate World Knowledge Forgetting in Large Language Models via MoE-Style Plugin

Shihan Dou,Enyu Zhou,Yan Liu,Songyang Gao,Jun Zhao,Wei Shen,Yuhao Zhou,Zhiheng Xi,Xiao Wang,Xiaoran Fan,Shiliang Pu,Jiang Zhu,Rui Zheng,Tao Gui,Qi Zhang,Xuanjing Huang
2024-03-08
Abstract:Supervised fine-tuning (SFT) is a crucial step for large language models (LLMs), enabling them to align with human instructions and enhance their capabilities in downstream tasks. Increasing instruction data substantially is a direct solution to align the model with a broader range of downstream tasks or notably improve its performance on a specific task. However, we find that large-scale increases in instruction data can damage the world knowledge previously stored in LLMs. To address this challenge, we propose LoRAMoE, a novelty framework that introduces several low-rank adapters (LoRA) and integrates them by using a router network, like a plugin version of Mixture of Experts (MoE). It freezes the backbone model and forces a portion of LoRAs to focus on leveraging world knowledge to solve downstream tasks, to alleviate world knowledge-edge forgetting. Experimental results show that, as the instruction data increases, LoRAMoE can significantly improve the ability to process downstream tasks, while maintaining the world knowledge stored in the LLM.
Computation and Language
What problem does this paper attempt to address?
This paper attempts to address the issue of world knowledge forgetting caused by increasing the amount of instruction data during the supervised fine-tuning (SFT) process of large-scale language models (LLMs). Specifically, the paper finds that with a significant increase in instruction data, although the performance on downstream tasks can be greatly improved, the world knowledge stored within the model is compromised. To resolve this contradiction, the authors propose the LoRAMoE framework, which introduces low-rank adapters (LoRA) as experts and integrates them using a router network. This approach aims to enhance the model's ability to handle downstream tasks while mitigating the forgetting of world knowledge. Experimental results show that LoRAMoE can effectively improve the model's performance on various downstream tasks without sacrificing the internal world knowledge of the model.