一种基于卷积神经元网络的中医舌诊模型的确定系统的制作方法_2

文档序号:10105909阅读:来源:国知局
9]本实用新型实施例中,参看图2所示,不仅需获取患者舌部图像信息作为训练集及交叉验证集,还需获取患者舌部图像信息作为测试集,再将所述测试集作为最优舌诊模型的输入,得到测试集中各测试样本的舌诊结果,通过统计得到该最优舌诊模型的诊断准确率为99.5%,因为测试集中的各测试样本代表患者实际病情,从而能够证明最优舌诊模型的诊断结果具备较高的准确率和可靠性。
[0050]本实用新型实施例中,优选的,获取到的训练集、交叉验证集、测试集中的各舌诊样本图像(舌部图像)的比例为6:2:2。
[0051]本实用新型实施例中,首先统一所述训练集、交叉验证集及测试集中的舌部图像的大小,并将统一大小后的舌部图像转化为一定的数据格式。
[0052]本实用新型实施例中,例如,可以将舌部图像转换为Caffe能处理的lmdb格式,Caffe是一种开源工具,lmdb是一种数据格式。
[0053]本实用新型实施例中,所述卷积神经元网络的结构包括:输入层、第一卷积层、第一池化层、第二卷积层、第二池化层及全连接输出层;
[0054]本实用新型实施例中,所述卷积神经元网络的结构具体包括:每层神经元构成、个数以及输入和输出。
[0055]本实用新型实施例中,参看图3所示,所述输入层由原始图像像素排列构成,用于输入患者舌部图像信息;
[0056]所述第一卷积层,用于根据输入层输入的患者舌部图像信息生成第一卷积特征图,卷积特征图相比于原始图像特征(像素)而言更抽象、概括,更能表征原始图像;
[0057]所述第一池化层,用于对第一卷积层输出的第一卷积特征图进行采样,得到采样区域的第一特征平均值(也称为第一池化特征图),以解决过拟合的情况,并降低计算量,提升计算效率;
[0058]所述第二卷积层,用于根据第一池化层输出的第一特征平均值生成第二卷积特征图;
[0059]所述第二池化层,用于对第二卷积层输出的第二卷积特征图进行采样,得到采样区域的第二特征平均值(也称为第二池化特征图);
[0060]所述全连接输出层,用于根据第二池化层输出的第二特征平均值进行训练,分类输出舌诊结果,所述舌诊结果包括:正常、肝硬化、肝癌、乙肝。
[0061]本实用新型实施例中,运用反向传播,利用所述训练集对设置后的卷积神经元网络进行训练,确定不同结构的卷积神经元网络的卷积神经元网络确定单元包括:
[0062]初始化模块,用于初始化卷积神经元网络中的所有权重,使它们近似等于0 ;
[0063]激活值生成模块,用于根据训练集中给定的训练样本进行前向传导,生成全连接输出层的激活值;
[0064]误差确定模块,用于利用全连接输出层产生的激活值与实际值之间的误差和全连接输出层的权重,确定第二池化层的误差,以此类推,算出每一层相应的误差,所述误差称为残差,表明了该层对最终输出值的残差产生了多少影响;以此类推,可以算出每一层相应的残差;
[0065]权重更新模块,用于利用每一层的残差更新所有权重;
[0066]训练结果保存模块,用于重复执行所述激活值生成模块、误差确定模块及权重更新模块的过程,直至完成设定的次数后,完成卷积神经元网络的训练过程,保存训练结果。
[0067]在前述基于卷积神经元网络的中医舌诊模型的确定系统的【具体实施方式】中,可选地,所述卷积神经元网络确定单元包括:
[0068]并行执行模块,用于将所述训练集拆成多个子训练集,每个子训练集对设置后的各个卷积神经元网络的训练过程由各子训练集对应的图形处理器进行并行处理;
[0069]卷积神经元网络确定模块,用于将各图形处理器的并行处理结果进行累加,完成所述训练集对设置后的卷积神经元网络的训练过程。
[0070]本实用新型实施例中,例如,原始Caffe只实现了单图形处理器(GraphicsProcessing Unit, GPU)计算,当计算量更大时,效率就会出现停滞,由于卷积神经元网络的的训练过程可以通过并行计算再累加的方式实现,因此,本实用新型实施例可以通过将多GPU的并行计算结果进行累加,使之可以实现多GPU计算,具体的,通过将所述训练集拆成多个子训练集,每个子训练集对设置后的各个卷积神经元网络的训练过程由各子训练集对应的GPU进行并行处理,每个子训练集交由一个GPU进行计算,最后再将并行处理结果进行累加从而实现了多GPU计算并完成所述训练集对设置后的卷积神经元网络的训练过程,使计算效率大大提升。与单GPU相比,GPU的并行运算能力能提升多倍的计算效率。
[0071]以上所述是本实用新型的优选实施方式,应当指出,对于本技术领域的普通技术人员来说,在不脱离本实用新型所述原理的前提下,还可以作出若干改进和润饰,这些改进和润饰也应视为本实用新型的保护范围。
【主权项】
1.一种基于卷积神经元网络的中医舌诊模型的确定系统,其特征在于,包括: 第一获取单元,用于获取患者舌部图像信息作为训练集、交叉验证集; 卷积神经元网络确定单元,用于设置卷积神经元网络每层神经元的个数,利用所述训练集对设置后的多个不同卷积神经元网络分别进行训练,确定多个不同结构的卷积神经元网络; 最优舌诊模型确定单元,用于将所述交叉验证集分别作为所述多个不同结构卷积神经元网络的输入,将舌诊结果准确率最高的卷积神经元网络作为最优舌诊模型。2.根据权利要求1所述的系统,其特征在于,还包括: 第二获取单元,用于获取患者舌部图像信息作为测试集; 测试单元,用于将所述测试集作为最优舌诊模型的输入,得到测试集中各测试样本的舌诊结果。3.根据权利要求1或2所述的系统,其特征在于,还包括: 预处理单元,用于将所述训练集、交叉验证集及测试集中的舌部图像大小进行归一化处理,并转化为一定的数据格式。4.根据权利要求1所述的系统,其特征在于,所述卷积神经元网络的结构包括:输入层、第一卷积层、第一池化层、第二卷积层、第二池化层及全连接输出层; 所述输入层,用于输入患者舌部图像信息; 所述第一卷积层,用于与输入层连接,并根据输入层输入的患者舌部图像信息生成第一卷积特征图; 所述第一池化层,用于与第一卷积层连接,并对第一卷积层输出的第一卷积特征图进行采样,得到采样区域的第一特征平均值; 所述第二卷积层,用于与第一池化层连接,并根据第一池化层输出的第一特征平均值生成第二卷积特征图; 所述第二池化层,用于与第二卷积层连接,并对第二卷积层输出的第二卷积特征图进行采样,得到采样区域的第二特征平均值; 所述全连接输出层,用于与第二池化层连接,并根据第二池化层输出的第二特征平均值,分类输出舌诊结果,所述舌诊结果包括:正常、肝硬化、肝癌、乙肝。5.根据权利要求4所述的系统,其特征在于,所述卷积神经元网络确定单元包括: 初始化模块,用于初始化卷积神经元网络中的所有权重; 激活值生成模块,用于根据训练集中给定的训练样本进行前向传导,生成全连接输出层的激活值; 误差确定模块,用于利用全连接输出层产生的激活值与实际值之间的误差和全连接输出层的权重,确定第二池化层的误差,以此类推,算出每一层相应的误差; 权重更新模块,用于利用每一层的误差更新所有权重; 训练结果保存模块,用于重复执行所述激活值生成模块、误差确定模块及权重更新模块的过程,直至完成设定的次数后,完成卷积神经元网络的训练过程,保存训练结果。6.根据权利要求1所述的系统,其特征在于,所述卷积神经元网络确定单元包括: 并行执行模块,用于将所述训练集拆成多个子训练集,每个子训练集对设置后的各个卷积神经元网络的训练过程由各子训练集对应的图形处理器进行并行处理; 卷积神经元网络确定模块,用于将各图形处理器的并行处理结果进行累加,完成所述训练集对设置后的卷积神经元网络的训练过程。
【专利摘要】本实用新型提供一种基于卷积神经元网络的中医舌诊模型的确定系统,能够提高中医舌诊模型诊断结果的准确率和可靠性。所述系统包括:第一获取单元,用于获取患者舌部图像信息作为训练集、交叉验证集;卷积神经元网络确定单元,用于利用所述训练集确定多个不同结构的卷积神经元网络;最优舌诊模型确定单元,用于将所述交叉验证集分别作为所述多个不同结构卷积神经元网络的输入,将舌诊结果准确率最高的卷积神经元网络作为最优舌诊模型;第二获取单元,用于获取患者舌部图像信息作为测试集;测试单元,用于将所述测试集作为最优舌诊模型的输入,得到测试集中各测试样本的舌诊结果。本实用新型适用于中医智能诊断技术领域。
【IPC分类】G06F19/00
【公开号】CN205015889
【申请号】CN201520743202
【发明人】王志良, 张佳伟
【申请人】北京科技大学
【公开日】2016年2月3日
【申请日】2015年9月23日
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1