提出新方法保护网络路径,避免遗忘旧知识且不增加参数量
Learning without Isolation: Pathway Protection for Continual Learning
- 以路径而非参数为核心,通过图匹配实现模型融合
- 在多个数据集上显著缓解灾难性遗忘,性能优于现有方法
- 适合资源受限场景,尤其适用于持续学习中的高效模型更新
深度网络在顺序任务学习中易发生灾难性遗忘,即学习新任务时遗忘旧任务知识。现有持续学习方法多聚焦于保护与旧任务相关的参数,但参数保护往往不切实际,因存储旧任务知识的参数规模随任务数量线性增长,难以有效保存。本文受神经科学与物理学启发,提出双重视角:在整体网络中,路径比参数更关键。据此,我们提出新型持续学习框架LwI,将模型融合建模为图匹配问题,保护旧任务占用的路径而不将其隔离。得益于深度网络中激活通道的稀疏性,LwI可自适应分配可用路径给新任务,实现路径保护并高效应对灾难性遗忘。在多个主流基准数据集上的实验表明,所提方法具有显著优势。
原文摘要 · Abstract (English)
Deep networks are prone to catastrophic forgetting during sequential task learning, i.e., losing the knowledge about old tasks upon learning new tasks. To this end, continual learning(CL) has emerged, whose existing methods focus mostly on regulating or protecting the parameters associated with the previous tasks. However, parameter protection is often impractical, since the size of parameters for storing the old-task knowledge increases linearly with the number of tasks, otherwise it is hard to preserve the parameters related to the old-task knowledge. In this work, we bring a dual opinion from neuroscience and physics to CL: in the whole networks, the pathways matter more than the parameters when concerning the knowledge acquired from the old tasks. Following this opinion, we propose a novel CL framework, learning without isolation(LwI), where model fusion is formulated as graph matching and the pathways occupied by the old tasks are protected without being isolated. Thanks to the sparsity of activation channels in a deep network, LwI can adaptively allocate available pathways for a new task, realizing pathway protection and addressing catastrophic forgetting in a parameter-efficient manner. Experiments on popular benchmark datasets demonstrate the superiority of the proposed LwI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。