arXiv:2503.21227cs.CL2025-03被引 3

让大模型持续学习新任务,不丢旧知识且不疯狂涨参数。

LLaVA-CMoE: Towards Continual Mixture of Experts for Large Vision-Language Models

  • 用探针专家动态判断何时何地加新专家,只在需要时扩展最少参数。
  • 在CoIN基准上8个视觉问答任务连续学习,遗忘率显著降低。
  • 无需任务标签也能自动分配专家,适合长期演进的智能系统。

Mixture of Experts(MoE)架构近期推动了大型语言模型在持续多模态学习中的可扩展性与适应性。然而,高效扩展模型以应对序列化任务仍具挑战:新增任务导致参数迅速膨胀,而修改共享路由机制常引发灾难性遗忘,损害已有知识。为此,我们提出LLaVA-CMoE,一种无需回放历史数据的持续学习框架,兼顾参数效率与知识鲁棒性。该方法引入探针引导的知识扩展机制,通过探针专家动态判断新专家的添加时机与位置,实现针对任务复杂度的自适应、最小化参数扩展。同时,提出概率任务定位器,为每项任务分配轻量级专用路由。针对推理阶段任务标签未知的实际问题,采用基于VAE的重建策略,通过匹配输入分布自动识别最优路由,实现精准专家分配。该设计缓解路由冲突与遗忘问题,支持无显式任务标签的鲁棒持续学习。在涵盖8个多样化视觉问答任务的CoIN基准上的大量实验表明,相较于先前方法,LLaVA-CMoE在紧凑模型规模下显著降低遗忘率与参数开销,验证了其在参数高效持续学习中的有效性与可扩展性。代码即将开源。

原文摘要 · Abstract (English)

Mixture of Experts (MoE) architectures have recently advanced the scalability and adaptability of large language models (LLMs) for continual multimodal learning. However, efficiently extending these models to accommodate sequential tasks remains challenging. As new tasks arrive, naive model expansion leads to rapid parameter growth, while modifying shared routing components often causes catastrophic forgetting, undermining previously learned knowledge. To address these issues, we propose LLaVA-CMoE, a continual learning framework for LLMs that requires no replay data of previous tasks and ensures both parameter efficiency and robust knowledge retention. Our approach introduces a Probe-Guided Knowledge Extension mechanism, which uses probe experts to dynamically determine when and where new experts should be added, enabling adaptive and minimal parameter expansion tailored to task complexity. Furthermore, we present a Probabilistic Task Locator that assigns each task a dedicated, lightweight router. To handle the practical issue that task labels are unknown during inference, we leverage a VAE-based reconstruction strategy to identify the most suitable router by matching input distributions, allowing automatic and accurate expert allocation. This design mitigates routing conflicts and catastrophic forgetting, enabling robust continual learning without explicit task labels. Extensive experiments on the CoIN benchmark, covering eight diverse VQA tasks, demonstrate that LLaVA-CMoE delivers strong continual learning performance with a compact model size, significantly reducing forgetting and parameter overhead compared to prior methods. These results showcase the effectiveness and scalability of our approach for parameter-efficient continual learning in large language models. Our code will be open-sourced soon.

持续学习MoE多模态参数效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。