arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.04741cs.LGcs.CVq-bio.NC
收藏

Keep Moving:识别任务相关子空间以最大化新学习任务的可塑性

Keep Moving: identifying task-relevant subspaces to maximise plasticity for newly learned tasks

Osnabrück University(奥斯纳布吕克大学);Institute of Cognitive Science(认知科学研究所)

Daniel Anthes, Sushrut Thorat, Peter König, Tim C. Kietzmann

AI总结:

该研究将神经网络激活空间分为影响旧任务的读出范围子空间与不影响的零空间,证明基于正则化的持续学习算法过度限制了可塑性,还提出深度网络功能相关子空间的估计方法,为开发兼顾稳定与高可塑性的持续学习算法提供指导。

摘要:

持续学习算法致力于在保留已有信息的同时获取新知识。这类算法通常强调稳定性,并在学习新任务时限制网络更新。在许多情况下,这类限制会损害模型的可塑性,即模型适应新任务需求的能力。但所有的网络变化都是有害的吗?本文通过提出神经网络的激活空间可分解为两个子空间来探讨这个问题:一个是读出范围(readout range),该范围内的变化会影响先前任务;另一个是零空间(null space),该范围内的变化不会改变先前任务的性能。基于这项新技术的实验,我们表明确实并非所有激活变化都与遗忘相关。相反,只有任务读出层可见的子空间内的变化才会导致稳定性下降,而限制该子空间外的变化只会造成可塑性损失。通过分析多种常用算法,我们发现基于正则化的技术未能完全分离这两个空间,因此对可塑性的限制超出了必要程度。我们通过研究一个线性模型拓展了结果:在该模型中我们可以直接操控两个子空间内的学习,从而将激活变化与稳定性和可塑性建立因果关联。对于分层非线性的情况,我们提出了一种近似方法,能够估计深度非线性网络每一层的功能相关子空间,印证了我们之前的结论。综上,这项工作为理解持续学习中稳定性与可塑性的背后机制提供了新方法,还可作为诊断工具,指导未来持续学习算法的开发,实现推理稳定的同时预留最大的学习空间。

英文摘要:

Continual learning algorithms strive to acquire new knowledge while preserving prior information. Often, these algorithms emphasise stability and restrict network updates upon learning new tasks. In many cases, such restrictions come at a cost to the model's plasticity, i.e. the model's ability to adapt to the requirements of a new task. But is all change detrimental? Here, we approach this question by proposing that activation spaces in neural networks can be decomposed into two subspaces: a readout range in which change affects prior tasks and a null space in which change does not alter prior performance. Based on experiments with this novel technique, we show that, indeed, not all activation change is associated with forgetting. Instead, only change in the subspace visible to the readout of a task can lead to decreased stability, while restricting change outside of this subspace is associated only with a loss of plasticity. Analysing various commonly used algorithms, we show that regularisation-based techniques do not fully disentangle the two spaces and, as a result, restrict plasticity more than need be. We expand our results by investigating a linear model in which we can manipulate learning in the two subspaces directly and thus causally link activation changes to stability and plasticity. For hierarchical, nonlinear cases, we present an approximation that enables us to estimate functionally relevant subspaces at every layer of a deep nonlinear network, corroborating our previous insights. Together, this work provides novel means to derive insights into the mechanisms behind stability and plasticity in continual learning and may serve as a diagnostic tool to guide developments of future continual learning algorithms that stabilise inference while allowing maximal space for learning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑