深度学习【17】物体检测:Focal Loss 反向求导及darknet上的实现

本文介绍了Focal Loss如何解决不平衡分类问题,详细推导了在softmax函数下Focal Loss的反向求导,并记录了在darknet框架中的具体实现。调整alpha参数至0.5,经过训练,平均精度map提升了2%。
摘要由CSDN通过智能技术生成

Focal Loss 反向求导及darknet上的实现

Focal Loss 可以解决不平衡分类问题,是在交叉熵损失函数上的扩展。详见,论文:Focal Loss for Dense Object Detection。
该文,主要推导FL在softmax函数下的求导,以及记录在darknet上的实现。

FL损失函数如下:

FL(pt)=α(1pt)γlog(pt) F L ( p t ) = − α ( 1 − p t ) γ l o g ( p t )

其中 pt p t 为:
pt={ p, 1p,if y==1 otherwise  p t = { p , if  y == 1     1 − p , otherwise 

softmax公式:

pi=exiek p i = e x i ∑ e k

我们的目标得到x对FL的导数:

dFLdxi=dFLdpidpidxi d F L d x i = d F L d p i ∗ d p i d x i

又,

dFLdpt
评论 29
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值