Sep, 2023

MS-UNet-v2: 医学图像分割的自适应降噪方法与小数据训练策略

TL;DR基于医学图像分割任务,我们提出了一种名为 MS-UNet 的新型 U-Net 模型。与 Swin-UNet 和 TransUnet 中使用的单层 U-Net 解码器结构不同,我们专门为 U-Net 设计了基于 Swin Transformer 的多尺度嵌套解码器。我们还提出了一种新颖的边界损失和即插即用微调去噪模块,这不仅有效提高了 MS-UNet 的分割性能,还可以单独应用于其他模型。实验证明,MS-UNet 在更高效的特征学习能力方面有效提升了网络性能,尤其是在少量训练数据的极端情况下,并且所提出的边界损失和去噪模块能显著增强 MS-UNet 的分割性能。