arXiv:2412.18177cs.LGcs.AI2024-12CVPR被引 9

通过可插拔模块提升在线持续学习的适应能力

Enhancing Online Continual Learning with Plug-and-Play State Space Model and Class-Conditional Mixture of Discretization

  • 引入S6MOD模块,基于选择性状态空间模型动态调整参数
  • 在多个基准上实现性能突破,显著提升模型适应性
  • 适合需要增量学习且注重泛化能力的研究与应用

在线持续学习(OCL)旨在从仅出现一次的数据流中学习新任务,同时保留已有知识。现有方法多依赖回放机制,侧重通过正则化或知识蒸馏增强记忆保持,但常忽视模型的适应性,难以从在线数据中逐步学习通用且判别性强的特征。为此,本文提出可插拔模块S6MOD,可集成于多数现有方法中以直接提升适应性。S6MOD在主干网络后引入额外分支,通过混合离散化选择性调整状态空间模型参数,丰富选择性扫描模式,使模型能自适应选择当前动态最敏感的离散化方式。进一步设计了基于类条件的路由算法实现动态、不确定性驱动的调整,并引入对比离散化损失进行优化。大量实验表明,将S6MOD与多种模型结合后,显著提升了模型适应性,带来明显性能提升,达到当前最优水平。

原文摘要 · Abstract (English)

Online continual learning (OCL) seeks to learn new tasks from data streams that appear only once, while retaining knowledge of previously learned tasks. Most existing methods rely on replay, focusing on enhancing memory retention through regularization or distillation. However, they often overlook the adaptability of the model, limiting the ability to learn generalizable and discriminative features incrementally from online training data. To address this, we introduce a plug-and-play module, S6MOD, which can be integrated into most existing methods and directly improve adaptability. Specifically, S6MOD introduces an extra branch after the backbone, where a mixture of discretization selectively adjusts parameters in a selective state space model, enriching selective scan patterns such that the model can adaptively select the most sensitive discretization method for current dynamics. We further design a class-conditional routing algorithm for dynamic, uncertainty-based adjustment and implement a contrastive discretization loss to optimize it. Extensive experiments combining our module with various models demonstrate that S6MOD significantly enhances model adaptability, leading to substantial performance gains and achieving the state-of-the-art results.

持续学习状态空间在线学习模型适应

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