Orthogonal Subspace Learning for Language Model Continual Learning
About
Benefiting from massive corpora and advanced hardware, large language models (LLMs) exhibit remarkable capabilities in language understanding and generation. However, their performance degrades in scenarios where multiple tasks are encountered sequentially, also known as catastrophic forgetting. In this paper, we propose orthogonal low-rank adaptation (O-LoRA), a simple and efficient approach for continual learning in language models, effectively mitigating catastrophic forgetting while learning new tasks. Specifically, O-LoRA learns tasks in different (low-rank) vector subspaces that are kept orthogonal to each other in order to minimize interference. Our method induces only marginal additional parameter costs and requires no user data storage for replay. Experimental results on continual learning benchmarks show that our method outperforms state-of-the-art methods. Furthermore, compared to previous approaches, our method excels in preserving the generalization ability of LLMs on unseen tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Continual Learning | Large Number of Tasks | Average Performance73.5 | 50 | |
| Continual Learning | Standard CL Benchmark | Avg Final Acc0.772 | 50 | |
| Continual Learning | Standard CL Benchmark | BWT (Avg Order 1-3)75.8 | 38 | |
| Continual Learning | TRACE | Avg Performance52.02 | 37 | |
| Continual Learning | Long CL benchmark N=15 | Long1 Performance71.5 | 18 | |
| Visual Question Answering | UCIT | ArxivQA80.93 | 16 | |
| Visual Question Answering | MLLM-DCL | Accuracy (Medical)46.89 | 16 | |
| Continual Learning | Long Sequence (test) | AP69.24 | 15 | |
| Continual Learning | SuperNI Benchmark | Average Score25.9 | 14 | |
| Continual Learning | Long Sequence Benchmark | OP69.6 | 14 |