Bi-Objective Continual Learning: Learning ‘New’ While Consolidating ‘Known’

  • Xiaoyu Tao Xi'an Jiaotong University
  • Xiaopeng Hong Xi'an Jiaotong University
  • Xinyuan Chang Xi'an Jiaotong University
  • Yihong Gong Xi'an Jiaotong University

Abstract

In this paper, we propose a novel single-task continual learning framework named Bi-Objective Continual Learning (BOCL). BOCL aims at both consolidating historical knowledge and learning from new data. On one hand, we propose to preserve the old knowledge using a small set of pillars, and develop the pillar consolidation (PLC) loss to preserve the old knowledge and to alleviate the catastrophic forgetting problem. On the other hand, we develop the contrastive pillar (CPL) loss term to improve the classification performance, and examine several data sampling strategies for efficient onsite learning from ‘new’ with a reasonable amount of computational resources. Comprehensive experiments on CIFAR10/100, CORe50 and a subset of ImageNet validate the BOCL framework. We also reveal the performance accuracy of different sampling strategies when used to finetune a given CNN model. The code will be released.

Published
2020-04-03
Section
AAAI Technical Track: Machine Learning