基于生成式模型的标签标注方法

1.本技术涉及标签数据领域，特别是涉及一种基于生成式模型的标签标注方法。


背景技术：

2.随着数据获取，存储，传输技术的发展，近些年来数据量呈现爆炸性的增长。有效的数据分类和检索方法可以更方便的管理大规模多样化的数据，从中发现有用的信息。值得注意的是，大多数现有的数据多分类的方法，无论是监督学习还是半监督学习，都假设训练数据的标签是完全合适的。本实施例认为在现实中这样的假设是比较难实现的，即使不是不可能实现的，也是不必要的和不切实际的假设。在现实中，多数训练数据的标签其实都是非完美的，存在着不完整标签或者是噪声标签。本实施例主要研究多标签分类中存在的不完整标注问题，并尝试用生成式模型来解决该问题。数据标注是一项既费时又费力的工作，由于人的主观性，不同的关注点，缺乏耐性去标注每一个细节，甚至是由于人的感知而忽略了某些细节，即使努力认真的给数据手工添加标注，一些标签仍然会有缺失。在这种情况下，这些标注的其实是不完整的，那些认为训练集中数据的标注都是完整的假设显然会误导分类学习过程。
3.因此，如何在标签分类中解决不完整标注成为了一个亟待解决的技术问题。
4.上述内容仅用于辅助理解本发明的技术方案，并不代表承认上述内容是现有技术。


技术实现要素：

5.本发明的主要目的在于提供了一种基于生成式模型的标签标注方法，旨在解决现有技术无法解决不完整标注的技术问题。
6.为实现上述目的，本发明提供了一种基于生成式模型的标签标注方法，所述方法包括：
7.获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型；
8.在所述corr-hdp模型中将β截断为k维；
9.获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架；
10.将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率；
11.通过更新后的所有样本的似然概率，加强所有标签和特征之间的关联性以实现标注分类的准确性和完整性。
12.可选地，所述获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型的步骤，包括：
13.获取corr-lda模型，使用折棒理论的记号法以获取标记：β～stick(α)，π～dp(α^π，β)；
14.设定随机变量y～uniform(1,
…
，ni)来关联同一个样本中的特征和标签；
15.给定一个样本i，生成个ni特征b
i,l
，服从以为参数的多项式分布；
16.对于mi个标签中的任一个标签，用y来随机选择ni个特征中的一个特征，以生成这个特征的隐藏变量z为条件生成相应的标签w
i,j
，其中，其中服从以为参数的多项式分布。
17.可选地，所述在所述corr-hdp模型中将β截断为k维的步骤，包括：
18.在所述corr-hdp模型中将β截断为k维，其中π～dp(α
π
，β)近似为π～dirichlet(α
π
β1，
…
，α
π
βk)。
19.可选地，所述获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架的步骤，包括：
20.获取不完整标注训练集d，对于每一个标签li∈l都存在一个集合对pj×
sj，其中pj∈d表示正样本集，sj∈d表示混合样本集，w
i,j
＝1
→ii
∈pj，w
i,j
＝0
→ii
∈sj。
21.可选地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤之前，还包括：
22.在所述训练集中，d于每一个j∈{1，2，
…
，m}都可以表示为一个集合对pj×
sj，因此，设定下列关系式：
23.ii∈pj＝》pr[w
i,j
＝1|bi]＝1
[0024]ii
∈pj≠》pr[w
i,j
＝1|bi]＝0
[0025]
其中
[0026]
可选地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤，包括：
[0027]
将所述半监督学习框架应用至所述corr-hdp模型上，更新每一个ik∈sj(1≤j≤m)的似然概率pr[w
k,j
＝1|bk]，同时对每一个ii∈pj(1≤j≤m)保持pr[w
i,j
＝1|bi]＝1。
[0028]
可选地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤之后，还包括：
[0029]
当某个样本此标签的更新值w
g,j
在sj中所有样本之间属于最小的t％时，将w
g,j
置为零。
[0030]
此外，为实现上述目的，本发明还提出一种基于生成式模型的标签标注装置，所述装置包括：
[0031]
模型构建模块，用于获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型；
[0032]
数据截断模块，用于在所述corr-hdp模型中将β截断为k维；
[0033]
样本获取模块，用于获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架；
[0034]
似然概率模块，用于将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率；
[0035]
标签标注模块，用于通过更新后的所有样本的似然概率，加强所有标签和特征之间的关联性以实现标注分类的准确性和完整性。
[0036]
此外，为实现上述目的，本发明还提出一种计算机设备，所述计算机设备包括：存储器，处理器以及存储在所述存储器上并可在所述处理器上运行的基于生成式模型的标签标注程序，所述基于生成式模型的标签标注程序配置为实现如上文所述的基于生成式模型的标签标注方法。
[0037]
此外，为实现上述目的，本发明还提出一种介质，所述介质上存储有基于生成式模型的标签标注程序，所述基于生成式模型的标签标注程序被处理器执行时实现如上文所述的基于生成式模型的标签标注方法的步骤。
[0038]
本发明根据获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型；在所述corr-hdp模型中将β截断为k维；获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架；将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率；通过更新后的所有样本的似然概率，加强所有标签和特征之间的关联性以实现标注分类的准确性和完整性，既能对训练集中存在的不完整标注样本进行标签补全，又能对新的完全未标注的样本进行准确的标签预测，实现了解决不完整标注的技术效果。
附图说明
[0039]
图1是本发明实施例方案涉及的硬件运行环境的基于生成式模型的标签标注设备的结构示意图；
[0040]
图2为本发明基于生成式模型的标签标注方法第一实施例的流程示意图；
[0041]
图3为本发明基于生成式模型的标签标注方法第一实施例corr-lda和corr-hdp的图模型；
[0042]
图4为本发明基于生成式模型的标签标注方法第一实施例ssc-hdp的图模型。
具体实施方式
[0043]
应当理解，此处所描述的具体实施例仅用以解释本发明，并不用于限定本发明。
[0044]
参照图1，图1为本发明实施例方案涉及的硬件运行环境的基于生成式模型的标签标注设备结构示意图。
[0045]
如图1所示，该基于生成式模型的标签标注设备可以包括：处理器1001，例如中央处理器(central processing unit，cpu)，通信总线1002、用户接口1003，网络接口1004，存储器1005。其中，通信总线1002用于实现这些组件之间的连接通信。用户接口1003可以包括显示屏(display)、输入单元比如键盘(keyboard)，可选用户接口1003还可以包括标准的有线接口、无线接口。网络接口1004可选的可以包括标准的有线接口、无线接口(如无线保真(wireless-fidelity，wi-fi)接口)。存储器1005可以是高速的随机存取存储器(random access memory，ram)存储器，也可以是稳定的非易失性存储器(non-volatile memory，nvm)，例如磁盘存储器。存储器1005可选的还可以是独立于前述处理器1001的存储装置。
[0046]
本领域技术人员可以理解，图1中示出的结构并不构成对基于生成式模型的标签标注设备的限定，可以包括比图示更多或更少的部件，或者组合某些部件，或者不同的部件
布置。
[0047]
如图1所示，作为一种存储介质的存储器1005中可以包括操作系统、数据存储模块、网络通信模块、用户接口模块以及基于生成式模型的标签标注程序。
[0048]
在图1所示的基于生成式模型的标签标注设备中，网络接口1004主要用于与网络服务器进行数据通信；用户接口1003主要用于与用户进行数据交互；本发明基于生成式模型的标签标注设备中的处理器1001、存储器1005可以设置在基于生成式模型的标签标注设备中，所述基于生成式模型的标签标注设备通过处理器1001调用存储器1005中存储的基于生成式模型的标签标注程序，并执行本发明实施例提供的基于生成式模型的标签标注方法。
[0049]
本发明实施例提供了一种基于生成式模型的标签标注方法，参照图2，图2为本发明基于生成式模型的标签标注方法第一实施例的流程示意图。
[0050]
本实施例中，所述基于生成式模型的标签标注方法包括以下步骤：
[0051]
步骤s10：获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型。
[0052]
需要说明的是，本实施例用到的样本主要为图像，本实施例将样本集表示为i，将样本的特征字典表示为v，将标签字典表示为l。本实施例用的形式来表示每个样本点的特征及其所包含的标注。对于每一个样本i
i ii∈i用一个维的向量b
i bi＝(b
i,1
，b
i,2
，
…
，b
i,n
)’做为其特征表示，其中b
i,j
表示特征字典中第j个特征vj∈v在ii中出现的次数。同时，用一个从m维的向量wi＝(w
i,1
，w
i,2
，
…
，w
i,m
)’做为ii的标注表示，其中w
i,j
∈{0,1}表示标签字典中第j个标注字是否在ii中出现。当ii是完全标注或者部分标注时，ii可以被表示为(bi，wi)；ii当人是未标注样本时，ii可以被表示为bi。因此，本实施例将包含有d个样本的不完整标注训练集表示为完整标注训练集表示为同时将包含有u个样本的未标注训练集表注训练集表
[0053]
进一步地，所述获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型的步骤，包括：获取corr-lda模型，使用折棒理论的记号法以获取标记：β～stick(α)，π～dp(α^π，β)；设定随机变量y～uniform(1,
…
，ni)来关联同一个样本中的特征和标签；给定一个样本i，生成个ni特征服从以为参数的多项式分布；对于mi个标签中的任一个标签，用y来随机选择ni个特征中的一个特征，以生成这个特征的隐藏变量z为条件生成相应的标签w
i,j
，其中服从以为参数的多项式分布。
[0054]
可以理解的是，corr-lda和corr-hdp的图模型如图3所示。图左边为corr-lda模型，图右边为corr-hdp模型，两个模型之间的区别在于在corr-hdp中先验β从一个折棒分布(stick-breaking distribution)中得到，而在corr-lda中先验从一个有限的狄利克雷分布(dirichlet distribution)中得到。corr-hdp的优点在于它能根据训练集而动态自适应的调整隐藏变量的数量，而不是将隐藏变量的数量作为一个先验固定下来。
[0055]
在具体实施中，从corr-hdp的图模型中可以看到，样本的特征b和标签w同时和分配变量z相关联，z是从一个以π为参数的多项式分布中得到的。聚类混合变量π是从一个以
为β参数的全局分布g0釆样得到的。折棒(stick-breaking)理论是建立狄利克雷过程(dirichlet process)的一种有效方法，使用折棒理论的记号法，得到以下标记：β～stick(α)，π～dp(α^π，β)。对于观测到的样本，我们设定一个随机变量y～uniform(1,
…
，ni)来关联同一个样本中的特征和标签。给定一个样本i首先生成个ni特征服从以为参数的多项式分布。然后，对于mi个标签中的任一个标签，用y来随机选择ni个特征中的一个特征，以生成这个特征的隐藏变量z为条件生成相应的标签w
i,j
，服从以为参数的多项式分布。以下算法描述了corr-hdp模型的生成过程。
[0056][0057][0058]
步骤s20：在所述corr-hdp模型中将β截断为k维。
[0059]
进一步地，所述在所述corr-hdp模型中将β截断为k维的步骤，包括：在所述corr-hdp模型中将β截断为k维，其中π～dp(α
π
，β)近似为π～dirichlet(α
π
β1，
…
，α
π
βk)。
[0060]
在具体实施中，在实际中为了使参数估计可行，需要将β截断为k维。因此，π～dp(α
π
，β)可认近似为π～dirichlet(α
π
β1，
…
，α
π
βk)。这里需要注意的是，在hdp中将β截断为k维不等同于使用k个固定的主题corr-lda模型。使用截断的狄利克雷过程(dp)作为先验可以根据训练集自适应的选择隐藏变量的数目，这个数目可以是小于等于k的任意值。而使用狄利克雷分布作为先验则将隐藏变量的数目固定为了k个。
[0061]
步骤s30：获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架。
[0062]
进一步地，所述获取正样本和未标注样本并将所述正样本和未标注样本作为训练
集的半监督学习框架的步骤，包括：获取不完整标注训练集d，对于每一个标签li∈l都存在一个集合对pj×
sj，其中pj∈d表示正样本集，sj∈d表示混合样本集，w
i,j
＝1
→ii
∈pj，w
i,j
＝0
→ii
∈sj。
[0063]
步骤s40：将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率。
[0064]
进一步地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤之前，还包括：在所述训练集中，d于每一个j∈{1，2，
…
，m}都可以表示为一个集合对pj×
sj，因此，设定下列关系式：
[0065]ii
∈pj＝》pr[w
i,j
＝1|bi]＝1
[0066]ii
∈pj≠》pr[w
i,j
＝1|bi]＝0
[0067]
其中
[0068]
进一步地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤，包括：将所述半监督学习框架应用至所述corr-hdp模型上，更新每一个ik∈sj(1≤j≤m)的似然概率pr[w
k,j
＝1|bk]，同时对每一个ii∈pj(1≤j≤m)保持pr[w
i,j
＝1|bi]＝1。
[0069]
进一步地，所述将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率的步骤之后，还包括：当某个样本此标签的更新值w
g,j
在sj中所有样本之间属于最小的t％时，将w
g,j
置为零。
[0070]
步骤s50：通过更新后的所有样本的似然概率，加强所有标签和特征之间的关联性以实现标注分类的准确性和完整性。
[0071]
在具体实施中，ssc-hdp的图模型如图4所示。基于理论分析，这里半监督学习的主要思想即是在corr-hdp的结构下更新每一个ik∈sj(1≤j≤m)的似然概率pr[w
k,j
＝1|bk]，同时对每一个ii∈pj(1≤j≤m)保持pr[w
i,j
＝1|bi]＝1。一下算法详述了ssc-hdp模型的算法。通过更新标签的似然概率，ssc-hdp模型不断的加强标签和特征之间的关联性，最终使得标注分类更加准确和完整。可以看出，未标注数据的信息可以被无缝地添加到ssc-hdp算法中。当初始的corr-hdp模型建立了之后，利用该模型的参数来计算未标注数据集u中每一个样本的每一个标签的似然概率，并把这些未标注样本全部归并到混合样本集sj(1≤j≤m)中。未标注数据的信息可以通过混合样本集被利用起来在连续的迭代过程中更新corr-hdp的参数。
[0072][0073][0074]
本实施例根据获取corr-lda模型，并根据折棒理论将所述corr-lda模型构建成corr-hdp模型；在所述corr-hdp模型中将β截断为k维；获取正样本和未标注样本并将所述正样本和未标注样本作为训练集的半监督学习框架；将所述半监督学习框架应用至所述corr-hdp模型上更新所述半监督学习框架中的每一个样本的似然概率；通过更新后的所有样本的似然概率，加强所有标签和特征之间的关联性以实现标注分类的准确性和完整性，既能对训练集中存在的不完整标注样本进行标签补全，又能对新的完全未标注的样本进行准确的标签预测，实现了解决不完整标注的技术效果。
[0075]
此外，本发明实施例还提出一种介质，所述介质上存储有基于生成式模型的标签标注程序，所述基于生成式模型的标签标注程序被处理器执行时实现如上文所述的基于生成式模型的标签标注方法的步骤。
[0076]
本发明基于生成式模型的标签标注装置的实施例或具体实现方式可参照上述各方法实施例，此处不再赘述。
[0077]
需要说明的是，在本文中，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、物品或者系统不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、物品或者系统所固有的要素。在没有更多限制的情况下，由语句“包括一个
……”
限定的要素，并不排除在包括该要素的过程、方法、物品或者系统中还存在另外的相同要素。
[0078]
上述本发明实施例序号仅仅为了描述，不代表实施例的优劣。
[0079]
通过以上的实施方式的描述，本领域的技术人员可以清楚地了解到上述实施例方法可借助软件加必需的通用硬件平台的方式来实现，当然也可以通过硬件，但很多情况下前者是更佳的实施方式。基于这样的理解，本发明的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质(如只读存储器/随机存取存储器、磁碟、光盘)中，包括若干指令用以使得一台终端设备(可以是手机，计算机，服务器，空调器，或者网络设备等)执行本发明各个实施例所述的方法。
[0080]
以上仅为本发明的优选实施例，并非因此限制本发明的专利范围，凡是利用本发明说明书及附图内容所作的等效结构或等效流程变换，或直接或间接运用在其他相关的技术领域，均同理包括在本发明的专利保护范围内。