用联合扩散模型实现持续学习,防止遗忘。
Joint Diffusion models in Continual Learning
- 将分类器与扩散生成模型联合优化,共享参数结构。
- 在多个基准上优于现有生成重放方法,无灾难性遗忘。
- 适用于半监督场景,适合需要稳定学习的AI系统。
本文提出JDCL——一种基于联合扩散模型的持续学习方法。神经网络在接收不同分布的新数据时会出现灾难性遗忘,即性能骤降。生成重放类方法通过从生成模型中采样旧数据与新数据混合训练来缓解该问题。本文将持续训练的分类器与基于扩散的生成模型整合为单一联合优化神经网络,并结合知识蒸馏技术,实现对新任务的稳定适应且无灾难性遗忘。在多个基准测试中,该方法优于当前最先进的生成重放技术。此外,我们还将方法扩展至半监督持续学习场景,在此设置下表现超越基于缓存的重放技术,并以自监督方式评估了所学表征的质量。
原文摘要 · Abstract (English)
In this work, we introduce JDCL - a new method for continual learning with generative rehearsal based on joint diffusion models. Neural networks suffer from catastrophic forgetting defined as abrupt loss in the model's performance when retrained with additional data coming from a different distribution. Generative-replay-based continual learning methods try to mitigate this issue by retraining a model with a combination of new and rehearsal data sampled from a generative model. In this work, we propose to extend this idea by combining a continually trained classifier with a diffusion-based generative model into a single - jointly optimized neural network. We show that such shared parametrization, combined with the knowledge distillation technique allows for stable adaptation to new tasks without catastrophic forgetting. We evaluate our approach on several benchmarks, where it outperforms recent state-of-the-art generative replay techniques. Additionally, we extend our method to the semi-supervised continual learning setup, where it outperforms competing buffer-based replay techniques, and evaluate, in a self-supervised manner, the quality of trained representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。