LoRAMoE: Alleviate World Knowledge Forgetting in Large Language Models via MoE-Style Plugin
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Reasoning | BBH | Accuracy41.74 | 672 | |
| Embodied Navigation | LENL (test) | SR-F (S1)78 | 44 | |
| Continual Unlearning | CIFAR-100 ResNet-20 (test) | Retention Accuracy60.46 | 24 | |
| Lifelong Embodied Navigation | LENL (test) | S1 Success Rate97 | 22 | |
| Robotic Manipulation | LLCRM 1.0 (test) | S1 Score73 | 22 | |
| pick place | AgileX PiPER real-world | Success Rate80 | 14 | |
| Commonsense Reasoning | Commonsense Reasoning Tasks (ARC-e, OBQA, SIQA, ARC-c, WinoG, PIQA, BoolQ, HellaS) LLaMA3-8B | ARC-e Accuracy87.8 | 13 | |
| Embodied Navigation | LENL 1.0 (test) | Success Rate (S1)15 | 13 | |
| Vision-Language Navigation | AML-VLN (test) | Task 1 Success Rate29 | 13 | |
| Press | AgileX PiPER real-world | Success Rate100 | 12 |