基于增量学习的图像分类方法及系统技术方案

本申请实施例中提供了一种基于增量学习的图像分类方法、系统及计算机介质，选取旧分类模型的旧类别数据，以及新增量数据，构建增量学习数据集；构建增量学习新分类模型；输入增量学习数据集，至增量学习新分类模型，在增量学习损失函数约束下进行增量学习训练，得到训练后的增量学习新分类模型；输入待分类图像至训练后的增量学习新分类模型，进行图像分类，得到图像分类结果。本申请在旧类别数据基础上连同新类别的数据一起进行学习，在增量学习损失函数约束下实现了使旧类别的映射向量的相似度能够在增量学习的过程中保持一致，进而在学习新类别的信息同时，保持了对旧类别的测试数据的分类性能。测试数据的分类性能。测试数据的分类性能。

【技术实现步骤摘要】
基于增量学习的图像分类方法及系统


[0001]本申请属于图像分类
，具体地，涉及一种基于增量学习的图像分类方法及系统。

技术介绍

[0002]增量学习是机器学习的一个重要研究课题，增量学习的目标是在不忘记已经学习到的知识的基础上，不断地通过新的数据学习新的知识。计算机视觉的很多现实应用都需要模型具备增量学习的能力。比如，人脸识别系统应该支持添加新的人脸数据，同时不应该降低其在旧的人脸数据上的学习性能。又比如，无人超市中的商品识别系统应该不断学习新增加的商品的知识，同时保持原有商品的识别准确率。增量学习和全量学习(即模型在所有新旧数据上重新训练)相比，增量学习的特点是，新的模型不再在旧的大数据上重新训练，而只在新添加的数据上对原有模型进行适当调整后进行训练。然而，目前大多数利用深度学习进行分类的方法在增量学习的过程中都会遭遇“灾难性遗忘”的问题，即在学习新类别的信息后，模型在旧类别的测试数据上的性能大幅度降低。
[0003]图像分类的神经网络通常包括两个部分：特征提取子网络和分类子网络。其中，特征提取网络将输入图像传入一系列的特征提本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于增量学习的图像分类方法，具体包括以下步骤：选取旧分类模型的旧类别数据，以及新增量数据，构建增量学习数据集；构建增量学习新分类模型；输入所述增量学习数据集，至所述增量学习新分类模型，在增量学习损失函数约束下进行增量学习训练，得到训练后的增量学习新分类模型；输入待分类图像至所述训练后的增量学习新分类模型，进行图像分类，得到图像分类结果。2.根据权利要求1所述的图像分类方法，其特征在于，所述构建增量学习新分类模型具体包括：将所述旧分类模型的特征提取网络，作为所述增量学习新分类模型的特征提取网络；在所述旧分类模型的分类网络的输出端，添加新增类别数量的维度后，作为所述增量学习新分类模型的分类网络。3.根据权利要求1所述的图像分类方法，其特征在于，所述增量学习损失函数具体包括交叉熵损失函数、蒸馏损失函数以及相似度矩阵的一致性损失函数。4.根据权利要求3所述的图像分类方法，其特征在于，所述相似度矩阵包括旧相似度矩阵以及新相似度矩阵；所述旧相似度矩阵由旧分类模型的各个旧类别的映射向量之间的相似度组成；所述新相似度矩阵由增量学习新分类模型的各个旧类别的映射向量之间的相似度组成。5.根据权利要求4所述的图像分类方法，其特征在于，所述相似度矩阵的一致性损失函数L(S
old
,S
new
)具体公式为：L(S
old
,S
new
)＝||S
old
‑
S
new
||1；其中，S
old
为旧相似度矩阵；S
new
为新相似度矩阵。6.根据权利要求3所述的图像分类方法，其特征在于，所述交叉熵损失函数L
CE
具体公式为：其中，y
i
表示网络的输出；g
i
表示...

【专利技术属性】
技术研发人员：廖丹萍，
申请(专利权)人：浙江智慧视频安防创新中心有限公司，
类型：发明
国别省市：