从loss处理图像分割中类别极度不均衡的状况---keras
前言场景loss function一、Log loss二、WBE Loss三、Focal loss四、Dice loss五、IOU loss六、Tversky loss七、敏感性–特异性 loss八、Generalized Dice loss九、BCE + Dice loss十、Dice + Focal loss十一、Exponential Logarithmic loss参考前言最近在做小目..
前言
- 场景
- loss function
- 参考
-
前言
最近在做小目标图像分割任务(医疗方向),往往一幅图像中只有一个或者两个目标,而且目标的像素比例比较小,使网络训练较为困难,一般可能有三种的解决方式:
- 选择合适的loss function,对网络进行合理的优化,关注较小的目标。
- 改变网络结构,使用attention机制(类别判断作为辅助)。
- 与2的根本原理一致,类属attention,即:先检测目标区域,裁剪之后进行分割训练。
通过使用设计合理的loss function,相比于另两种方式更加简单易行,能够保留图像所有信息的情况下进行网络优化,达到对小目标精确分割的目的。
场景
- 使用U-Net作为基准网络。
- 实现使用keras
- 小目标图像分割场景,如下图举例。
AI Challenger眼底水肿病变区域自动分割,背景占据了很大的一部分
segthor医疗影像器官分割
loss function
一、Log loss
对于二分类而言,对数损失函数如下公式所示: − 1 N ∑ N i = 1 ( y i l o g p i + ( 1 − y i ) l o g ( 1 − p i ) ) − 1 N ∑ i = 1 N ( y i l o g p i + ( 1 − y i ) l o g ( 1 − p i ) ) − 1 N ∑ i = 1 N ( y i log p i + ( 1 − y i ) log ( 1 − p i ) ) −1N∑Ni=1(yilogpi+(1−yi)log(1−pi))−1N∑i=1N(yilogpi+(1−yi)log(1−pi)) -\frac{1}{N}\sum_{i=1}^{N}(y_i\log p_i + (1-y_i)\log (1-p_i)) −1N∑Ni=1(yilogpi+(1−yi)log(1−pi))−1N∑i=1N(yilogpi+(1−yi)log(1−pi))−N1i=1∑N(yilogpi+(1−yi)log(1−pi))T(A,B)=∣A⋂B∣+α∣A−B∣+β∣B−A∣∣A⋂B∣
再抄一遍Dice系数公式:
D S C ( A , B ) = 2 ∣ A ⋂ B ∣ ∣ A ∣ + ∣ B ∣ DSC(A,B)=2∣A⋂B∣∣A∣+∣B∣ DSC(A,B)=2∣A⋂B∣∣A∣+∣B∣
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)