基于K-means聚类原型的增量学习方法

PROTOTYPE REINFORCED INCREMENTAL LEARNING BASED ON K-MEANS CLUSTERING

  • 摘要: 增量学习旨在缓解深度神经网络的灾难性遗忘问题。现有的增量学习方法面临的主要问题包括:隐私泄露、消耗额外内存、模型参数量线性化增长等。针对现有方法的不足,以知识蒸馏的正则化手段为基础,提出一种基于k-means自动聚类算法的原型采样机制,对旧类原型和新数据的深层特征进行联合分类训练,以保持新旧类别的区分性和平衡性。实验结果表明,该方法在两个公开数据集CIFAR100和Tiny-ImageNet上比现有方法分类精度平均提高17.4百分点和16.9百分点,验证了该方法的有效性和优势。

     

    Abstract: Incremental learning aims to alleviate the catastrophic forgetting problem of deep neural networks. The main problems faced by existing incremental learning methods include: privacy leakage, consumption of additional memory, and linear growth of model parameters. In view of the shortcomings of the existing methods, based on the regularization method of knowledge distillation, the paper proposed a prototype sampling mechanism based on the k-means automatic clustering algorithm, and performed joint classification training on the old class prototypes and deep features of the new data to maintain distinction and balance between old and new categories. The experiment results show that the method has an average classification accuracy improvement of 17.4 and 16.9 percentage points compared with the existing methods on two public datasets CIFAR100 and Tiny-ImageNet, which verifies the effectiveness and advantages of the method in the paper.

     

/

返回文章
返回