Overcoming Catastrophic Forgetting by Incremental Moment Matching

Abstract

Catastrophic forgetting is a problem which refers to losing the information of the first task after training from the second task in continual learning of neural networks. To resolve this problem, we propose the incremental moment matching (IMM), which uses the Bayesian neural network framework. IMM assumes that the posterior distribution of parameters of neural networks is approximated with Gaussian distribution and incrementally matches the moment of the posteriors, which are trained for the first and second task, respectively. To make our Gaussian assumption reasonable, the IMM procedure utilizes various transfer learning techniques including weight transfer, L2-norm of old and new parameters, and a newly proposed variant of dropout using old parameters. We analyze our methods on the MNIST and CIFAR-10 datasets, and then evaluate them on a real-world life-log dataset collected using Google Glass. Experimental results show that IMM produces state-of-the-art performance in a variety of datasets.

7 Figures and Tables

Cite this paper

@article{Lee2017OvercomingCF, title={Overcoming Catastrophic Forgetting by Incremental Moment Matching}, author={Sang-Woo Lee and Jin-Hwa Kim and JungWoo Ha and Byoung-Tak Zhang}, journal={CoRR}, year={2017}, volume={abs/1703.08475} }