当前位置: 首页 > 专利查询>中南大学专利>正文

数据偏差下的眼底图分类方法及成像方法技术

技术编号:27847355 阅读:40 留言:0更新日期:2021-03-30 12:59
本发明专利技术公开了一种数据偏差下的眼底图分类方法,包括获取历史的眼底图像数据、标记并分类;构建目标网络和在线网络的基础模型;共享目标网络和在线网络的参数权重;为目标网络提供课程模块;为在线网络构建监督对抗损失函数;进行第一阶段学习和进行第二阶段学习,得到最终的网络模型;采用最终的网络模型对实时获取的眼底图像数据进行分类。本发明专利技术还公开了一种包括所述数据偏差下的眼底图分类方法的成像方法。本发明专利技术能够较好地处理样本中存在的偏差问题;同时在线网络和目标网络的网络参数相互自适应调整,动态引导模型对样本进行学习,从而使得模型的预测从有偏转向无偏,提升了模型的敏感度,而且可靠性高,敏感性好,分类准确性高。准确性高。准确性高。

【技术实现步骤摘要】
数据偏差下的眼底图分类方法及成像方法


[0001]本专利技术书属于图像处理领域,具体涉及一种数据偏差下的眼底图分类方法及成像方法。

技术介绍

[0002]随着经济技术的发展和人们生活水平的提高,人工智能算法已经广泛应用于人们的生产和生活中,给人们的生产和生活带来了无尽的便利。
[0003]目前,基于人工智能算法的计算机辅助图像处理技术,已经广泛应用于医院等单位。在医疗领域,基于人工智能算法的计算机辅助图像处理技术已经广泛应用于眼底图像的分类、识别和成像等,给相关人员带来了巨大的便利,也极大地提高了效率。
[0004]但是,基于人工智能算法的计算机辅助图像处理技术在具体应用时,依旧存在一些问题:由于异常图像的数目较少,以及异常图像难以具体正确分类的问题,这使得目前的基于人工智能算法的计算机辅助图像处理技术,存在较大的数据偏差;而假阳性的判断指标的引入,会在模型训练时会导致模型产生一些偏差,从而严重影响模型的分类结果和分类精确性。

技术实现思路

[0005]本专利技术的目的之一在于提供一种可靠性高、敏感性好和分类准确性高的数据偏差下的眼底图分类方法。
[0006]本专利技术的目的之二在于提供一种包括了所述数据偏差下的眼底图分类方法的成像方法。
[0007]本专利技术提供的这种数据偏差下的眼底图分类方法,包括如下步骤:
[0008]S1.获取历史的眼底图像数据,标记并分类;
[0009]S2.构建目标网络和在线网络的基础模型;
[0010]S3.采用指数移动平均法共享目标网络和在线网络的参数权重;
[0011]S4.为目标网络提供课程模块;
[0012]S5.为在线网络构建监督对抗损失函数;
[0013]S6.进行第一阶段学习:通过在线网络的对抗学习,使得模型收敛至某个表征空间;
[0014]S7.进行第二阶段学习:固定步骤S6得到的在线网络的表征参数并传递给目标网络,同时根据双课程样本加权方法,更新目标网络的分类器参数,从而得到最终的网络模型;
[0015]S8.采用步骤S7得到的最终的网络模型,对实时获取的眼底图像数据进行分类。
[0016]步骤S1所述的历史的眼底图像数据,标记并分类,具体为获取历史的眼底图像数据,根据设定的要求进行标记分类,同时将数据分为训练集和测试集。
[0017]步骤S2所述的构建目标网络和在线网络的基础模型,具体为采用如下步骤构建模
型:
[0018]A.目标网络包括双分支结构和分类器;双分支结构包括空间注意力网络和通道注意力网络;分类器采用双线性融合分类器;
[0019]B.空间注意力网络的输入为原始图像,监督标签是注意力图或者分割标签图;网络为注意力Unet结构,具有五层编码和五层解码;从编码最底层输出的向量,为双线性融合提供一条高级语义信息;
[0020]C.通道注意力网络为SE

Resnet50结构,在每一个网络大模块结构前均加入通道注意力模块,最后删除分类器层,输出一条和空间注意力网络输出的向量维度所匹配的向量;
[0021]D.分类器为双线性融合分类器,对于空间注意力网络和通道注意力网络输出的卷积,操作后得到的增强的鉴别特征计算公式为式中为定义的鉴别特征图,F为来源于卷积层的输入特征图,m(F)为三维注意力特征图,
×
为点乘;
[0022]E.将步骤B得到的空间注意力网络的输出向量进行转置,然后与步骤C得到的通道注意力网络的通道向量,采用如下公式进行计算,并将计算的融合的结果进行矩阵向量化,并上传至分类器从而得到最终的预测概率:
[0023][0024]式中m(F)为注意力的特征图,且提供了一个权重给每个局部的像素和通道;sqrt为数学符号开根号;m
s
(F)为空间注意力特征图;m
c
(F)为通道注意力特征图;为叉乘计算;
[0025]F.对在线网络模型进行训练。
[0026]步骤S3所述的采用指数移动平均法共享目标网络和在线网络的参数权重,具体为采用如下步骤进行共享:
[0027]设定表示第k次迭代后的在线网络的权重,表示第k次迭代后的目标网络的权重,为第k

1次迭代后的目标网络的权重;
[0028]采用如下公式计算:式中γ为权重参数且γ∈[0,1);
[0029]通过上述计算,使用目标网络学到的历史参数来更新在线网络的参数;
[0030]同时,采用指数移动平均法来对目标网络和在线网络的参数进行共享。
[0031]步骤S4所述的为目标网络提供课程模块,具体为采用如下步骤提供课程模块:
[0032]a.构建样本课程模块:动态的编码损失函数的权重,来平衡训练中的损失贡献度;采用如下公式作为样本损失加权公式:
[0033][0034]式中α
i
为每个样本损失的权重;p
i
为依据目标网络进行模型评估标签为y=1的预测分数,且测分数,且为第t步迭代时判定为难样本的概率分数;η为超参数;t为迭代次数;
[0035]b.采用如下公式作为难样本的判别公式:
[0036][0037]式中为难样本鉴别分数;
[0038]c.采用如下算式计算迭代t次内识别难样本的预测分数阈值:
[0039][0040]式中为第0步迭代时判定为难样本的概率分数;
[0041]d.构建特征课程模块:通过每个样本上一组空间的权重对局部特征进行编码;通过提高高度鉴别区域和相应的特异性证据特征的权重,从而提高识别精度;采用如下公式计算非线性加权,从而加强课程学习中的卷积特征:
[0042][0043]式中β
i
为空间位置卷积特征的权重;σ为sigmoid激活函数;MLP为多层感知器;UpConv为上采样卷积算子;E
i
为空间注意力特征图;F
i
为从MLP输出的特征映射;为点乘;
[0044]e.设计了1x1卷积层,将多维矩阵转化为单通道;Sigmoid函数用于限定取值的范围在[0,1],上采样算子使得矩阵恢复至与原始图像同样维度,并对位置的每个特征施加一个权重;
[0045]f.将构建的样本课程模块和特征课程模块融入损失函数,进行参数更新。
[0046]步骤S5所述的为在线网络构建监督对抗损失函数,具体为通过为每个样本和相应的特征向量分配自适应的权值来进行在线网络的再平衡训练;采用如下公式进行计算:
[0047][0048][0049]式中ζ为定义的重平衡损失函数;为边界损失函数;α
i
为权重函数;为具有权重函数α
i
和β
i
的样本i上的交叉熵损失;β
i
为权重函数;为网络权重;N为样本数;Ι(y
i
=y
j
)为样本为自身时二值化的指示函数;M(z
本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种数据偏差下的眼底图分类方法,包括如下步骤:S1.获取历史的眼底图像数据,标记并分类;S2.构建目标网络和在线网络的基础模型;S3.采用指数移动平均法共享目标网络和在线网络的参数权重;S4.为目标网络提供课程模块;S5.为在线网络构建监督对抗损失函数;S6.进行第一阶段学习:通过在线网络的对抗学习,使得模型收敛至某个表征空间;S7.进行第二阶段学习:固定步骤S6得到的在线网络的表征参数并传递给目标网络,同时根据双课程样本加权方法,更新目标网络的分类器参数,从而得到最终的网络模型;S8.采用步骤S7得到的最终的网络模型,对实时获取的眼底图像数据进行分类。2.根据权利要求1所述的数据偏差下的眼底图分类方法,其特征在于步骤S1所述的历史的眼底图像数据,标记并分类,具体为获取历史的眼底图像数据,根据设定的要求进行标记分类,同时将数据分为训练集和测试集。3.根据权利要求1或2所述的数据偏差下的眼底图分类方法,其特征在于步骤S2所述的构建目标网络和在线网络的基础模型,具体为采用如下步骤构建模型:A.目标网络包括双分支结构和分类器;双分支结构包括空间注意力网络和通道注意力网络;分类器采用双线性融合分类器;B.空间注意力网络的输入为原始图像,监督标签是注意力图或者分割标签图;网络为注意力Unet结构,具有五层编码和五层解码;从编码最底层输出的向量,为双线性融合提供一条高级语义信息;C.通道注意力网络为SE

Resnet50结构,在每一个网络大模块结构前均加入通道注意力模块,最后删除分类器层,输出一条和空间注意力网络输出的向量维度所匹配的向量;D.分类器为双线性融合分类器,对于空间注意力网络和通道注意力网络输出的卷积,操作后得到的增强的鉴别特征计算公式为式中为定义的鉴别特征图,F为来源于卷积层的输入特征图,m(F)为三维注意力特征图,
×
为点乘;E.将步骤B得到的空间注意力网络的输出向量进行转置,然后与步骤C得到的通道注意力网络的通道向量,采用如下公式进行计算,并将计算的融合的结果进行矩阵向量化,并上传至分类器从而得到最终的预测概率:式中m(F)为注意力的特征图,且提供了一个权重给每个局部的像素和通道;sqrt为数学符号开根号;m
s
(F)为空间注意力特征图;m
c
(F)为通道注意力特征图;为叉乘计算;F.对在线网络模型进行训练。4.根据权利要求3所述的数据偏差下的眼底图分类方法,其特征在于步骤S3所述的采用指数移动平均法共享目标网络和在线网络的参数权重,具体为采用如下步骤进行共享:设定表示第k次迭代后的在线网络的权重,表示第k次迭代后的目标网络的权重,
为第k

1次迭代后的目标网络的权重;采用如下公式计算:式中γ为权重参数且γ∈[0,1);通过上述计算,使用目标网络学到的历史参数来更新在线网络的参数;同时,采用指数移动平均法来对目标网络和在线网络的参数进行共享。5.根据权利要求4所述的数据偏差下的眼底图分类方法,其特征在于步骤S4所述的为目标网络提供课程模块,具体为采用如下步骤提供课程模块:a.构建样本课程模块:动态的编码损失函数的权重,来平衡训练中的损失贡献度;采用如下公式作为样本损失加权公式:式中α
i
为每个样本损失的权重;p
i
为依据目标网络进行模型评估标签为y=1的预测分数,且数,且为第t步迭代时判定为难样本的概率分数标准;η为超参数;t为迭代次数;b.采用如下公式作为难样本的判别公式:式中为难样本鉴别分数;c.采用如下算式计算迭代t次内识别难样本的预测分数阈值...

【专利技术属性】
技术研发人员:赵荣昌陈宣霖何博文贾孝良
申请(专利权)人:中南大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1