Prototype Representation Expansion in Incremental Learning
Keming Mao,Yong Luo,Yitao Ren,Ruixiang Wang
DOI: https://doi.org/10.1007/s11063-023-11317-x
IF: 2.565
2023-01-01
Neural Processing Letters
Abstract:Deep neural networks have made outstanding achievements in many static tasks, however, when faced with incremental scenario, they suffer from catastrophic forgetting since the previous data is usually inaccessible. Stored data and generative models are commonly used for maintaining model performance, but there exist problems of memory utilization and privacy safety. In this paper, a novel non-exemplar based incremental learning model, Prototype Representation Expansion (PRE), which provides a great degree to retain the feature space of old tasks, is proposed. Firstly, prototypes are generated to meet the stability and robustness. The mean value of feature embedding for each class is used as prototype to maintain the model stability. Meanwhile, PRE also selects prototypes according to their responses of classifier by feature disturbance noise injection, and the decision boundary can better be maintained. Secondly, prototypes of various classes are linearly combined to construct the hybrid prototype with mixed labels. Along with prototype augmentation, they are used for incremental training phase. We conduct extensive experiments on two benchmark datasets, CIFAR-100 and ImageNet-Subset. It shows that PRE can be combined with some non-exemplar based methods to significantly improve their ability and achieve comparable performance to exemplar based methods.