基于合成错误增强的医学图像分割标签精修网络/文献速递-基于人工智能的医学影像技术

2026.1.23

本文提出一种新颖的标签精修网络,通过生成逼真的合成结构错误并结合标签外观模拟网络,隐式编码标签结构信息,显著改善了气道和脑血管等树状结构医学图像的分割精度和完整性,尤其在纠正不连续性和缺失末端分支方面表现出色。

Title 题目

01

Label refinement network from synthetic error augmentation for medical image segmentation

基于合成错误增强的医学图像分割标签精修网络

文献速递介绍

02

CNNs在医学图像分割中表现卓越,但大多数方法(包括U-Net)未能充分利用和编码对象的结构信息,常导致如肺部气道等管状结构分割中出现不连续等结构性错误。显式编码全局结构信息难度大。本文提出一个标签精修框架,通过生成合成结构错误并训练网络进行修正,隐式地将结构信息融入CNNs,旨在提升初始分割的准确性和泛化能力。该方法在气道和脑血管分割任务上进行了验证,并与多种现有方法进行了比较和消融研究。

Aastract摘要

02

深度卷积神经网络在图像分割中通常不会显式学习标签结构,可能导致分割结果出现结构性错误,例如气道或血管等树状结构中出现不连续。为解决此问题,本文提出了一种新颖的标签精修方法,通过隐式融入标签结构信息来修正初始分割中的这类错误。该方法包含两个创新部分:一是生成合成结构错误的模型;二是标签外观模拟网络,用于生成外观上与真实初始分割相似的合成错误分割。利用这些带有合成错误的分割和原始图像,训练标签精修网络来纠正错误并改进初始分割。该方法在胸部CT气道分割和脑部3D CT血管造影(CTA)图像脑血管分割两项任务上进行了验证。结果显示,与标准3D U-Net、四种现有标签精修方法以及针对管状结构损失函数训练的U-Net相比,本方法显著提高了性能。当使用额外的未标注数据进行模型训练时,改进效果更为显著。消融研究也证明了所提方法各组件的价值。

Method 方法

03

所提出的方法包括四个步骤:首先,使用基线分割网络(3D U-Net)生成初始分割;其次,基于对常见错误的分析,向真实标签中添加两类合成错误(缺失末端分支和分支不连续),以训练精修网络;再次,利用基于对抗学习的标签外观模拟网络(LASN)减小合成错误分割与真实初始分割之间的外观差异,提高泛化能力;最后,训练标签精修网络,将带有外观增强合成错误的分割或初始分割与原始图像作为输入,以真实标签为参考,学习纠正错误并输出最终精修的分割结果。

Discussion讨论

04

本文提出的标签精修方法通过生成逼真的合成错误样本来训练网络纠正初始分割中的结构错误,并在气道和脑血管分割任务中展现了优越的性能,尤其在提高完整性和纠正连通性错误方面具有临床重要性,有助于早期疾病诊断和生物标志物测量。该方法优于多种现有标签精修技术和专门的损失函数,并且结合半监督学习可进一步提升性能。与现有方法相比,本文方法提供了一种更通用、更强大的方式来利用带有合成错误的训练数据。作者讨论了合成错误和标签外观模拟网络的重要性,以及该方法应用于其他树状结构分割任务的潜力,甚至是非树状结构分割,前提是能识别并模拟常见的错误类型。也探讨了其在区分病理异常与分割错误的可能性。

Conclusion结论

05

本文提出了一种新颖的标签精修方法,该方法通过从合成错误中学习来优化基线分割网络的初始分割。引入的标签外观模拟网络成功减少了合成错误分割与真实初始分割之间的外观差异,从而增强了方法的泛化能力。在针对分支结构的两项分割任务中,与四种现有的标签精修方法以及采用管状结构损失函数训练的U-Net相比,所提出的方法取得了显著更优的分割结果。通过采用半监督学习技术并利用额外的未标注数据进行训练,该方法的分割性能得到了进一步提升。

Results结果

06

本文提出的标签精修方法在气道和脑血管分割任务上均取得了最佳的Dice系数和完整性得分,并显著减少了断裂数量,同时保持了适度的渗漏。这表明该方法能够有效地从合成错误中学习并纠正真实数据中的错误,尤其在处理树状结构时能更完整地检测分支。与现有方法相比,本方法在完整性方面表现更优,且不易出现过分割。消融研究证实了合成错误和标签外观模拟网络对性能提升的关键作用。半监督学习进一步利用未标注数据,显著提高了Dice得分和气道分割的完整性或血管分割的连通性。

Figure

07

图1. 由U-Net训练的气道分割中常见的结构错误。真阳性以黄色显示,假阴性以蓝色显示,假阳性以红色显示。详细视图a-b显示了缺失末端分支的错误,视图c显示了分支中的不连续错误。

图2. 所提出的标签精修方法的示意图。首先,训练一个基础分割网络f1以获得初始分割x。其次,创建与x中错误相似的合成错误分割xs。第三,训练一个标签外观改进网络fa(与判别器D一起)以获得外观增强的合成错误分割xsˆ。最后,训练标签精修网络f2来纠正这些合成错误,输入可以是xsˆ或x与图像I的组合。

图3. 针对气道定义的合成分割错误的示意图。左侧显示随机选择的末端分支的定义,右侧显示非末端分支的定义。bO:分支起点,bE:分支终点,bM:分支中点,mO:掩膜起点,L:掩膜长度。气道分支的被掩膜部分以蓝色显示(包括所选分支及附近其他分支)。

图4. 所提方法不同组件获得的气道分割示例。在详细视图中,真阳性以黄色显示,假阴性以蓝色显示,假阳性以红色显示。

图5. 所提方法超参数(最大合成错误率)对气道和脑血管分割性能的影响。结果显示为三个随机数据分割上Dice和完整性指标的平均性能及标准偏差(误差条)。基线(LR)结果以虚线显示。

相关推荐
shangjian0072 小时前
AI大模型-深度学习-循环神经网络RNN-编码器和解码器
人工智能·rnn·深度学习
lixin5565562 小时前
基于迁移学习的图像分类增强器
java·人工智能·pytorch·python·深度学习·语言模型
落雨盛夏6 小时前
深度学习|李哥考研4图片分类比较详细说明
人工智能·深度学习·分类
就这个丶调调12 小时前
VLLM部署全部参数详解及其作用说明
深度学习·模型部署·vllm·参数配置
轴测君13 小时前
SE Block(Squeeze and Excitation Block)
深度学习·机器学习·计算机视觉
飞Link15 小时前
深度学习里程碑:ResNet(残差网络)从理论到实战全解析
人工智能·python·深度学习
翱翔的苍鹰16 小时前
完整的“RNN + jieba 中文情感分析”项目之一:终极版
人工智能·rnn·深度学习
茶栀(*´I`*)17 小时前
PyTorch实战:CNN实现CIFAR-10图像分类的思路与优化
pytorch·深度学习·cnn