arXiv:2507.09471cs.CV2025-07中稿 · IEEE Transactions …被引 5

解决持续学习中任务标识误导导致的误判问题

CKAA: Cross-subspace Knowledge Alignment and Aggregation for Robust Continual Learning

  • 通过跨子空间特征对齐提升模型鲁棒性
  • 在多个基准上显著优于现有参数高效微调方法
  • 适合需要高可靠性的连续学习应用场景

持续学习使AI模型能从连续任务流中不断学习。近期基于参数高效微调(PEFT)的方法因性能优越受到关注,通常为每个任务分配独立子模块,并用任务识别器选择测试图像对应的子模块。然而,由于各子模块独立训练导致特征子空间错位,当任务标识错误时模型易产生模糊决策。为此,本文提出跨子空间知识对齐与聚合框架(CKAA),包含两项创新:(1) 双层次知识对齐(DKA):通过跨子空间对齐类内特征分布,并借助特征模拟过程学习鲁棒全局分类器,使模型在训练中能区分正确与错误子空间的特征;(2) 基于任务置信度的适配器混合(TC-MoA):一种鲁棒推理机制,根据任务置信度分数自适应聚合相关子模块的知识,避免对错误任务标识过度自信。大量实验表明,CKAA显著优于现有PEFT类持续学习方法。代码已开源。

原文摘要 · Abstract (English)

Continual Learning (CL) empowers AI models to continuously learn from sequential task streams. Recently, parameter-efficient fine-tuning (PEFT)-based CL methods have garnered increasing attention due to their superior performance. They typically allocate a unique sub-module for learning each task, with a task recognizer to select the appropriate sub-modules for testing images. However, due to the feature subspace misalignment from independently trained sub-modules, these methods tend to produce ambiguous decisions under misleading task-ids. To address this, we propose Cross-subspace Knowledge Alignment and Aggregation (CKAA), a novel framework that enhances model robustness against misleading task-ids through two key innovations: (1) Dual-level Knowledge Alignment (DKA): By aligning intra-class feature distributions across different subspaces and learning a robust global classifier through a feature simulation process, DKA enables the model to distinguish features from both correct and incorrect subspaces during training. (2) Task-Confidence-guided Mixture of Adapters (TC-MoA): A robust inference scheme that adaptively aggregates task-specific knowledge from relevant sub-modules based on task-confidence scores, avoiding overconfidence in misleading task-id predictions. Extensive experiments demonstrate that CKAA outperforms existing PEFT-based CL methods. Code is available at https://github.com/HHHLF/CKAA_TIP2026.

持续学习参数高效鲁棒性

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