【深度学习实战（22）】解决样本类别不均衡问题之Focal Loss

BILLY BILLY

已于 2024-07-04 12:38:38 修改

阅读量1.2k

点赞数 5

分类专栏：深度学习基础文章标签：深度学习人工智能

于 2024-04-23 16:17:15 首次发布

本文链接：https://blog.csdn.net/m0_51579041/article/details/138129323

版权

深度学习基础专栏收录该内容

53 篇文章 9 订阅

订阅专栏

一、Focal Loss公式介绍

Focal loss是何恺明大神提出的一种新的loss计算方案。其具有两个重要的特点。

1、控制正负样本的权重
2、控制容易分类和难分类样本的权重
论文：

二分类问题交叉熵损失

公式：
在这里插入图片描述
我们可以利用如下Pt简化交叉熵loss。

此时：

代码：

BCE_loss = F.binary_cross_entropy_with_logits(inputs, targets, reduction='none')

正负样本平衡项

-想要降低负样本的影响，可以在常规的损失函数前增加一个系数αt。与Pt类似，当label=1的时候，αt=α；当label=otherwise的时候，αt=1 - α，a的范围也是0到1。此时我们便可以通过设置α实现控制正负样本对loss的贡献。
公式：
在这里插入图片描述
其中：

分解开就是：

难易样本平衡项

样本属于某个类，且预测结果中该类的概率越大，其越容易分类，在二分类问题中，正样本的标签为1，负样本的标签为0，p代表样本为1类的概率。

对于正样本而言，1-p的值越大，样本越难分类。
对于负样本而言，p的值越大，样本越难分类。

Pt的定义如下
在这里插入图片描述
所以利用1-Pt就可以计算出每个样本属于容易分类或者难分类。
具体实现方式如下。

两种权重控制方法合并，就得到了Focal Loss

通过如下公式就可以实现控制正负样本的权重和控制容易分类和难分类样本的权重。
在这里插入图片描述
分解开就是：

二、Focal Loss代码实现

import torch
import torch.nn as nn
import torch.functional as F

class WeightedFocalLoss(nn.Module):
    "Non weighted version of Focal Loss"    
    def __init__(self, alpha=.25, gamma=2):
            super(WeightedFocalLoss, self).__init__()  
            # --------------#
            #   平衡正负样本系数
            # --------------#      
            self.alpha = torch.tensor([alpha, 1-alpha]).cuda()      
            # --------------#
            #   平衡难易样本系数
            # --------------#   
            self.gamma = gamma
            
    def forward(self, inputs, targets):
            # --------------#
            #   分类交叉熵损失
            # --------------# 
            BCE_loss = F.binary_cross_entropy_with_logits(inputs, targets, reduction='none')   
            # --------------#
            #   标签GT
            # --------------#      
            targets = targets.type(torch.long)     
            # --------------#
            #   计算at
            # --------------#    
            at = self.alpha.gather(0, targets.data.view(-1))   
            # --------------#
            #   计算pt: BEC_loss = -log(pt)  --> pt = torch.exp(-BCE_loss)   
            # --------------#       
            pt = torch.exp(-BCE_loss)   
            # --------------#
            #   计算Focal Loss
            # --------------#       
            F_loss = at*(1-pt)**self.gamma * BCE_loss        
            return F_loss.mean()