softmax+交叉熵
Last updated
Was this helpful?
Last updated
Was this helpful?
一般来说,多分类问题指的就是单标签分类问题,即从n个候选类别中选1个目标类别。loss一般是softmax+交叉熵
多标签分类问题,即从n个候选类别中选k个目标类别。这种情况下我们一种朴素的做法是用sigmoid激活,然后变成n个二分类问题,用二分类的交叉熵之和作为loss。
显然,当n≫kn≫k时,这种做法会面临着严重的类别不均衡问题,这时候需要一些平衡策略,比如手动调整正负样本的权重、focal loss等。训练完成之后,还需要根据验证集来进一步确定最优的阈值。
构建组合形式的softmax来作为单标签softmax的推广。在这部分内容中,我们会先假设k是一个固定的常数,然后再讨论一般情况下k的自动确定方案,最后确实能得到一种有效的推广形式。
组合softmax
自动确定阈值
用于多标签分类
这便是我们最终得到的Loss形式了——“softmax + 交叉熵”在多标签分类任务中的自然、简明的推广,它没有类别不均衡现象,因为它不是将多标签分类变成多个二分类问题,而是变成目标类别得分与非目标类别得分的两两比较,并且借助于logsumexp的良好性质,自动平衡了每一项的权重。
Keras下的参考实现
要提示的是,除了标准的多标签分类问题外,还有一些常见的任务形式也可以认为是多标签分类,比如基于0/1标注的序列标注,典型的例子是“半指针-半标注”标注设计。因此,从这个角度看,能被视为多标签分类来测试式(11)的任务就有很多了,苏神也在之前的三元组抽取例子task_relation_extraction.py中尝试了(11),最终能取得跟这里一致的效果。
当然,最后还是要说明一下,虽然理论上式(11)作为多标签分类的损失函数能自动地解决很多问题,但终究是不存在绝对完美、保证有提升的方案,所以当你用它替换掉你原来多标签分类方案时,也不能保证一定会有提升,尤其是当你原来已经通过精调权重等方式处理好类别不平衡问题的情况下,式(11)的收益是非常有限的。毕竟式(11)的初衷,只是让我们在不需要过多调参的的情况下达到大部分的效果。