提出通用拓扑修复方法,无需重训练即可提升医学图像分割的拓扑正确性。
Universal Topology Refinement for Medical Image Segmentation with Polynomial Feature Synthesis
- 用正交多项式合成多种拓扑错误,训练与模型无关的修复网络。
- 在多个数据集上验证,修复后分割结果拓扑正确性显著提升。
- 可无缝接入现有分割流程,适合医疗影像分析场景。
尽管现有医学图像分割方法在像素级精度上表现优异,但常忽视拓扑正确性,导致分割结果无法用于下游任务。重新训练模型并引入拓扑驱动损失虽可行,但计算成本高且不实用。更优方案是设计一种兼容任意领域分割流水线的通用即插即用拓扑修复方法。然而,直接训练后处理模型往往因对目标分割网络的拓扑错误产生偏差而失败,尤其在小样本数据集下,错误类型受限于标注数据提供的信息。本文通过训练一个模型无关的拓扑修复网络,使用涵盖广泛拓扑错误的合成分割数据进行训练。受Stone-Weierstrass定理启发,采用随机采样的正交多项式基系数生成拓扑扰动掩码,确保表示的完整性和无偏性。实验表明,该方法兼容多种多项式基族,其通用即插即用拓扑修复网络优于现有的基于学习的拓扑驱动方法和后处理方法。此外,将其与学习型模型结合可轻松提升性能,实现零成本增益。
原文摘要 · Abstract (English)
Although existing medical image segmentation methods provide impressive pixel-wise accuracy, they often neglect topological correctness, making their segmentations unusable for many downstream tasks. One option is to retrain such models whilst including a topology-driven loss component. However, this is computationally expensive and often impractical. A better solution would be to have a versatile plug-and-play topology refinement method that is compatible with any domain-specific segmentation pipeline. Directly training a post-processing model to mitigate topological errors often fails as such models tend to be biased towards the topological errors of a target segmentation network. The diversity of these errors is confined to the information provided by a labelled training set, which is especially problematic for small datasets. Our method solves this problem by training a model-agnostic topology refinement network with synthetic segmentations that cover a wide variety of topological errors. Inspired by the Stone-Weierstrass theorem, we synthesize topology-perturbation masks with randomly sampled coefficients of orthogonal polynomial bases, which ensures a complete and unbiased representation. Practically, we verified the efficiency and effectiveness of our methods as being compatible with multiple families of polynomial bases, and show evidence that our universal plug-and-play topology refinement network outperforms both existing topology-driven learning-based and post-processing methods. We also show that combining our method with learning-based models provides an effortless add-on, which can further improve the performance of existing approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。