前言

  • 场景
  • loss function
  • 参考
  • 前言

    最近在做小目标图像分割任务(医疗方向),往往一幅图像中只有一个或者两个目标,而且目标的像素比例比较小,使网络训练较为困难,一般可能有三种的解决方式:

    1. 选择合适的loss function,对网络进行合理的优化,关注较小的目标。
    2. 改变网络结构,使用attention机制(类别判断作为辅助)。
    3. 与2的根本原理一致,类属attention,即:先检测目标区域,裁剪之后进行分割训练。

    通过使用设计合理的loss function,相比于另两种方式更加简单易行,能够保留图像所有信息的情况下进行网络优化,达到对小目标精确分割的目的。

    场景

    1. 使用U-Net作为基准网络。
    2. 实现使用keras
    3. 小目标图像分割场景,如下图举例。
      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(yilog⁡pi+(1−yi)log⁡(1−pi)) -\frac{1}{N}\sum_{i=1}^{N}(y_i\log p_i + (1-y_i)\log (1-p_i)) 1NNi=1(yilogpi+(1yi)log(1pi))1Ni=1N(yilogpi+(1yi)log(1pi))N1i=1N(yilogpi+(1yi)log(1pi))T(A,B)=AB+αAB+βBAAB
    再抄一遍Dice系数公式:
    D S C ( A , B ) = 2 ∣ A ⋂ B ∣ ∣ A ∣ + ∣ B ∣ DSC(A,B)=2∣A⋂B∣∣A∣+∣B∣ DSC(A,B)=2ABA+B

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐