不用回放数据,用变分自编码器实现高效增量生成建模
Frugal Incremental Generative Modeling using Variational Autoencoders
- 基于多模态潜在空间设计无回放增量生成模型
- 参数量几乎不变,仍保持SOTA生成准确率
- 适合资源受限场景下的持续学习应用
持续学习在深度学习中潜力巨大,但面临灾难性遗忘等挑战。随着强大基础与生成模型的发展,该范式更具可行性,但传统回放方法导致数据量持续增长,带来可扩展性问题。本文提出一种基于变分自编码器(VAE)的无回放增量生成建模方法,核心贡献包括:(i) 设计了基于精心构建的多模态潜在空间的增量生成模型;(ii) 引入正交性准则以缓解学习过程中的灾难性遗忘。所提方法包含静态与动态两种变体,参数量基本不增长(或可控增长)。大量实验表明,该方法在生成性能达到当前最优(SOTA)的同时,内存消耗比相近方法低一个数量级。
原文摘要 · Abstract (English)
Continual or incremental learning holds tremendous potential in deep learning with different challenges including catastrophic forgetting. The advent of powerful foundation and generative models has propelled this paradigm even further, making it one of the most viable solution to train these models. However, one of the persisting issues lies in the increasing volume of data particularly with replay-based methods. This growth introduces challenges with scalability since continuously expanding data becomes increasingly demanding as the number of tasks grows. In this paper, we attenuate this issue by devising a novel replay-free incremental learning model based on Variational Autoencoders (VAEs). The main contribution of this work includes (i) a novel incremental generative modelling, built upon a well designed multi-modal latent space, and also (ii) an orthogonality criterion that mitigates catastrophic forgetting of the learned VAEs. The proposed method considers two variants of these VAEs: static and dynamic with no (or at most a controlled) growth in the number of parameters. Extensive experiments show that our method is (at least) an order of magnitude more ``memory-frugal'' compared to the closely related works while achieving SOTA accuracy scores.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。