本发明涉及机器学习领域,更具体的说是涉及一种基于主题模型的半监督哈希算法。
背景技术:
在大规模数据集检索中,哈希算法是解决效率问题的有效途径。给定一组数据的特征向量
目前,关于半监督哈希的典型代表是核化监督哈希(kernel-basedsupervisedhashing,ksh)。ksh由两部分组成,即核方法与监督训练。核方法可以处理实际问题中近似线性可分的数据。监督训练通过优化算法使哈希码的相似性等效于数据标签的相似性,提升检索准确率。但是ksh仅仅使用数据集中有标签的数据,忽略了无标签数据的分布规律;ksh中核方法在其目标(特征线性可分性增强)中效果并不令人满意;ksh中核方法引入了随机因素,导致稳定性较差。
因此,如何提供一种不但能够保证检索的准确率,而且能够保证算法稳定性的基于主题模型的半监督哈希算法是本领域技术人员亟需解决的问题。
技术实现要素:
有鉴于此,本发明提供了一种基于主题模型的半监督哈希算法,将ksh中核方法替换为主题模型,对单词特征进行哈希编码;通过主题模型利用数据集中所有数据的单词特征,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述,从而获得更准确的哈希码;通过替换核方法,减少随机因素,使得模型更加稳定。
为了实现上述目的,本发明提供如下技术方案:
一种基于主题模型的半监督哈希算法,其特征在于,具体包括以下步骤:
s1:根据所有数据训练主题模型,由单词特征得到主题特征,并对所述主题模型进行求解;
s2:通过半监督训练将主题特征转化为哈希码;
s3:对未知数据进行哈希编码并检索。
本发明的技术效果:将ksh中核方法替换为主题模型,对单词特征进行哈希编码,其中主题模型为潜在狄利克雷分配(latentdirichiletallocation,lda);通过主题模型利用数据集中所有数据的单词特征,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述,从而获得更准确的哈希码;通过替换核方法,减少随机因素,使得模型更加稳定。
优选的,在上述一种基于主题模型的半监督哈希算法中,在s1中具体步骤包括:
s11:设主题z为隐含变量,表示为主题库中的一个索引,z∈{1,…,k},其中k为主题总类数;
s12:定义超参数α和β;
s13::定义数据集w={w1,…,wm},wi=[wi,1,…,wi,ni],其中wij∈{1,…,v},m为数据数,ni为第i个数据中单词个数,v为单词总类数;
s14:通过主题模型求解每个数据的主题概率分布特征θm=[θm1,…,θmk](m=1,…,m)以及主题-单词概率分布
本发明的技术效果:通过主题模型利用数据集中所有数据的单词特征,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述。
优选的,在上述一种基于主题模型的半监督哈希算法中,对所述超参数进行优化,选择一组最优超参数。
本发明的技术效果:对超参数进行优化,选择一组最优超参数,以提高学习的性能和效果。
优选的,在上述一种基于主题模型的半监督哈希算法中,在s14中,求解所述主题模型采用吉布斯采样法具体包括:
s141:给每个特征的每个单词w随机分配一个主题z;
s142:然后每个单词的主题通过下式重新采样并更新;
其中
s143:令i在(1,1)到(m,nm)之间循环变化,反复执行上述重采样与更新过程,直到收敛;
s144:通过统计得到主题-单词的共现矩阵[n(k,v)]k×v和数据-主题的共现矩阵[n(m,k)]m×k;
s145:通过下式计算得到θmk和
本发明的技术效果:通过主题模型利用数据集中所有数据的单词特征,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述。
优选的,在上述一种基于主题模型的半监督哈希算法中,在s2中具体步骤包括:
s21:设主题特征数据集为x={x(1),…,x(n)},其中
其中,
s22:从x中随机选择l个带标签的样本,其中l远远小于n,构建标签矩阵s;根据第i个样本与第j个样本标签是否相同对sij与sji赋予1或-1;
s23:不失一般性地假设带标签样本集为xl={x(1),…,x(l)},定义
s24:定义函数σ(x)=2/(1+exp(-x))-1,使用梯度下降法优化以下目标函数
s25:最终解得a并根据s21得到所有数据的哈希码。
本发明的技术效果:为了保持检索精度,保证生成的每位编码都有均衡且互相独立的分布,以保证每位编码的最大信息以及所有位之间的最小冗余。
优选的,在上述一种基于主题模型的半监督哈希算法中,在s3中具体步骤包括:
s31:对于未知数据的单词特征wq=[wq,1,…,wq,nq],wqj∈{1,…,v},根据s1和s2中训练的模型依次求其主题特征与哈希码;
s32:将
本发明的技术效果:针对未知数据采用同样的方法,实现模型对数据分布更准确的描述,从而获得更准确的哈希码,进一步减少随机因素,使得模型更加稳定。
经由上述的技术方案可知,与现有技术相比,本发明公开提供了一种基于主题模型的半监督哈希算法,使用主题模型将单词特征转化为主题特征,然后对主题特征进行半监督训练,实现了半监督条件下准确、稳定并能涵盖所有数据,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述,从而获得更准确的哈希码;通过替换核方法,减少随机因素,使得模型更加稳定。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据提供的附图获得其他的附图。
图1附图为本发明的流程图;
图2附图为本发明吉布斯采样的流程图;
图3附图为本发明转化哈希码的流程图。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
本发明实施例公开了一种基于主题模型的半监督哈希算法,使用主题模型将单词特征转化为主题特征,然后对主题特征进行半监督训练,实现了半监督条件下准确、稳定并能涵盖所有数据,增强了模型对数据的泛化能力,使模型对数据分布有更准确的描述,从而获得更准确的哈希码;通过替换核方法,减少随机因素,使得模型更加稳定。
实施例1
请参阅附图1-3为本发明提供了一种基于主题模型的半监督哈希算法,具体包括以下步骤:
s1:根据所有数据训练主题模型,由单词特征得到主题特征,并对所述主题模型进行求解;
s2:通过半监督训练将主题特征转化为哈希码;
s3:对未知数据进行哈希编码并检索。
为了优化上述技术方案,在s1中具体步骤包括:
s11:设主题z为隐含变量,表示为主题库中的一个索引,z∈{1,…,k},其中k为主题总类数;
s12:定义超参数α和β;
s13:定义数据集w={w1,…,wm},wi=[wi,1,…,wi,ni],其中wij∈{1,…,v},m为数据数,ni为第i个数据中单词个数,v为单词总类数;
s14:通过主题模型求解每个数据的主题概率分布特征θm=[θm1,…,θmk]
(m=1,…,m)以及主题-单词概率分布
为了优化上述技术方案,在s12中定义所述超参数,对超参数进行优化,选择一组最优超参数。
为了优化上述技术方案,在s14中,求解所述主题模型采用吉布斯采样法,且所述吉布斯采样法具体包括:
s141:给每个特征的每个单词w随机分配一个主题z;
s142:每个单词的主题z通过下式重新采样并更新;
其中
s143:令i在(1,1)到(m,nm)之间循环变化,反复执行上述重采样与更新过程,直到收敛;
s144:通过统计得到主题-单词的共现矩阵[n(k,v)]k×v和数据-主题的共现矩阵[n(m,k)]m×k;
s145:通过下式计算得到θmk和
为了优化上述技术方案,在s2中具体步骤包括:
s21:设主题特征数据集为x={x(1),…,x(n)},其中
其中,
s22:从x中随机选择l个带标签的样本,其中l远远小于n,构建标签矩阵s;根据第i个样本与第j个样本标签是否相同对sij与sji赋予1或-1;
s23:不失一般性地假设带标签样本集为xl={x(1),…,x(l)},定义
s24:定义函数σ(x)=2/(1+exp(-x))-1,使用梯度下降法优化以下目标函数
s25:最终解得a并根据s21得到所有数据的哈希码。
为了优化上述技术方案,在s3中具体步骤包括:
s31:对于未知数据的单词特征wq=[wq,1,…,wq,nq],wqj∈{1,…,v},根据s1和s2中训练的模型依次求其主题特征与哈希码;
s32:将
实施例2
一种基于主题模型的半监督哈希算法,其中主题模型采用潜在语义分析,是用于知识获取和展示的计算理论和方法,它使用统计计算的方法对大量的文本集进行分析,从而提取出词与词之间潜在的语义结构,并用这种潜在的语义结构来表示词和文本,达到消除词之间的相关性和简化文本向量实现降维的目的。潜在语义分析的基本观点是:把高维的向量空间模型(vsm)表示中的文档映射到低维的潜在语义空间中。这个映射是通过对项/文档矩阵的奇异值分解(svd)来实现的。
求解主题模型的方法采用期望最大化,首先初始化分布函数;估计未知参数的期望值,给出当前的参数估计;重新估计分布函数,以使得数据的似然性最大,给出未知变量的期望估计。
本说明书中各个实施例采用递进的方式描述,每个实施例重点说明的都是与其他实施例的不同之处,各个实施例之间相同相似部分互相参见即可。对于实施例公开的装置而言,由于其与实施例公开的方法相对应,所以描述的比较简单,相关之处参见方法部分说明即可。
对所公开的实施例的上述说明,使本领域专业技术人员能够实现或使用本发明。对这些实施例的多种修改对本领域的专业技术人员来说将是显而易见的,本文中所定义的一般原理可以在不脱离本发明的精神或范围的情况下,在其它实施例中实现。因此,本发明将不会被限制于本文所示的这些实施例,而是要符合与本文所公开的原理和新颖特点相一致的最宽的范围。