Multilabel Neural Networks with Applications to Functional Genomics and Text Categorization
2006 IEEE Transactions on Knowledge and Data Engineering
Min-Ling Zhang and Zhi-Hua Zhou
介绍
本文提出了一种名为BP-MLL的神经网络算法,即多标记学习的反向传播,这是第一个多标签神经网络算法。顾名思义,BP-MLL源自流行的反向传播算法,通过用定义的新误差函数替换其错误函数来捕获多标记学习的特征,也就是说,属于实例的标签的排名应高于不属于该实例的标签。
多标签学习
假设给定一个实例
xi 及其相关的标签集
Yi,一个成功的学习系统倾向于对于属于
Yi 的标签值比不属于
Yi 的标签值大
f(xi,y1)>f(xi,y2) for any y1∈Yi and y2∈/Yi,对应的分类器
h(⋅)可以通过$f ( \cdot , \cdot ) $得出
h(xi)={y∣f(xi,y)>t(xi),y∈Y}, 其中
t(⋅)是一个阈值函数,通常情况下设定为零(因为这里神经网络使用的激活函数是tanh,值域为[-1,1])
BP-MLL
神经网络
在本文中,传统的多层前馈神经网络适用于从多标签示例中学习。前馈网络将神经元排列成层,第一层采用输入,最后一层采用输出。中间层与外部世界没有联系,因此被称为隐藏层。最小化平方和误差函数的最流行的方法是反向传播算法。
实际上,使传统的前馈神经网络从处理单标签示例到多标签示例有两个关键问题,第一个关键是设计一些特定的误差函数,而不是简单的平方和函数,以捕捉多标记学习的特征。其次,必须对经典学习算法进行一些修改,以便最小化新设计的误差函数。
架构
对于数据集,其中每个样本
X=Rd,
Y={1,2,…,Q}
{(x1,Y1),(x2,Y2),…,(xm,Ym)}
定义每个实例
xi∈X 为d维特征向量,
Yi⊆Y是每个实例所对应的标签集合,维度小于等于Q。
xi 做为神经网络输入层,通过三层神经网络得到输出 $c^i, c^i \in \mathbb { R }^ { Q } , -1\le c^i \le1 $(因为激活函数为tanh),传统意义上的误差函数为
E=i=1∑mEi,Ei=j=1∑Q(cji−dji)2
其中
dji为第i个样本的第j个标签,
di∈RQ,dji=+1(j∈Yi),dji=−1(j∈/Yi)。其中定义的误差函数仅考虑单个标签的误差,没有考虑不同标签之间的相关性,因此我们更改误差函数为
E=i=1∑mEi=i=1∑m∣Yi∣∣∣Yi∣∣1(k,l)∈Yi×Yi∑exp(−(cki−cli))
对于单个实例的误差函数:
⎝⎛∣Yi∣∣∣Yi∣∣1(k,l)∈Yi×Yi∑exp(−(cki−cli))⎠⎞
其中
cki−cli衡量神经网络输出中属于
xi(k∈Yi) 的标签和输出与
xi(l∈Yi) 标签的差别。很明显,差异越大,性能越好。此外,如果
cki(属于
xi 的标签上的输出)远小于
cli(不属于
xi 的标签上的输出),则将这种差异的否定反馈送到指数函数以严重惩罚第i个误差项,进行优化。
在本文中,全局误差函数的最小化是通过梯度下降与误差反向传播策略相结合来实现的。
训练和测试
在训练和测试前要移除数据集当中标签全为0或者全为1的情况,因为到标签为全0或者全1时,误差函数当中的正则项
∣Yi∣∣Yi∣1分母为0,公式无法计算。
通过不断迭代训练模型,得到最终的BPMLL神经网络模型,这时对于每个新输入
x 输出为
c(−1≤c≤1),这个
cj(j=1,2,…,Q) 将被用于标签排序,
x 的关联标签集由阈值函数
t(x) 确定,也就是
Y={j∣cj>t(x),j∈Y}.
通常情况下
t(x) 设定为0,然而,在本文中,采用文献A kernel method for multi-labelled classification中使用的阈值学习机制,推广了上述自然解。这里把
t(x)看成是
c(x)的线性函数,
t(x)=wT⋅c(x)+b,其中
c(x)=(c1(x),c2(x),…,cQ(x))。
对于每一个训练样本
(xi,Yi) ,得到
c(xi)=(c1i,c2i,…,cQi),则
t(xi)=argmint(∣{k∣k∈Yi,cki≤t}∣+∣{l∣l∈Yi,cli≥t}∣),通过
c(xi)和t(xi)训练得到线性模型,以便于预测新样本的阈值函数。
代码实现
https://github.com/hinanmu/BPMLL