将医学病灶分割中的背景细分为多个类别,显著提升小病灶识别效果。
BackSplit: The Importance of Sub-dividing the Background in Biomedical Lesion Segmentation
- 将背景拆分为多个细粒度类别,增强模型对解剖结构的理解。
- 在多个数据集上,小病灶分割性能普遍提升,最高增益达12.7%。
- 无需增加推理成本,适用于自动或人工生成的辅助标签。
医学图像中小病灶的分割仍极具挑战性。现有方法多聚焦于改进网络架构、损失函数或数据增强,以及扩充标注数据。本文提出不同视角:问题部分源于背景建模方式。传统方法将所有非病灶像素合并为单一“背景”类,忽视了病灶出现时复杂的解剖上下文。实际上,背景由组织、器官等多样结构组成,如今可通过手动标注或现有分割模型自动推断。本文主张采用细粒度的背景划分策略(称为BackSplit),在不增加推理开销的前提下,显著提升性能。从信息论角度,证明BackSplit提升了期望费舍尔信息量,带来更紧的渐近界和更稳定的优化。在多个数据集与架构上的实验证明,即使使用预训练模型自动生成辅助标签,BackSplit仍能持续提升小病灶分割表现。此外,交互式分割框架生成的辅助标签同样有效,验证了该方法的鲁棒性、简洁性与广泛适用性。
原文摘要 · Abstract (English)
Segmenting small lesions in medical images remains notoriously difficult. Most prior work tackles this challenge by either designing better architectures, loss functions, or data augmentation schemes; and collecting more labeled data. We take a different view, arguing that part of the problem lies in how the background is modeled. Common lesion segmentation collapses all non-lesion pixels into a single "background" class, ignoring the rich anatomical context in which lesions appear. In reality, the background is highly heterogeneous-composed of tissues, organs, and other structures that can now be labeled manually or inferred automatically using existing segmentation models. In this paper, we argue that training with fine-grained labels that sub-divide the background class, which we call BackSplit, is a simple yet powerful paradigm that can offer a significant performance boost without increasing inference costs. From an information theoretic standpoint, we prove that BackSplit increases the expected Fisher Information relative to conventional binary training, leading to tighter asymptotic bounds and more stable optimization. With extensive experiments across multiple datasets and architectures, we empirically show that BackSplit consistently boosts small-lesion segmentation performance, even when auxiliary labels are generated automatically using pretrained segmentation models. Additionally, we demonstrate that auxiliary labels derived from interactive segmentation frameworks exhibit the same beneficial effect, demonstrating its robustness, simplicity, and broad applicability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。