EVCL: Elastic Variational Continual Learning with Weight Consolidation

Hunar Batra, Ronald Clark
6/23/2024
cs.LGstat.ML

No official code found. Best implementation: hunarbatra/elastic-variational-continual-learning (3 stars, last commit 2024-04-28).

Abstract

Continual learning aims to allow models to learn new tasks without forgetting what has been learned before. This work introduces Elastic Variational Continual Learning with Weight Consolidation (EVCL), a novel hybrid model that integrates the variational posterior approximation mechanism of Variational Continual Learning (VCL) with the regularization-based parameter-protection strategy of Elastic Weight Consolidation (EWC). By combining the strengths of both methods, EVCL effectively mitigates catastrophic forgetting and enables better capture of dependencies between model parameters and task-specific data. Evaluated on five discriminative tasks, EVCL consistently outperforms existing baselines in both domain-incremental and task-incremental learning scenarios for deep discriminative models.

View on arXivView PDF

Code Implementations(1)

EVCL-WC: Elastic Variational Continual Learning with Weight Consolidation. Paper: https://arxiv.org/abs/2406.15972

30PythonApr 9, 20242 years ago
bayesian-deep-learningcontinual-learning

Cite this paper

@article{batra2024evcl,
  title  = {EVCL: Elastic Variational Continual Learning with Weight Consolidation},
  author = {Hunar Batra and Ronald Clark},
  year   = {2024},
  eprint = {2406.15972},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.15972v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.15972.svg)](https://codeofpaper.com/paper/2406.15972)

Discussion