Timezone: »
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)
Research Scientist at SK T-Brain
Jaehyun Jun (Seoul National University)
Jung-Woo Ha (Clova AI Research, NAVER Corp.)
Byoung-Tak Zhang (Seoul National University & Surromind Robotics)
Related Events (a corresponding poster, oral, or spotlight)
-
2017 Poster: Overcoming Catastrophic Forgetting by Incremental Moment Matching »
Thu Dec 7th 02:30 -- 06:30 AM Room Pacific Ballroom #37
More from the Same Authors
-
2020 Workshop: BabyMind: How Babies Learn and How Machines Can Imitate »
Byoung-Tak Zhang · Gary Marcus · Angelo Cangelosi · Pia Knoeferle · Klaus Obermayer · David Vernon · Chen Yu -
2020 Poster: Self-supervised Auxiliary Learning with Meta-paths for Heterogeneous Graphs »
Dasol Hwang · Jinyoung Park · Sunyoung Kwon · KyungMin Kim · Jung-Woo Ha · Hyunwoo Kim -
2020 Social: NeurIPS 2020 Social ML in Korea »
Jung-Woo Ha -
2018 Poster: Answerer in Questioner's Mind: Information Theoretic Approach to Goal-Oriented Visual Dialog »
Sang-Woo Lee · Yu-Jung Heo · Byoung-Tak Zhang -
2018 Spotlight: Answerer in Questioner's Mind: Information Theoretic Approach to Goal-Oriented Visual Dialog »
Sang-Woo Lee · Yu-Jung Heo · Byoung-Tak Zhang -
2018 Poster: Bilinear Attention Networks »
Jin-Hwa Kim · Jaehyun Jun · Byoung-Tak Zhang -
2016 Poster: Multimodal Residual Learning for Visual QA »
Jin-Hwa Kim · Sang-Woo Lee · Donghyun Kwak · Min-Oh Heo · Jeonghee Kim · Jung-Woo Ha · Byoung-Tak Zhang -
2010 Poster: Generative Local Metric Learning for Nearest Neighbor Classification »
Yung-Kyun Noh · Byoung-Tak Zhang · Daniel Lee