Prototypes Sampling Mechanism for Class Incremental Learning
Zhe Tao,Shucheng Huang,Gang Wang
DOI: https://doi.org/10.1109/access.2023.3301123
IF: 3.9
2023-08-12
IEEE Access
Abstract:Incremental learning aims to alleviate the catastrophic forgetting problem of deep neural networks during learning sequential data stream. This problem is even more challenging when old data is unavailable, since learning system can only be trained under the supervision of current data. To address this problem, we proposed a prototype sampling mechanism based on K-means clustering method. On the one hand, we proposed to use K-means clustering to pick out class-representative prototypes for each old class. During incremental stages, prototypes and deep features from current data are trained together to maintain the distinction and balance between old and new classes. On the other hand, we proposed to attach a mask to the loss function based on the cosine similarity between the prototypes and the current data. Which further enhances the discrimination between old and new classes compared to naive knowledge distillation schemes. Extensive experiments conducted on three benchmark datasets including CIFAR100, Tiny-ImageNet and vggface2 verified the effectiveness and advantages of our proposed method. Specifically, we improved class incremental performance by 1.6%, 1.2% and 1.7% on three datasets respectively.
computer science, information systems,telecommunications,engineering, electrical & electronic