通过分层残差梯度适配,实现视觉语言模型持续学习的三重平衡。
KeepLoRA++: Continual Learning with Layer-Scaled Residual Gradient Adaptation

- 将参数空间分解为主成分与残差子空间,分层控制更新。
- 在图像分类、视觉问答等任务上优于现有基线方法。
- 适合需要长期学习新任务且不遗忘旧知识的研究场景。
预训练视觉语言模型的持续学习需平衡三重目标:保留预训练知识、维持序列任务的知识、保持获取新知识的能力。本文提出KeepLoRA++,通过统一的双维度知识保留机制实现该平衡。我们从层间与层内两个视角分析Transformer结构中的知识分布:层间分析揭示浅层主要编码通用可迁移知识,深层则聚焦任务特定适应;层内分析表明,主成分子空间承载通用知识,残差子空间用于任务特化。基于此发现,KeepLoRA++引入分层残差梯度适配策略:新任务仅在残差子空间更新LoRA参数,并采用由浅至深的层尺度缩放,以避免干扰已有能力。具体而言,新任务梯度被投影到同时正交于预训练模型主成分子空间及前序任务主导方向的子空间,同时对浅层赋予较小更新幅度,深层赋予较大更新幅度。理论分析与实证评估均表明,KeepLoRA++能有效平衡三重目标,在图像分类、视觉问答和视频理解任务中持续优于代表性基线。
原文摘要 · Abstract (English)
Continual learning for pre-trained vision-language models requires balancing three competing objectives: retaining pre-trained knowledge, preserving knowledge from a sequence of learned tasks, and maintaining the plasticity to acquire new knowledge. This paper presents KeepLoRA++, balancing these objectives through a unified dual-dimensional knowledge retention mechanism. We analyze knowledge distribution of Transformer architecture from both inter-layer and intra-layer perspectives. The inter-layer perspective examines how retention is distributed across layers, while the intra-layer perspective focuses on the parameter space within each layer. Our analysis reveals a structural property: general transferable knowledge is mainly encoded in the shallow layers and the principal subspace of the parameters, while task-specific adaptations are localized in the deep layers and the residual subspace. Motivated by this insight, KeepLoRA++ introduces a layer-scaled residual gradient adaptation method. New tasks are learned by restricting LoRA parameter updates to the residual subspace, combined with a shallow-to-deep layer scaling, to prevent interference with previously acquired capabilities. Specifically, the gradient of a new task is projected onto a subspace orthogonal to both the principal subspace of the pre-trained model and the dominant directions of previous task features, while simultaneously assigning smaller update magnitudes to shallow layers and larger ones to deeper layers. Our theoretical analysis and empirical evaluations confirm that KeepLoRA++ successfully balances these three competing objectives, consistently outperforming representative baselines across image classification, visual question answering, and video understanding tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。