Dynamic Feature Learning and Matching for Class-Incremental Learning

Sunyuan Qiang,Yanyan Liang,Jun Wan,Du Zhang
2024-05-14
Abstract:Class-incremental learning (CIL) has emerged as a means to learn new classes incrementally without catastrophic forgetting of previous classes. Recently, CIL has undergone a paradigm shift towards dynamic architectures due to their superior performance. However, these models are still limited by the following aspects: (i) Data augmentation (DA), which are tightly coupled with CIL, remains under-explored in dynamic architecture scenarios. (ii) Feature representation. The discriminativeness of dynamic feature are sub-optimal and possess potential for refinement. (iii) Classifier. The misalignment between dynamic feature and classifier constrains the capabilities of the model. To tackle the aforementioned drawbacks, we propose the Dynamic Feature Learning and Matching (DFLM) model in this paper from above three perspectives. Specifically, we firstly introduce class weight information and non-stationary functions to extend the mix DA method for dynamically adjusting the focus on memory during training. Then, von Mises-Fisher (vMF) classifier is employed to effectively model the dynamic feature distribution and implicitly learn their discriminative properties. Finally, the matching loss is proposed to facilitate the alignment between the learned dynamic features and the classifier by minimizing the distribution distance. Extensive experiments on CIL benchmarks validate that our proposed model achieves significant performance improvements over existing methods.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper addresses the problem in Class-Incremental Learning (CIL). Existing CIL methods suffer from limitations in exploring data augmentation strategies, suboptimal dynamic feature representation, and mismatch between classifiers and features in dynamic architectures. To tackle these issues, the paper proposes a Dynamic Feature Learning and Matching (DFLM) model, which includes: 1) introducing class weight information and non-static function for enhanced mixed data augmentation strategy; 2) utilizing von Mises-Fisher (vMF) classifier to effectively model the dynamic feature distribution; 3) proposing a matching loss to minimize the distribution distance between features and classifiers, promoting their alignment. Experimental results demonstrate that the proposed model significantly improves performance on CIL benchmarks.