arXiv:2505.21874eess.IVcs.CV2025-05被引 2

针对医学图像分割中的混杂因素,提出因果感知的优化网络

MAMBO-NET: Multi-Causal Aware Modeling Backdoor-Intervention Optimization for Medical Image Segmentation Network

  • 用多高斯分布建模混杂因素,引入因果干预机制
  • 在五个数据集上显著提升分割准确率,有效降低混杂影响
  • 适合关注医学图像分割鲁棒性的研究者与临床应用开发

医学图像分割通常假设图像到分割结果的过程无偏,使用神经网络建立条件概率模型完成任务。然而这一假设忽略了复杂解剖变异和成像模态限制等混杂因素的影响,导致相关性与因果性模糊,分割效果不佳。为此,本文提出多因果感知建模后门干预优化网络(MAMBO-NET),借鉴因果推断思想,采用多高斯分布自建模方式拟合混杂因素,并将因果干预引入分割过程。设计合理的后验概率约束以有效训练混杂因素分布。为使分布能有效指导分割并缓解其负面影响,引入经典后门干预技术,并分析其在分割任务中的可行性。在五个医学图像数据集上进行大量实验,结果表明该方法显著降低了混杂因素影响,提升了分割精度。

原文摘要 · Abstract (English)

Medical image segmentation methods generally assume that the process from medical image to segmentation is unbiased, and use neural networks to establish conditional probability models to complete the segmentation task. This assumption does not consider confusion factors, which can affect medical images, such as complex anatomical variations and imaging modality limitations. Confusion factors obfuscate the relevance and causality of medical image segmentation, leading to unsatisfactory segmentation results. To address this issue, we propose a multi-causal aware modeling backdoor-intervention optimization (MAMBO-NET) network for medical image segmentation. Drawing insights from causal inference, MAMBO-NET utilizes self-modeling with multi-Gaussian distributions to fit the confusion factors and introduce causal intervention into the segmentation process. Moreover, we design appropriate posterior probability constraints to effectively train the distributions of confusion factors. For the distributions to effectively guide the segmentation and mitigate and eliminate the Impact of confusion factors on the segmentation, we introduce classical backdoor intervention techniques and analyze their feasibility in the segmentation task. To evaluate the effectiveness of our approach, we conducted extensive experiments on five medical image datasets. The results demonstrate that our method significantly reduces the influence of confusion factors, leading to enhanced segmentation accuracy.

医学图像因果推理分割优化

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