RTRA: Rapid Training of Regularization-based Approaches in Continual Learning
Machine Learning
2023-12-18 v1 Computer Vision and Pattern Recognition
Abstract
Catastrophic forgetting(CF) is a significant challenge in continual learning (CL). In regularization-based approaches to mitigate CF, modifications to important training parameters are penalized in subsequent tasks using an appropriate loss function. We propose the RTRA, a modification to the widely used Elastic Weight Consolidation (EWC) regularization scheme, using the Natural Gradient for loss function optimization. Our approach improves the training of regularization-based methods without sacrificing test-data performance. We compare the proposed RTRA approach against EWC using the iFood251 dataset. We show that RTRA has a clear edge over the state-of-the-art approaches.
Cite
@article{arxiv.2312.09361,
title = {RTRA: Rapid Training of Regularization-based Approaches in Continual Learning},
author = {Sahil Nokhwal and Nirman Kumar},
journal= {arXiv preprint arXiv:2312.09361},
year = {2023}
}