提出新方法实现持续学习中遗忘为零且正向知识迁移。
Continual Learning of Achieving Forgetting-free and Positive Knowledge Transfer
- 将持续学习建模为优化问题,约束正向与反向知识迁移
- 通过稀疏子网络和梯度对齐避免遗忘,提升新旧任务性能
- 适合需长期学习且依赖知识积累的AI系统应用
现有持续学习研究主要关注如何缓解灾难性遗忘,以平衡新任务的学习灵活性与旧任务的记忆稳定性。然而,理想的持续学习智能体不仅应克服遗忘,还应促进正向和反向知识迁移(FKT、BKT)——即利用过往知识帮助新任务学习,同时借助新知识提升旧任务表现。本文首次将持续学习建模为优化问题,要求每个任务在确保正向与反向知识迁移为正的条件下达到最优性能。为此,提出新型增强型任务持续学习(ETCL)方法,实现无遗忘且正向知识迁移。理论分析进一步推导出导致负向知识迁移的边界,并据此设计在线任务相似性检测策略以促进正向迁移。为克服遗忘,ETCL为每项任务学习一组特定二值掩码,保留稀疏子网络的同时维持密集网络性能。在新任务开始时,通过将新任务梯度与前一相似任务子网络梯度对齐,保证正向知识迁移。结合双目标优化与正交梯度投影方法,仅更新分类层中先前相似任务的权重,实现正向反向知识迁移。大量实验表明,所提方法在差异大、相似、混合任务序列上均显著优于强基线。
原文摘要 · Abstract (English)
Existing research on continual learning (CL) of a sequence of tasks focuses mainly on dealing with catastrophic forgetting (CF) to balance the learning plasticity of new tasks and the memory stability of old tasks. However, an ideal CL agent should not only be able to overcome CF, but also encourage positive forward and backward knowledge transfer (KT), i.e., using the learned knowledge from previous tasks for the new task learning (namely FKT), and improving the previous tasks' performance with the knowledge of the new task (namely BKT). To this end, this paper first models CL as an optimization problem in which each sequential learning task aims to achieve its optimal performance under the constraint that both FKT and BKT should be positive. It then proposes a novel Enhanced Task Continual Learning (ETCL) method, which achieves forgetting-free and positive KT. Furthermore, the bounds that can lead to negative FKT and BKT are estimated theoretically. Based on the bounds, a new strategy for online task similarity detection is also proposed to facilitate positive KT. To overcome CF, ETCL learns a set of task-specific binary masks to isolate a sparse sub-network for each task while preserving the performance of a dense network for the task. At the beginning of a new task learning, ETCL tries to align the new task's gradient with that of the sub-network of the previous most similar task to ensure positive FKT. By using a new bi-objective optimization strategy and an orthogonal gradient projection method, ETCL updates only the weights of previous similar tasks at the classification layer to achieve positive BKT. Extensive evaluations demonstrate that the proposed ETCL markedly outperforms strong baselines on dissimilar, similar, and mixed task sequences.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。