arXiv:2502.09500cs.LG2025-02

提出Eidetic学习方法,可证明性地解决神经网络灾难性遗忘问题。

Eidetic Learning: an Efficient and Provable Solution to Catastrophic Forgetting

  • 通过数据条件路由机制,动态分配网络容量应对新任务。
  • 训练后无需重放或记忆,对多个任务均保持零遗忘表现。
  • 适合需要持续学习的场景,兼顾理论保证与实际效率。

灾难性遗忘——神经网络在学习任务t1后,再训练任务t2时丧失t1能力——是长期存在的难题。本文提出Eidetic Learning方法,可证明性地解决该问题。采用该方法训练的EideticNet无需重放或记忆即可保持所有任务性能。针对连续离散任务,模型在推理时能自动根据输入进行数据条件路由,无需额外任务信息。其结构类似稀疏门控Mixture-of-Experts,将网络容量按任务划分并实现自适应路由。EideticNet实现简单、训练高效,时间与空间复杂度随参数量线性增长。该方法在预训练和微调阶段对现代神经网络中的归一化层均具理论保证。多种架构与任务组合下,EideticNet均表现出无遗忘特性。代码已开源:https://github.com/amazon-science/eideticnet-training。

原文摘要 · Abstract (English)

Catastrophic forgetting -- the phenomenon of a neural network learning a task t1 and losing the ability to perform it after being trained on some other task t2 -- is a long-standing problem for neural networks [McCloskey and Cohen, 1989]. We present a method, Eidetic Learning, that provably solves catastrophic forgetting. A network trained with Eidetic Learning -- here, an EideticNet -- requires no rehearsal or replay. We consider successive discrete tasks and show how at inference time an EideticNet automatically routes new instances without auxiliary task information. An EideticNet bears a family resemblance to the sparsely-gated Mixture-of-Experts layer Shazeer et al. [2016] in that network capacity is partitioned across tasks and the network itself performs data-conditional routing. An EideticNet is easy to implement and train, is efficient, and has time and space complexity linear in the number of parameters. The guarantee of our method holds for normalization layers of modern neural networks during both pre-training and fine-tuning. We show with a variety of network architectures and sets of tasks that EideticNets are immune to forgetting. While the practical benefits of EideticNets are substantial, we believe they can be benefit practitioners and theorists alike. The code for training EideticNets is available at https://github.com/amazon-science/eideticnet-training.

持续学习灾难性遗忘神经网络可证明性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。