用扩散模型生成多样医学图像原型,提升少样本分割精度。
DPL: Spatial-Conditioned Diffusion Prototype Enhancement for One-Shot Medical Segmentation
- 用扩散过程从少量标注数据生成多样化原型变体
- 在腹部MRI/CT上实现新最佳性能,提升显著
- 适合需要小样本医学图像分割的研究者
单样本医学图像分割面临原型表示困难,因标注数据有限且患者间解剖差异大。传统方法依赖确定性平均,难以捕捉类内多样性。本文提出扩散原型学习(DPL),将原型建模为可学习的概率分布,通过扩散过程从极少标注数据中生成语义一致的多样化原型。核心创新包括:(1) 基于扩散的原型增强模块,通过前后向扩散生成原型变体;(2) 空间感知条件机制,利用原型特征统计的几何属性;(3) 保守融合策略,在保持原型保真度的同时最大化表示多样性。训练与推理使用相同流程,确保一致性。实验在腹部MRI和CT数据集上显著提升性能,达到当前最优水平。
原文摘要 · Abstract (English)
One-shot medical image segmentation faces fundamental challenges in prototype representation due to limited annotated data and significant anatomical variability across patients. Traditional prototype-based methods rely on deterministic averaging of support features, creating brittle representations that fail to capture intra-class diversity essential for robust generalization. This work introduces Diffusion Prototype Learning (DPL), a novel framework that reformulates prototype construction through diffusion-based feature space exploration. DPL models one-shot prototypes as learnable probability distributions, enabling controlled generation of diverse yet semantically coherent prototype variants from minimal labeled data. The framework operates through three core innovations: (1) a diffusion-based prototype enhancement module that transforms single support prototypes into diverse variant sets via forward-reverse diffusion processes, (2) a spatial-aware conditioning mechanism that leverages geometric properties derived from prototype feature statistics, and (3) a conservative fusion strategy that preserves prototype fidelity while maximizing representational diversity. DPL ensures training-inference consistency by using the same diffusion enhancement and fusion pipeline in both phases. This process generates enhanced prototypes that serve as the final representations for similarity calculations, while the diffusion process itself acts as a regularizer. Extensive experiments on abdominal MRI and CT datasets demonstrate significant improvements respectively, establishing new state-of-the-art performance in one-shot medical image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。