ML-kNN 多标签k近邻算法

来源:互联网 发布:mac office 彻底关闭 编辑:程序博客网 时间:2024/04/30 03:21

传统kNN


  k近邻算法(k-Nearest Neighbour, KNN)是机器学习中最基础,最简单的常用算法之一。其思想非常直接:如果一个样本在特征空间中的k个最相似(即特征空间中距离最邻近)的样本中的大多数属于某一个类别,则该样本也属于这个类别。 
  如下图的Xu,它最近的邻居中属于ω1的最多,因此他被归类于ω1类。 
这里写图片描述 
  这个思想很容易理解,就是俗话中常说的“近朱者赤,近墨者黑”。在单标签学习中,与一个实例在特征空间中越相近(即距离越近)的实例,他们之间标签相同的可能性就越大。

多标签kNN


  而在多标签问题中,我们仍可根据这个思想推导出多标签学习的kNN算法,即ML-kNN算法。 
  多标签kNN的主要思想是对于每一个新实例(instance),距离它最近的k个实例(特征空间中与它的距离最小的k个实例)可以首先得到,然后得到这些实例的标签集合,之后通过最大后验概率准则来确定新实例的标签集合。


这里给出算法的具体数学计算方法:

变量定义: 
k为取最近邻个数 
Y为所有标签的集合,总标签个数可以定义为n 
l为一个标签,lY 
x为一个实例 
Yx为实例x对应的标签集合,YxY 
y⃗ xx的标记向量,是一个1×n的行向量,它的元素y⃗ x(l)若为1,代表lYx,若为0,则lYx 
N(x)记录xk个最近邻的索引


然后,我们可以得到对应k近邻实例的标签信息:

C⃗ x(l)=aN(x)y⃗ xa(l),lY

这里,C⃗ x是一个1×n的行向量,它的元素C⃗ x(l)指的是对于标签lxk个近邻中有多少个近邻拥有这个标签。

则,对于新的实例t,首先得到其k个近邻索引集合N(t),定义事件Hl1t有标签l,事件Hl0t无标签l,定义事件Elj ( j{0,1,,k})为对于标签lk个近邻中有j个包含这个标签。则基于向量C⃗ t,可以通过最大后验概率准则和贝叶斯准则得到:

y⃗ t(l)=argmaxb{0,1}P(Hlb)P(ElC⃗ t(l)|Hlb)

其中,y⃗ t(l)即为我们要求的结果,代表t实例是否有l标签。


其中P(Hlb)代表t是否有l标签的先验概率,可以用l标签在整个训练集上出现的次数除以标签总次数来求出。 

P(Hl1)=mi=1y⃗ xi(l)m

即样本中拥有l标签的向量数目除以向量总数。


后验概率P(ElC⃗ t(l)|Hlb)计算方法为:

P(ElC⃗ t(l)|Hl1)=c[j]kp=0c[p]

其中c[j]j等于C⃗ t(l),即tk近邻中有标签l的个数。

c[]的意思是:若xnk近邻中有l标签的个数为δ,且xn有标签l,则c[δ]+1n{1,2,,m}

c[j]代表的就是所有向量中,其k近邻有jl标签,且其自身也有l标签的向量的个数。

由此,kp=0c[p]即为这个向量有l标签,其k近邻有0m个有l标签的情况的向量个数总和。

算出P(Hlb)P(ElC⃗ t(l)|Hlb)之后,我们只需看b={0,1}中哪种情况使得这个乘积值最大,若b=1时最大,则向量tl标签,反之则没有。

0 0
原创粉丝点击