arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2602.03473cs.LGcs.CV

将Bi-Level路由混合专家扩展到300+任务的持续学习

Scaling Continual Learning to 300+ Tasks with Bi-Level Routing Mixture-of-Experts

School of Computing and Data Science, The University of Hong Kong(计算与数据科学学院,香港大学);Center for Embodied Artificial Intelligence and Computer Vision, Shenzhen Loop Area Institute(具身人工智能与计算机视觉中心,深圳环宇区研究所)

Meng Lou, Yunxiang Fu, Yizhou Yu

AI总结:

本文提出CaRE,通过双级路由混合专家机制实现长任务序列的持续学习,提出OmniBenchmark-1K数据集,展示在多种数据集和任务设置上优于基线的性能。

摘要:

持续学习,尤其是类增量学习(CIL),基于预训练模型(PTM)近年来引起了广泛关注。然而,如何在非常长的任务序列中有效学习判别性和全面的特征表示,同时保持稳定性和可塑性仍是一个开放问题。我们提出了CaRE,一种可扩展的持续学习器,采用高效的双级路由混合专家(BR-MoE)。BR-MoE的核心思想是双级路由机制:一个路由选择阶段动态激活相关任务特定的路由器,随后是专家路由阶段动态激活和聚合专家,旨在将判别性和全面的表示注入到每个中间网络层中。另一方面,我们引入了一个具有挑战性的数据集,OmniBenchmark-1K,用于评估在数百个任务的非常长的任务序列上的CIL性能。大量实验表明,CaRE在各种数据集和任务设置上表现出领先性能,包括常用的CIL数据集和经典CIL设置(例如5-20个任务)。据我们所知,CaRE是第一个能够扩展到非常长的任务序列(从100到超过300个不重叠任务)的持续学习器,且在这样的任务序列上大幅优于所有基线。我们希望这项工作能激发进一步研究在极长任务序列上的持续学习。代码和数据集已公开发布在https://github.com/LMMMEng/CaRE。

英文摘要:

Continual learning, especially class-incremental learning (CIL), on the basis of a pre-trained model (PTM) has garnered substantial research interest in recent years. However, how to effectively learn both discriminative and comprehensive feature representations while maintaining stability and plasticity over very long task sequences remains an open problem. We propose CaRE, a scalable {C}ontinual Le{a}rner with efficient Bi-Level {R}outing Mixture-of-{E}xperts (BR-MoE). The core idea of BR-MoE is a bi-level routing mechanism: a router selection stage that dynamically activates relevant task-specific routers, followed by an expert routing phase that dynamically activates and aggregates experts, aiming to inject discriminative and comprehensive representations into every intermediate network layer. On the other hand, we introduce a challenging dataset, OmniBenchmark-1K, for CIL performance evaluation on very long task sequences with hundreds of tasks. Extensive experiments show that CaRE demonstrates leading performance across a variety of datasets and task settings, including commonly used CIL datasets with classical CIL settings (e.g., 5-20 tasks). To the best of our knowledge, CaRE is the first continual learner that scales to very long task sequences (ranging from 100 to over 300 non-overlapping tasks), while outperforming all baselines by a large margin on such task sequences. We hope that this work will inspire further research into continual learning over extremely long task sequences. Code and dataset are publicly released at https://github.com/LMMMEng/CaRE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑