3 个月前

S2S2:面向医学影像鲁棒语义分割的语义堆叠

S2S2:面向医学影像鲁棒语义分割的语义堆叠

摘要

医学图像分割的鲁棒性与泛化能力常常受到训练数据稀缺性及多样性不足的制约,这与推理阶段所面临的实际变异性形成鲜明对比。尽管传统方法——如特定领域增强、专用网络架构以及定制化训练流程——能够在一定程度上缓解上述问题,但这些方法依赖于领域知识的可用性与可靠性。当领域知识缺失、存在误导性或被误用时,模型性能反而可能下降。针对这一挑战,我们提出一种新颖的、与领域无关的、可附加的、数据驱动的策略,其灵感源自图像去噪中的图像堆叠(image stacking)技术。该方法被称为“语义堆叠”(semantic stacking),在训练过程中通过估计一个去噪后的语义表征,以补充传统的分割损失。该方法不依赖于特定领域的假设,因此在多种图像模态、模型架构及数据增强技术下均具有广泛的适用性。通过大量实验验证,我们证明了该方法在不同条件下的分割性能均显著优于现有方法。相关代码已开源,地址为:https://github.com/ymp5078/Semantic-Stacking。

代码仓库

ymp5078/semantic-stacking
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
medical-image-segmentation-on-automaticTransUNet
Avg DSC: 90.4
medical-image-segmentation-on-cvc-clinicdbFCBFormer
mean Dice: 0.9488
medical-image-segmentation-on-kvasir-segFCBFormer
mean Dice: 0.932
medical-image-segmentation-on-synapse-multiTransUNet
Avg DSC: 81.19

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
S2S2:面向医学影像鲁棒语义分割的语义堆叠 | 论文 | HyperAI超神经