Scaling Continual Learning to 300+ Tasks with Bi-Level Routing Mixture-of-Experts
Abstract
Lay Summary
When AI models learn new tasks, they often forget what they learned before, a problem called "catastrophic forgetting". To address this, Continual Learning (CL) has become a promising solution. However, existing methods work well for a small number of tasks (e.g., 20 tasks), but it is unclear whether they remain effective when faced with hundreds of consecutive tasks. Additionally, there was no suitable benchmark to evaluate a model's ability to learn over such long task sequences. We propose CaRE, a strong CL method that can dynamically retrieve and combine relevant knowledge from past tasks when learning new ones, mimicking how humans learn. We also created OmniBenchmark-1K, a challenging dataset with 1,000 diverse classes designed specifically for evaluating very long-sequence CL. CaRE is the first CL system to effectively handle over 300 non-overlapping tasks, outperforming all existing methods by a large margin. Our code and dataset are publicly available, enabling further research into more practical AI systems that can learn continuously in real-world environments.