Instance-level and Class-level Contrastive Incremental Learning for Image Classification
Jia-yi Han, Jian–wei Liu · 2022 International Joint Conference on Neural Networks (IJCNN) · 2022
Recently, people pay more attention to catastrophic forgetting problem, that is, the ability of the model to recognize old tasks decreases dramatically when new tasks are added incrementally. Previous studies focused on making the outputs or intermediate features of the new model as similar as possible the old model but ignored the inner-class assignment information. We consider that the inner-class information can effectively reflect the association pattern and intrinsic nature of the samples with each other, so that maintaining the inner-class relationship among task data is helpful to alleviate the negative impact of catastrophic forgetting. Contrastive learning exhibits excellent performance under self-supervising tasks, which can enhance robustness and make representation more compact. We propose an Incremental Learning algorithm with Instance-level and Class-level Contrastive loss and Knowledge Distillation (IL-ICCKD) as common constraints. Specifically, we encourage our model to maintain the knowledge learned in the past from perspectives of instance characteristics and inner-class assignment distribution. At the same time, our model uses a spatial group-wise enhanced attention mechanism to make the learned representations grasp the spatial distribution of subfeatures. We extensively evaluate our framework on three popular benchmark datasets and demonstrate the performance beyond other models.