arXiv:2511.06857cs.CV2025-11AAAI被引 3

提出新方法提升医学图像分割的准确率与多样性,解决预测模糊问题。

Ambiguity-aware Truncated Flow Matching for Ambiguous Medical Image Segmentation

  • 设计分层推理框架,分别在数据分布和样本层面提升准确率与多样性。
  • 用高斯截断表示增强预测保真度和截断分布可靠性,优于采样近似。
  • 引入语义感知流匹配,让多样的分割结果更合理可信,适合临床不确定场景。

在模糊医学图像分割(AMIS)中,同时提升预测准确率与多样性仍面临挑战,源于二者固有的权衡。尽管截断扩散概率模型(TDPM)具有优化潜力,但现有方法存在准确率与多样性纠缠、预测保真度和合理性不足的问题。为此,本文提出模糊感知截断流匹配(ATFM),引入新型推理范式与专用模型组件。首先,提出数据分层推理(Data-Hierarchical Inference),重新定义针对AMIS的推理范式,分别在数据分布层与数据样本层提升准确率与多样性,实现有效解耦。其次,引入高斯截断表示(GTR),将$T_{\text{trunc}}$时刻的截断分布显式建模为高斯分布,而非依赖采样近似,从而增强预测保真度与截断分布可靠性。第三,提出分割流匹配(SFM),通过扩展流匹配(FM)中的语义感知流变换,提升多样预测的合理性。在LIDC与ISIC3数据集上的综合评估表明,ATFM优于当前最优方法,且推理更高效。相比先进方法,其GED与HM-IoU分别提升最高达12%和7.3%。

原文摘要 · Abstract (English)

A simultaneous enhancement of accuracy and diversity of predictions remains a challenge in ambiguous medical image segmentation (AMIS) due to the inherent trade-offs. While truncated diffusion probabilistic models (TDPMs) hold strong potential with a paradigm optimization, existing TDPMs suffer from entangled accuracy and diversity of predictions with insufficient fidelity and plausibility. To address the aforementioned challenges, we propose Ambiguity-aware Truncated Flow Matching (ATFM), which introduces a novel inference paradigm and dedicated model components. Firstly, we propose Data-Hierarchical Inference, a redefinition of AMIS-specific inference paradigm, which enhances accuracy and diversity at data-distribution and data-sample level, respectively, for an effective disentanglement. Secondly, Gaussian Truncation Representation (GTR) is introduced to enhance both fidelity of predictions and reliability of truncation distribution, by explicitly modeling it as a Gaussian distribution at $T_{\text{trunc}}$ instead of using sampling-based approximations. Thirdly, Segmentation Flow Matching (SFM) is proposed to enhance the plausibility of diverse predictions by extending semantic-aware flow transformation in Flow Matching (FM). Comprehensive evaluations on LIDC and ISIC3 datasets demonstrate that ATFM outperforms SOTA methods and simultaneously achieves a more efficient inference. ATFM improves GED and HM-IoU by up to $12\%$ and $7.3\%$ compared to advanced methods.

医学图像分割不确定性建模流匹配生成多样性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。