arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.03709cs.CV

面向分布外息肉分割的原型学习

Prototype Learning for Out-of-Distribution Polyp Segmentation

Northwestern University(西北大学)

Nikhil Kumar Tomar, Debesh Jha, Ulas Bagci

AI总结:

本文提出PrototypeLab模型,通过融合多种光照模式并学习对象原型,在分布外息肉分割中实现Dice≥90%、mIoU≥85%的接近实时性能,优于16种先进分割架构。

摘要:

现有的结肠镜图像息肉分割模型在不同中心的数据集上往往无法提供可靠的分割结果,限制了其适用性。本研究的目标是构建一个稳健且泛化良好的分割模型,名为PrototypeLab,以辅助息肉分割。为实现这一目标,我们将白光成像(WLI)、蓝光成像(BLI)、联动成像(LCI)和柔性光谱成像色彩增强(FICE)等多种光照模式纳入新的分割模型,该模型学习为图像中存在的每一类对象创建原型。这些原型表示对象的特征性属性,例如形状、纹理和颜色。我们的模型被设计为能够在来自多个中心的分布外(OOD)数据集上有效运行。我们首先生成一个粗略掩码,用于学习主要对象类别的原型,然后利用这些原型生成最终的分割掩码。通过使用原型来表示主要类别,我们的方法能够应对医学图像中存在的变异性,并能很好地泛化到新数据,因为原型捕获了数据的底层分布。PrototypeLab提供了一种有前景的解决方案,其Dice系数≥90%,mIoU≥85%,且息肉分割的处理速度接近实时。与16种最先进的图像分割架构相比,它在OOD数据集上取得了优越的性能,有望改善临床结果。代码可在https://github.com/xxxxx/PrototypeLab获取。

英文摘要:

Existing polyp segmentation models from colonoscopy images often fail to provide reliable segmentation results on datasets from different centers, limiting their applicability. Our objective in this study is to create a robust and well-generalized segmentation model named PrototypeLab that can assist in polyp segmentation. To achieve this, we incorporate various lighting modes such as White light imaging (WLI), Blue light imaging (BLI), Linked color imaging (LCI), and Flexible spectral imaging color enhancement (FICE) into our new segmentation model, that learns to create prototypes for each class of object present in the images. These prototypes represent the characteristic features of the objects, such as their shape, texture, color. Our model is designed to perform effectively on out-of-distribution (OOD) datasets from multiple centers. We first generate a coarse mask that is used to learn prototypes for the main object class, which are then employed to generate the final segmentation mask. By using prototypes to represent the main class, our approach handles the variability present in the medical images and generalize well to new data since prototype capture the underlying distribution of the data. PrototypeLab offers a promising solution with a dice coefficient of $\geq$ 90\% and mIoU $\geq$ 85\% with a near real-time processing speed for polyp segmentation. It achieved superior performance on OOD datasets compared to 16 state-of-the-art image segmentation architectures, potentially improving clinical outcomes. Codes are available at https://github.com/xxxxx/PrototypeLab.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑