神经网络训练方法、图片分类方法及图片分类系统技术方案

本发明专利技术公开一种神经网络训练方法、图片分类方法及图片分类系统。神经网络训练方法S1:准备数据集：源领域图片和第一目标领域图片、第二目标领域图片；S2:将源领域图片和第一目标领域图片输入至第一神经网络，得到第一特征F1；S3:设置第一损失函数，根据第一特征F1计算第一损失，用以对第一神经网络进行训练；S4:将源领域图片输入训练好的第一神经网络,得到的特征F1

【技术实现步骤摘要】
神经网络训练方法、图片分类方法及图片分类系统


[0001]本专利技术涉及信息检索
，尤其涉及一种神经网络训练方法、图片分类方法及图片分类系统。

技术介绍

[0002]深度学习是从数据中学习的一种技术，目前在业界得到广泛的应用，基于深度学习的图片分类具有较高的准确率。
[0003]深度学习之所以强大，是因为其可以从数据中能够学习到特征，从而挖掘新的规律，但深度学习也有其局限性，只适用于其在和训练集的数据里相似特征的，当同样的深度学习网络或者模型应用在新的领域时候，还得需要重新进行训练或者调整。也就是说迁移到新的领域或者当出现的新图片，此在原来训练集中都没有出现过的领域则识别不出来。
[0004]为了解决深度学习的局限性，目前的unsupervised domain adaptation(UDA)领域自适应是一种提升不同领域图片分类准确率的算法。但目前UDA只能是一种领域迁移到另外一种领域，属于一对一的迁移,此外业界也有提出一对多迁移的尝试，但准确率并不理想。

技术实现思路

[0005]本专利技术为解决现有技术中存在的技术问题，提出一种神经网络训练方法，包括步骤：
[0006]S1:准备数据集：源领域图片和第一目标领域图片、第二目标领域图片；
[0007]S2:将源领域图片和第一目标领域图片输入至第一神经网络，得到第一特征F1；
[0008]S3:设置第一损失函数，根据第一特征F1计算第一损失，用以对第一神经网络进行训练；
[0009]S4:将源领域图片输入训练好的第一神经网络,得到的特征F1
’
,将特征F1
’
以及第二目标领域图片输入到第二神经网络，得到第二特征F2；
[0010]S5:设置第二损失函数，根据第二特征F2计算第二损失，用以对第二神经网络进行训练。
[0011]进一步地，还包括步骤S6:训练第三神经网络，其方法为：
[0012]准备源领域图片和第三目标领域图片；
[0013]将源领域图片输入至训练得到的第一神经网络和第二神经网络，得到第二特征F2
’
；
[0014]将特征F2
’
以及第三目标领域图片输入到第三神经网络，计算第三神经网络的损失值并进行训练。
[0015]进一步地，所述第一神经网络为densenet。
[0016]进一步地，所述第二神经网络为densenet。
[0017]进一步地，第一损失函数为MMD损失和CE损失加权求和，其中：
[0018]MMD损失为源域和第一目标域特征求得；
[0019]CE损失为源域图片和第一目标域图片交叉熵之和。
[0020]进一步地，第二损失函数为MMD损失和CE损失加权求和，其中：
[0021]MMD损失为源域和第二目标域特征求得；
[0022]CE损失为源域图片和第二目标域图片交叉熵之和。
[0023]本专利技术还提供一种图片分类方法，包括步骤：
[0024]接收输入的图片；
[0025]神经网络模型对图片进行分析并分类；
[0026]返回输出结果，
[0027]所述神经网络为所述神经网络训练方法得到的神经网络模型。
[0028]本专利技术还提供一种图片分类系统，包括交互单元、处理模块，其中：
[0029]交互单元用以接收输入的图片；
[0030]处理模块用以对输入的图片进行处理并向交互单元返回分类结果，
[0031]所述神经网络为所述神经网络训练方法得到的神经网络模型。
[0032]本专利技术还提供一种计算机可读存储介质，所述存储介质中存储有指令或者程序，所述指令或者程序由处理器加载并执行如所述的图片分类方法。
[0033]本专利技术还提供一种电子设备，包括：处理器、存储介质和总线，所述存储介质存储有所述处理器可执行的机器可读指令，当电子设备运行时，所述处理器与所述存储介质之间通过总线通信，所述处理器执行所述机器可读指令，以执行所述的图片分类方法。
[0034]本专利技术公开的方法和系统中所述模块，在实际应用中，即可以在一台目标服务器上部署多个模块，也可以每一模块独立部署在不同的目标服务器上，特别的，根据需要，为了提供更强大的计算处理能力，也可以根据需要将模块部署到集群目标服务器上。
[0035]由此可见，本专利技术能够实现一对多的领域迁移，能够有效解决解决领域漂移问题，使得目标领域能够取得更好的学习效果也即一种源领域迁移到多种目标领域的图片分类的技术方案。
[0036]为了对本专利技术有更清楚全面的了解，下面结合附图，对本专利技术的具体实施方式进行详细描述。
附图说明
[0037]为了更清楚地说明本申请实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍。显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
[0038]图1为本申请实施例的神经网络训练方法流程示意图。
[0039]图2为本申请实施例的神经网络训练的结构示意图。
具体实施方式
[0040]本文提出的技术方案，实现一对多的领域迁移，能够有效解决领域漂移问题，使得目标领域能够取得更好的学习效果也即一种源领域迁移到多种目标领域的图片分类的技
术方案。
[0041]本申请所述的领域，泛指在迁移学习技术中的概念中，即把一个领域(即源领域)的知识，迁移到另外一个领域(即目标领域)。也包括不同的场景下的图片，比如灯光较强时候拍的图片是一种领域，灯光较弱又是另外一种领域。
[0042]本申请首先训练使用源领域图片和第一个目标领域图片训练第一个神经网络，从而可以实现第一目标领域的知识迁移，即能对第一目标领域图片的识别和分类。
[0043]然后再用训练好的第一神经网络作为源领域图片的输入，以及第二目标领域图片一起训练第二神经网络，又实现对第二目标图片的识别和分类。
[0044]基于上述训练好的第一神经网络、第二神经网络和新的目标图片(即第三目标图片)一起训练一个新的神经网络(即第三神经网络)，从而又可以对第三目标图片(也即另外一个新领域的目标图片)进行识别和分类。
[0045]此外，作为变化的实施例方式，还可以根据第二神经网络的损失函数进行训练第二神经网络的时候，也可以同时训练(更新)第一神经网络的。
[0046]如此不断迭代，可以不断地迁移到另一个新的目标领域，从而实现一对多的领域迁移。
[0047]请参阅图1以及图2，本申请针对现有技术的技术缺陷，公开一种神经网络训练方法，包括步骤：
[0048]S1:准备数据集：源领域图片和第一目标领域图片、第二目标领域图片；
[0049]S2:将源领域图片和第一目标领域图片输入至第一神经网络，得到第一特征F1；
[0050]S3:设置第一损失函数，根据本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种神经网络训练方法，其特征是，包括步骤：S1:准备数据集：源领域图片和第一目标领域图片、第二目标领域图片；S2:将源领域图片和第一目标领域图片输入至第一神经网络，得到第一特征F1；S3:设置第一损失函数，根据第一特征F1计算第一损失，用以对第一神经网络进行训练；S4:将源领域图片输入训练好的第一神经网络,得到的特征F1
’
,将特征F1
’
以及第二目标领域图片输入到第二神经网络，得到第二特征F2；S5:设置第二损失函数，根据第二特征F2计算第二损失，用以对第二神经网络进行训练。2.如权利要求1所述的神经网络训练方法，其特征是，还包括步骤S6:训练第三神经网络，其方法为：准备源领域图片和第三目标领域图片；将源领域图片输入至训练得到的第一神经网络和第二神经网络，得到第二特征F2
’
；将特征F2
’
以及第三目标领域图片输入到第三神经网络，计算第三神经网络的损失值并进行训练。3.如权利要求1所述的神经网络训练方法，其特征是，所述第一神经网络为densenet。4.如权利要求1或2所述的神经网络训练方法，其特征是，所述第二神经网络为densenet。5.如权利要求1所述的神经网络训练方法，其特征是，第...

【专利技术属性】
技术研发人员：曾祥云，朱姬渊，
申请(专利权)人：上海易康源医疗健康科技有限公司，
类型：发明
国别省市：