一种基于半监督变分自编码器的主动众包图像学习方法技术

技术编号:28943439 阅读:16 留言:0更新日期:2021-06-18 21:52
本发明专利技术公开了一种基于半监督变分自编码器的主动众包图像学习方法,具体为:获取图片集,随机抽选一定比例的图像,将其分发给网络上的标注者进行标注,获得众包标记;搭建基于半监督变分自编码器的众包学习网络模型;将数据集输入模型,构造损失函数;基于随机梯度下降,端到端地对模型进行训练;选择预测真实标记的熵最大的图像,查询真实标记;与上一轮迭代所使用的训练集组合,生成新的数据集,返回第三步,直到当前迭代次数到达阈值为止;删去模型重构部分和众包映射层部分,将网络的剩余部分作为分类器。本发明专利技术同时利用众包数据和无标记数据,降低数据的标注成本。并通过引入少量真实标记,来缓解模型对于噪声的过拟合,提高模型的泛化性能。

【技术实现步骤摘要】
一种基于半监督变分自编码器的主动众包图像学习方法
本专利技术涉及一种基于半监督变分自编码器的主动众包图像学习方法,属于图像标注

技术介绍
传统的监督学习需要大量的专家图像,需要比较高的标注成本。为了降低成本,人们提出了多种方法,比如众包学习、半监督学习和主动学习。众包学习的流程是首先发布标注任务在网络平台上,任何人都可以进行这个标注任务,所收集到的数据比专家标注噪声比例更多。为了解决众包中非专家标注者的不确定性,通常会让每张图像由多个标注者进行标注,即重复标注。如何利用这种具有噪声和重复标记的众包数据成为了众包学习的关键点。代表性的方式有从每张图像的多个标签集估计其真实标记的真实标记推断,其中最为简单的是Majority-voting,对每个标注者的可靠度进行建模的DS模型及其变种,使用EM式算法联合学习分类器模型和标注者噪声模型,使用预测模型来提高数据质量,与主动学习结合等。其中,尽管众包学习中的标签聚合方法可以与任何类型的机器学习算法结合使用,但与同样联合学习分类器本身的方法相比,它们是次优的。半监督学习通过引入大量的无标记数据,这些无标记数据和有标记数据是同分布的,通过无标记数据来对模型进行约束,缓解模型的过拟合,提高它的性能同时降低对专家标记数量的需求。不过,将众包学习和半监督学习结合起来的工作都没有考虑过引入少量的额外监督信息来缓解模型对于众包数据中噪声的拟合。主动学习则是通过某种指标或策略选择出无标记数据中最具有信息量的图像,之后询问专家这些图像的标记。将众包学习和主动学习结合起来的工作均聚焦于询问图像的众包标注而非专家标注。但是在实际情况中,少量图像的真实标记不会增加太多的标记成本,但却很有可能较大地提高模型的泛化性能。
技术实现思路
本专利技术所要解决的技术问题是:提供一种基于半监督变分自编码器的主动众包图像学习方法,同时利用众包数据和无标记数据,降低数据的标注成本。并通过引入少量真实标记,来缓解模型对于噪声的拟合,提高模型的泛化性能。本专利技术为解决上述技术问题采用以下技术方案:一种基于半监督变分自编码器的主动众包图像学习方法,包括如下步骤:步骤1,获取图像集X,将图像集X按照预设比例随机划分为第一图像集和第二图像集,将第一图像集分发给网络上的标注者进行标注,获得相应的众包标记;步骤2,搭建半监督众包学习网络模型,所述半监督众包学习网络模型包括半监督变分自编码器、众包映射层和主动学习框架;步骤3,将第一图像集、第二图像集、众包标记、第三图像集和真实标记集作为训练集,输入半监督众包学习网络模型,基于训练集和网络参数构造变分下界损失函数;第一次迭代时,和均为空集;步骤4,基于随机梯度下降方法,端到端地对半监督众包学习网络模型进行训练,得到训练好的半监督众包学习网络模型;步骤5,利用训练好的半监督众包学习网络模型对第二图像集中每幅图像的真实标记概率进行预测,从第二图像集中选择真实标记概率预测值的熵最大的图像,获取图像的真实标记;步骤6,将图像加入第三图像集,将真实标记加入真实标记集,返回步骤3,直到当前迭代次数达到阈值为止,得到最终的半监督众包学习网络模型;步骤7,对最终的半监督众包学习网络模型,删掉重构部分和众包映射层部分,将剩余部分作为众包图像分类器;步骤8,利用众包图像分类器对待标注图像进行标注,得到众包学习标记。作为本专利技术的一种优选方案,步骤2所述众包映射层,具体形式如下:其中,表示第r个标注者的众包标记,表示第r个众包映射层,y表示真实标记,,R为标注者的数量。作为本专利技术的一种优选方案,步骤3所述变分下界损失函数的构造方法为:对于,其变分下界构造过程如下所示:其中,表示第一图像集中的图像,表示的真实标记,表示隐变量,表示与的似然函数,表示期望,表示的先验分布,表示的先验分布,表示的后验分布,表示对的近似分布,表示和的损失函数,表示分布参数,为变分参数;对于,其变分下界构造过程如下所示:其中,表示第二图像集的图像,表示关于的似然函数,表示的后验分布,表示近似的,是对的近似分布,而是关于的先验联合分布,表示关于和的损失函数,表示的熵,表示关于的损失函数;最终的变分下界损失函数为:当第一次迭代时,损失函数的第四项为0;其中,表示变分下界损失函数,均表示超参数,表示第一图像集的分布,表示第二图像集的分布,表示第三图像集的分布,分别表示第三图像集的单张图像和其真实标记,表示第r个众包映射层,y表示真实标记,表示由经过众包映射层反向映射得到的目标标记,表示代入了的。作为本专利技术的一种优选方案,步骤4所述端到端地对半监督众包学习网络模型进行训练所采用的优化器为SGD或Adam。作为本专利技术的一种优选方案,所述步骤5的具体过程如下:1)利用训练好的半监督众包学习网络模型对第二图像集中每幅图像的真实标记概率进行预测,得到真实标记概率预测值;2)计算的熵;3)从第二图像集中挑选最大图像;4)获取图像对应的真实标记。本专利技术采用以上技术方案与现有技术相比,具有以下技术效果:本专利技术结合半监督学习、众包学习以及主动学习的优势,使用半监督变分自编码器完成众包数据集共享表示学习,使用众包映射层完成真实标记层到众包标记层的映射,使用主动挑选不断迭代的方式引入少量真实标记,实现了端到端训练,提高了图像估计的效果和效率。附图说明图1是本专利技术一种基于半监督变分自编码器的主动众包图像学习方法的流程图。图2是本专利技术模型的结构图。图3是本专利技术迭代挑选的具体流程图。具体实施方式下面详细描述本专利技术的实施方式,所述实施方式的示例在附图中示出。下面通过参考附图描述的实施方式是示例性的,仅用于解释本专利技术,而不能解释为对本专利技术的限制。如图1所示,展示了基于半监督变分自编码器与主动学习的众包图像学习方法的具体流程,包括以下步骤:步骤1:构造图像众包数据集。获取图像后,随机抽选t比例的图像,将它们记为,之后通过众包平台,如AmazonMechanicalTurk和Crowdflower等,分发给网络上的标注者进行预标注,得到标注。剩余的没有被标注的部分我们记作。步骤2:搭建半监督众包学习网络模型,图2是模型的结构图。在步骤2中,我们所搭建的半监督主动众包学习网络由一个半监督变分自编码器、一个映射层与一套迭代的主动学习框架组成。首先,我们将表示整个数据集表示为,其中,为单张图片,为的真实标记,为对应的隐变量,n是样本数量。在半监督的设定下,只有一部分有着对应的,其中,,而。其中是有标记图像集,我们设之为本文档来自技高网
...

【技术保护点】
1.一种基于半监督变分自编码器的主动众包图像学习方法,其特征在于,包括如下步骤:/n步骤1,获取图像集

【技术特征摘要】
1.一种基于半监督变分自编码器的主动众包图像学习方法,其特征在于,包括如下步骤:
步骤1,获取图像集X,将图像集X按照预设比例随机划分为第一图像集和第二
图像集,将第一图像集分发给网络上的标注者进行标注,获得相应的众包
标记;
步骤2,搭建半监督众包学习网络模型,所述半监督众包学习网络模型包括半监督变分自编码器、众包映射层和主动学习框架;
步骤3,将第一图像集、第二图像集、众包标记、第三图像集和真实标记集作为训练集,输入半监督众包学习网络模型,基于训练集和网络参数
构造变分下界损失函数;第一次迭代时,和均为空集;
步骤4,基于随机梯度下降方法,端到端地对半监督众包学习网络模型进行训练,得到训练好的半监督众包学习网络模型;
步骤5,利用训练好的半监督众包学习网络模型对第二图像集中每幅图像的
真实标记概率进行预测,从第二图像集中选择真实标记概率预测值的熵最大的图
像,获取图像的真实标记;
步骤6,将图像加入第三图像集,将真实标记加入真实标记集,返回步骤
3,直到当前迭代次数达到阈值为止,得到最终的半监督众包学习网络模型;
步骤7,对最终的半监督众包学习网络模型,删掉重构部分和众包映射层部分,将剩余部分作为众包图像分类器;
步骤8,利用众包图像分类器对待标注图像进行标注,得到众包学习标记。


2.根据权利要求1所述基于半监督变分自编码器的主动众包图像学习方法,其特征在于,步骤2所述众包映射层,具体形式如下:



其中,表示第r个标注者的众包标记,表示第r个众包映射层,y表示真实
标记,,R为标注者的数量。


3.根据权利要求1所述基于半监督变分自编码器的主动众包图像学习方法,其特征在于,步骤3所...

【专利技术属性】
技术研发人员:李绍园侍野黄圣君
申请(专利权)人:南京航空航天大学
类型:发明
国别省市:江苏;32

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

1