CKDF: Cascaded Knowledge Distillation Framework for Robust Incremental Learning

Kunchi Li

doi:10.1109/TIP.2022.3176130

CKDF: Cascaded Knowledge Distillation Framework for Robust Incremental Learning

2022-01-01

18

PRE

AI

DOI：10.1109/TIP.2022.3176130

Original

Original request for help

Save

Abstract

En 中文

Recently, owing to the superior performances, knowledge distillation-based (kd-based) methods with the exemplar rehearsal have been widely applied in class incremental learning (CIL). However, we discover that they suffer from the feature uncalibration problem, which is caused by directly transferring knowledge from the old model immediately to the new model when learning a new task. As the old model confuses the feature representations between the learned and new classes, the kd loss and the classification loss used in kd-based methods are heterogeneous. This is detrimental if we learn the existing knowledge from the old model directly in the way as in typical kd-based methods. To tackle this problem, the feature calibration network (FCN) is proposed, which is used to calibrate the existing knowledge to alleviate the feature representation confusion of the old model. In addition, to relieve the task-recency bias of FCN caused by the limited storage memory in CIL, we propose a novel image-feature hybrid sample rehearsal strategy to train FCN by splitting the memory budget to store the image-and-feature exemplars of the previous tasks. As feature embeddings of images have much lower-dimensions, this allows us to store more samples to train FCN. Based on these two improvements, we propose the Cascaded Knowledge Distillation Framework (CKDF) including three main stages. The first stage is used to train FCN to calibrate the existing knowledge of the old model. Then, the new model is trained simultaneously by transferring knowledge from the calibrated teacher model through the knowledge distillation strategy and learning new classes. Finally, after completing the new task learning, the feature exemplars of previous tasks are updated. Importantly, we demonstrate that the proposed CKDF is a general framework that can be applied to various kd-based methods. Experimental results show that our method achieves state-of-the-art performances on several CIL benchmarks.

Keywords:

Task analysis

Computational modeling

Adaptation models

Data models

Training

Knowledge engineering

Feature extraction

Feature calibration

hybrid exemplars

cascaded knowledge distillation

incremental learning

CKDF: Cascaded Knowledge Distillation Framework for Robust Incremental Learning

Abstract

Journal

Organization

Cited Papers