arXiv:2507.19575cs.CVcs.LG2025-07被引 1

用可交换性替代独立同分布,解决医学图像分割中的数据融合难题

Is Exchangeability better than I.I.D to handle Data Distribution Shifts while Pooling Data for Data-scarce Medical image segmentation?

  • 以可交换性为假设,控制深层特征的前景背景差异
  • 在五个数据集上实现顶尖分割性能,尤其在新增超声数据集上表现突出
  • 适合数据稀缺场景下的医学图像分割研究者参考

医学影像数据稀缺是深度学习面临的主要挑战。尽管通过数据池化(整合多源数据)或数据增加(引入新数据集)可提升模型性能,但会引发分布偏移,导致性能下降,即“数据增加困境”。传统i.i.d.假设在多源场景下不成立,而假设数据集间可交换性更具实用性。本文基于因果框架,提出一种控制深度网络各层前景-背景特征差异的方法,改善特征表示,在数据增加场景中尤为关键。该方法在五个数据集(包括自建的新超声数据集)上实现了最先进的分割效果,且在三种模型架构下均生成更精细准确的分割图,优于主流基线。

原文摘要 · Abstract (English)

Data scarcity is a major challenge in medical imaging, particularly for deep learning models. While data pooling (combining datasets from multiple sources) and data addition (adding more data from a new dataset) have been shown to enhance model performance, they are not without complications. Specifically, increasing the size of the training dataset through pooling or addition can induce distributional shifts, negatively affecting downstream model performance, a phenomenon known as the "Data Addition Dilemma". While the traditional i.i.d. assumption may not hold in multi-source contexts, assuming exchangeability across datasets provides a more practical framework for data pooling. In this work, we investigate medical image segmentation under these conditions, drawing insights from causal frameworks to propose a method for controlling foreground-background feature discrepancies across all layers of deep networks. This approach improves feature representations, which are crucial in data-addition scenarios. Our method achieves state-of-the-art segmentation performance on histopathology and ultrasound images across five datasets, including a novel ultrasound dataset that we have curated and contributed. Qualitative results demonstrate more refined and accurate segmentation maps compared to prominent baselines across three model architectures.

医学图像数据融合可交换性分割

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