处理深度学习中数据集不平衡问题方法汇总

本文探讨了数据集不平衡在分类问题中的挑战,如欺诈检测和疾病诊断,并介绍了两种处理方法:数据增强和权值均衡。数据增强包括旋转缩放、RGB通道提取等,而权值均衡则通过调整损失函数中的类别权重,如Focal Loss,来优化少数类别的性能。
摘要由CSDN通过智能技术生成

一. 数据集不平衡带来的问题:

        在一个分类问题中,如果在所有你想要预测的类别里有一个或者多个类别的样本量非常少,那你的数据也许就面临不平衡类别的问题。如:

1.欺诈预测(欺诈的数量远远小于真实交易的数量)

2.自然灾害预测(不好的事情远远小于好的事情)

3.在图像分类中识别恶性肿瘤(训练样本中含有肿瘤的图像远比没有肿瘤的图像少)

用不平衡的数据训练出来的模型一定会导致样本少的种类预测性能很差,甚至无法预测。但请注意,用不平衡的数据训练出来的模型纵使有以上问题,但是一般情况下往往会得到最高的准确率AP(因为大多数的准确率来自于包含了大量样本的类别中,交叉熵的损失即使是在不均衡的数据的时候,也是趋向于得到最高的准确率)。

       所以我们目的是得到最高的准确率的话,情况下,做数据均衡并没什么意义。但是若少数样本的种类如欺诈检测,肿瘤检测等问题来讲,意义重大且必要。

二. 处理不平衡的方法

处理不平衡的方法总的来说可以分为两大类:1. 数据增强,2.损失函数权值均衡

1. 数据增强

kaggle比赛中有这么一个案例:

问题 - 我们在 kaggle 网站上选择「座头鲸识别挑战」,我们期望解决不平衡类别的挑战(理想情况下,所分类的鲸鱼数量少于未分类的鲸类,并且也有少数罕见鲸类我们有的图像数量更少。)

来自 kaggle :「在这场比赛中,你面临着建立一个算法来识别图像中的鲸鱼种类的挑战。您将分析 Happy Whale 数据库中的超过2

  • 14
    点赞
  • 136
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值