扫盲：如何提升医学图像分割性能-to do list

导读：本文主要讨论了如何改进图像分割项目中的分割性能，包括一般性和具体性的问题解决方案，以及如何通过调整模型参数、改善数据集质量、优化模型架构、调整超参数、增加训练时长、改善图像分辨率和后处理技术等方法来提高分割效果。

图像分割（Image segmentation）是人工智能的无名英雄，它能逐个像素地分解图像，揭示形状、物体和细节，甚至是我们肉眼无法看到的东西。许多计算机视觉项目都采用分割算法对图像中的标签进行分类。

本文目录

基本介绍

我不会对细分的基本原理进行简单介绍，但我会对其进行初步介绍

分割是根据类别将图像分成不同的部分。在一般情况下，分割算法会考虑这个像素是否应该属于哪个类别，并因此给它分配一个类别。大多数情况下，分割分为 3 类： 语义分割 | 实例分割 | 全景分割

关于分割类型可以查看我之前的文章
【添加链接】

你也可以点击下面的链接进行深度学习：
语义分割 实例分割 全景分割

一些分割架构具有特殊的编码器-解码器结构，它们是架构的基本组成部分。编码器负责从输入图像中提取特征并缩小其空间尺寸。解码器的目的是从编码器获得的抽象特征中重建分割输出。在下图中，您可以看到我们的架构。

如果参数设置正确、超调得当，或者有适当的数据，分割通常都能很好地发挥作用。然而，许多分割项目都会遇到这样的问题：即使经过适当的训练，也无法获得正确的 masks。在这种情况下，区域提取也不正确，从而影响了项目的性能。

【添加引用】分割结果可以叫segmentation results, masks

分割性能不理想的一般情况

我们可以通过一些方法来改进输出mask。我们可以考虑以下几点。

1. 数据集质量（Dataset Quality）

• 规模不足

小规模的数据集可能无法提供足够的示例，让模型有效地学习各种物体的形状和边界。

可以查看此博客，了解计算机视觉项目的最小数据量。

• 标注质量

标注过程中不准确或不一致的分割掩码会对模型性能产生负面影响。

• 类别不平衡

如果某些对象类别的实例数量明显少于其他类别，模型可能难以准确地分割这些类别。

2. 模型架构（Model Architecture）

• 复杂性

有些模型结构在捕捉细粒度细节或处理复杂物体形状方面可能不那么有效。

• 超参数
  调整超参数（如学习率、批量大小和优化器设置）可显著影响分割精度

3. 训练过程（Training Process）

• 训练不足
  模型可能需要更多的历时才能完全收敛并学会准确的分割。请查看此博客，了解最佳epoch次数。

例如，如果验证误差开始增加，这可能是过度拟合的迹象。你应该尽可能多地设置epoch次数，并在验证误差开始增加时终止训练。

• 过度拟合

如果模型过度贴合训练数据，则可能无法很好地泛化到未见过的图像上，从而导致不完美的masks。

4. 图像特征（Image Characteristics）

• 图像分辨率低

低分辨率图像会使模型难以准确划分物体边界

• 遮挡
  
• 可变光照

光照条件不一致会影响物体外观，增加分割难度

5. 图像后处理（ Post-Processing）

• 阈值

对预测掩码概率的阈值处理不当，可能会产生伪影或遗漏部分对象

• 平滑

过度平滑masks会模糊边界，降低精确度

6. 总体改进策略

• 增加数据量
• 处理类别不平衡问题
• 优化模型架构
• 调整超参数
• 加长训练时间
• 改进图像分辨率模型或者注意力机制来处理遮挡问题。
• 标准化光照条件
• 微调阈值
• 应用平滑代码改进

本文来源
：https://medium.com/@BH_Chinmay/how-you-can-improve-output-masks-for-segmentation-projects-fedcba338967