论文标题

BoxPolyp:使用额外的粗边界盒注释来增强广义息肉分割

BoxPolyp:Boost Generalized Polyp Segmentation Using Extra Coarse Bounding Box Annotations

论文作者

Wei, Jun, Hu, Yiwen, Li, Guanbin, Cui, Shuguang, Zhou, S Kevin, Li, Zhen

论文摘要

准确的息肉分割对于结直肠癌的诊断和治疗非常重要。但是,由于产生准确的面具注释的高成本,现有的息肉分割方法遭受了严重的数据短缺和模型概括受损。相反,粗息息框注释更容易访问。因此,在本文中,我们提出了一个增强的Boxpolyp模型,以完全使用准确的面膜和额外的粗盒注释。实际上,盒子注释用于减轻先前息肉细分模型的过度问题,该模型通过迭代增强分段模型产生细粒息肉区域。为了实现这一目标,首先提出了融合过滤器采样(FFS)模块,以从噪声较少的盒子注释中生成像素伪标签,从而导致大量的性能改善。此外,考虑到同一息肉的外观一致性,设计了图像一致性(IC)损失。这种IC损失明确缩小了两个不同网络提取的特征之间的距离,从而改善了模型的鲁棒性。请注意,我们的BoxPolyp是一种插件型号,可以合并为任何吸引人的主链。对五个具有挑战性的基准的定量和定性实验结果证实,我们提出的模型的表现优于先前的最新方法。

Accurate polyp segmentation is of great importance for colorectal cancer diagnosis and treatment. However, due to the high cost of producing accurate mask annotations, existing polyp segmentation methods suffer from severe data shortage and impaired model generalization. Reversely, coarse polyp bounding box annotations are more accessible. Thus, in this paper, we propose a boosted BoxPolyp model to make full use of both accurate mask and extra coarse box annotations. In practice, box annotations are applied to alleviate the over-fitting issue of previous polyp segmentation models, which generate fine-grained polyp area through the iterative boosted segmentation model. To achieve this goal, a fusion filter sampling (FFS) module is firstly proposed to generate pixel-wise pseudo labels from box annotations with less noise, leading to significant performance improvements. Besides, considering the appearance consistency of the same polyp, an image consistency (IC) loss is designed. Such IC loss explicitly narrows the distance between features extracted by two different networks, which improves the robustness of the model. Note that our BoxPolyp is a plug-and-play model, which can be merged into any appealing backbone. Quantitative and qualitative experimental results on five challenging benchmarks confirm that our proposed model outperforms previous state-of-the-art methods by a large margin.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源