An ultimate objective in continual learning is to preserve knowledge learned in preceding tasks while learning new tasks. To mitigate forgetting prior knowledge, we propose a novel knowledge distillation technique that takes into the account the manifold structure of the latent/output space of a neural network in learning novel tasks. To achieve this, we propose to approximate the data manifold up-to its first order, hence benefiting from linear subspaces to model the structure and maintain the knowledge of a neural network while learning novel concepts. We demonstrate that the modeling with subspaces provides several intriguing properties, including robustness to noise and therefore effective for mitigating Catastrophic Forgetting in continual learning. We also discuss and show how our proposed method can be adopted to address both classification and segmentation problems. Empirically, we observe that our proposed method outperforms various continual learning methods on several challenging datasets including Pascal VOC, and Tiny-Imagenet. Furthermore, we show how the proposed method can be seamlessly combined with existing learning approaches to improve their performances. The codes of this article will be available at https://github.com/csiro-robotics/SDCL.
翻译:持续学习的一个最终目标是:在学习新任务的同时,保留先前任务中习得的知识。为缓解遗忘先前知识的问题,我们提出了一种新颖的知识蒸馏技术,该技术在学习新任务时考虑了神经网络潜在/输出空间的流形结构。为实现此目标,我们提出对数据流形进行一阶近似近似,从而利用线性子空间来建模结构并维持神经网络的知识,同时学习新概念。我们证明了基于子空间的建模具有若干引人注目的特性,包括对噪声的鲁棒性,因此能有效缓解持续学习中的灾难性遗忘。我们还讨论并展示了所提方法如何应用于解决分类和分割问题。实验结果表明,所提方法在Pascal VOC和Tiny-Imagenet等多个具有挑战性的数据集上优于多种持续学习方法。此外,我们还展示了所提方法如何与现有学习方法无缝结合以提升其性能。本文代码将开源在https://github.com/csiro-robotics/SDCL。