针对数据不平衡情况,提出一种能让模型快速收敛的损失函数。其主要思想就是添加一个一个抑制参数,使得在训练计算损失时,抑制容易分类的样本的权重,从而让模型能够重点学习辨别那些不易区分的样本。
1、Cross entropy
2、Balanced Cross Entropy
添加平衡参数alpha,优化正负样本数据不平衡情况。
3、Focal Loss
添加抑制参数,重点学习不易分类样本。
Focal 损失,其中gamma大于等于0
实际应用中,使用的是带平衡参数alpha的Focal损失,如下:
平衡Focal 损失