Variational autoencoder (VAE) is a deep generative model for unsupervised learning, allowing to encode observations into the meaningful latent space. VAE is prone to catastrophic forgetting when tasks arrive sequentially, and only the data for the current one is available. We address this problem of continual learning for VAEs. It is known that the choice of the prior distribution over the latent space is crucial for VAE in the non-continual setting. We argue that it can also be helpful to avoid catastrophic forgetting. We learn the approximation of the aggregated posterior as a prior for each task. This approximation is parametrised as an additive mixture of distributions induced by encoder evaluated at trainable pseudo-inputs. We use a greedy boosting-like approach with entropy regularisation to learn the components. This method encourages components diversity, which is essential as we aim at memorising the current task with the fewest components possible. Based on the learnable prior, we introduce an end-to-end approach for continual learning of VAEs and provide empirical studies on commonly used benchmarks (MNIST, Fashion MNIST, NotMNIST) and CelebA datasets. For each dataset, the proposed method avoids catastrophic forgetting in a fully automatic way.


    Zugriff

    Download


    Exportieren, teilen und zitieren



    Titel :

    BooVAE: Boosting Approach for Continual Learning of VAE


    Beteiligte:
    Kuzina, Anna (Autor:in) / Egorov, Evgenii (Autor:in) / Burnaev, Evgeny (Autor:in)

    Erscheinungsdatum :

    2019


    Anmerkungen:

    Accepted to NeurIPS 2021. 14 pages, 4 figures, 4 tables


    Medientyp :

    Preprint


    Format :

    Elektronische Ressource


    Sprache :

    Englisch




    Continual SLAM: Beyond Lifelong Simultaneous Localization and Mapping Through Continual Learning

    Vödisch, Niclas / Cattaneo, Daniele / Burgard, Wolfram et al. | Springer Verlag | 2023


    Expandable Orbit Decay Prediction Using Continual Learning

    Junhua He / Hua Wang / Haitao Wang et al. | DOAJ | 2024

    Freier Zugriff

    Evaluating Differential Privacy in Federated Continual Learning

    Ouyang, Junyan / Han, Rui / Liu, Chi Harold | IEEE | 2023


    CONTINUAL PROACTIVE LEARNING FOR AUTONOMOUS ROBOT AGENTS

    DAVIDSON JAMES / MASON JULIAN / POURSOHI ARSHAN | Europäisches Patentamt | 2022

    Freier Zugriff

    CONTINUAL PROACTIVE LEARNING FOR AUTONOMOUS ROBOT AGENTS

    DAVIDSON JAMES / MASON JULIAN / POURSOHI ARSHAN | Europäisches Patentamt | 2021

    Freier Zugriff