Timezone: »

Overcoming Catastrophic Forgetting by Incremental Moment Matching
Sang-Woo Lee · Jin-Hwa Kim · Jaehyun Jun · Jung-Woo Ha · Byoung-Tak Zhang

Wed Dec 06 05:45 PM -- 05:50 PM (PST) @ Hall A

Catastrophic forgetting is a problem of neural networks that loses the information of the first task after training the second task. Here, we propose incremental moment matching (IMM) to resolve this problem. IMM incrementally matches the moment of the posterior distribution of neural networks, which is trained for the first and the second task, respectively. To make the search space of posterior parameter smooth, the IMM procedure is complemented by various transfer learning techniques including weight transfer, L2-norm of the old and the new parameter, and a variant of dropout with the old parameter. We analyze our approach on various datasets including the MNIST, CIFAR-10, Caltech-UCSD-Birds, and Lifelog datasets. Experimental results show that IMM achieves state-of-the-art performance in a variety of datasets and can balance the information between an old and a new network.

Author Information

Sang-Woo Lee (Naver Corp.)
Jin-Hwa Kim (SK T-Brain)
Jaehyun Jun (Seoul National University)
Jung-Woo Ha (Clova AI Research, NAVER Corp.)
Jung-Woo Ha

- Head, AI Innovation, NAVER Cloud - Research Fellow, NAVER AI Lab - Datasets and Benchmarks Co-Chair, NeurIPS 2023 - Socials Co-Chair, ICML 2023 - Socials Co-Chair, NeurIPS 2022 - BS, Seoul National University - PhD, Seoul National University

Byoung-Tak Zhang (Seoul National University & Surromind Robotics)

Related Events (a corresponding poster, oral, or spotlight)

More from the Same Authors