通过形状与强度先验知识蒸馏,提升医学图像分割的准确性和泛化能力。
Shape-intensity knowledge distillation for robust medical image segmentation

- 用类平均图像训练教师网络提取形状-强度先验
- 将先验信息通过知识蒸馏注入学生网络,提升分割精度
- 显著增强跨数据集泛化能力,适合医疗影像领域应用
许多医学图像分割方法已取得显著成果,但大多未考虑形状-强度先验信息,可能导致在未见数据集上产生不合理分割结果。本文提出一种新方法,将联合形状-强度先验信息融入分割网络:首先在类别平均训练图像上训练教师网络以提取有价值的信息,再通过知识蒸馏将其转移至具有相同架构的学生网络。该学生网络作为最终分割模型,可有效整合先验信息,获得更准确的分割结果。尽管方法简单,但在五种不同模态的医学图像分割任务上,SIKD在同数据集评估中持续提升多个基线模型(包括MaxStyle和SAMed),并显著增强跨数据集泛化能力。
原文摘要 · Abstract (English)
Many medical image segmentation methods have achieved impressive results. Yet, most existing methods do not take into account the shape-intensity prior information. This may lead to implausible segmentation results, in particular for images of unseen datasets. In this paper, we propose a novel approach to incorporate joint shape-intensity prior information into the segmentation network. Specifically, we first train a segmentation network (regarded as the teacher network) on class-wise averaged training images to extract valuable shape-intensity information, which is then transferred to a student segmentation network with the same network architecture as the teacher via knowledge distillation. In this way, the student network regarded as the final segmentation model can effectively integrate the shape-intensity prior information, yielding more accurate segmentation results. Despite its simplicity, experiments on five medical image segmentation tasks of different modalities demonstrate that the proposed Shape-Intensity Knowledge Distillation (SIKD) consistently improves several baseline models (including recent MaxStyle and SAMed) under intra-dataset evaluation, and significantly improves the cross-dataset generalization ability. The code is available at https://github.com/whdong-whu/SIKD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。