arXiv:2411.15469cs.CV2024-11被引 5

用正交投影避免遗忘,让Mamba模型持续学习新任务

Mamba-CL: Optimizing Selective State Space Model in Null Space for Continual Learning

  • 在旧任务特征空间的正交方向更新参数,防止干扰已有知识
  • 在4个增量分类基准上超越现有方法,显著减少遗忘
  • 适合需要长期学习的新任务场景,尤其是视觉领域

持续学习(CL)旨在使AI模型能够随时间顺序学习一系列任务而不遗忘旧知识。近期,状态空间模型(SSMs),特别是Mamba模型,在计算机视觉中取得显著进展。本文基于SSM的优势,探索将其应用于持续学习。为此,提出Mamba-CL框架,通过在大模型核心SSM中更新与先前任务特征子空间正交的参数,实现对大规模Mamba基础模型的连续微调。该方法理论上保证了每个SSM模块在新旧任务间输出的一致性,从而缓解灾难性遗忘问题。具体而言,推导出Mamba模型中四个关键时不变参数的整体一致性约束,简化其循环状态空间结构和非线性离散化过程。实践中,采用零空间投影高效实现参数正交性。在四个类增量基准上的大量实验表明,Mamba-CL在抗遗忘方面表现优异,性能优于当前最优方法。代码见附录。

原文摘要 · Abstract (English)

Continual Learning (CL) aims to equip AI models with the ability to learn a sequence of tasks over time, without forgetting previously learned knowledge. Recently, State Space Models (SSMs), particularly the Mamba model, have achieved notable success in computer vision. Building on the strengths of SSMs, this study explores leveraging the Mamba model for CL. Therefore, we introduce Mamba-CL, a framework that continuously fine-tunes the core SSMs of the large-scale Mamba foundation model by updating parameters orthogonal to the feature subspace of previous tasks. This approach theoretically guarantees the consistency objective aiming to preserves consistent output for each SSM module across both previous and current tasks, so as to overcome catastrophic forgetting issue. Specifically, we achieve this goal by deducing the overall consistency constraints on four key time-invariant parameters in the Mamba model, streamlining its recurrent state-space structure and non-linear discretization process in SSM. In practice, we apply the null-space projection to efficiently implement the orthogonality within Mamba model. Extensive experiments on four class-incremental benchmarks demonstrate the effectiveness of Mamba-CL for anti-forgetting, achieving superior performances to state-of-the-art methods. Code is available in the supplementary materials.

持续学习Mamba模型抗遗忘

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