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

用于医学图像分割的多尺度渐进式文本提示网络

Multiscale Progressive Text Prompt Network for Medical Image Segmentation

Xianjun Han, Qianqian Chen, Zhaoyang Xie, Xuejun Li, Hongyu Yang

AI总结:

提出多尺度渐进式文本提示网络,通过两阶段对比学习预训练先验提示编码器并结合多尺度特征融合与UpAttention块,以低标注成本实现高精度的医学图像分割。

摘要:

医学图像的准确分割是获取可靠形态学统计数据的关键步骤。然而,为该任务训练深度神经网络需要大量标注数据以确保高精度结果。为解决这一问题,我们提出使用渐进式文本提示作为先验知识来指导分割过程。我们的模型包含两个阶段。第一阶段,我们在自然图像上执行对比学习以预训练强大的先验提示编码器(PPE)。该PPE利用文本先验提示生成多模态特征。第二阶段,将医学图像和文本先验提示送入继承自第一阶段的PPE,以完成下游医学图像分割任务。多尺度特征融合块(MSFF)结合来自PPE的特征,生成多尺度多模态特征。这两种渐进式特征不仅弥合了语义鸿沟,还提高了预测精度。最后,UpAttention块通过融合图像和文本特征来优化预测结果。该设计提供了一种简单且准确的方法,利用多尺度渐进式文本先验提示进行医学图像分割。与仅使用图像相比,我们的模型以低数据标注成本实现了高质量结果。此外,我们的模型不仅在医学图像上具有出色的可靠性和有效性,在自然图像上也表现良好。在不同图像数据集上的实验结果表明,我们的模型对图像分割有效且鲁棒。

英文摘要:

The accurate segmentation of medical images is a crucial step in obtaining reliable morphological statistics. However, training a deep neural network for this task requires a large amount of labeled data to ensure high-accuracy results. To address this issue, we propose using progressive text prompts as prior knowledge to guide the segmentation process. Our model consists of two stages. In the first stage, we perform contrastive learning on natural images to pretrain a powerful prior prompt encoder (PPE). This PPE leverages text prior prompts to generate multimodality features. In the second stage, medical image and text prior prompts are sent into the PPE inherited from the first stage to achieve the downstream medical image segmentation task. A multiscale feature fusion block (MSFF) combines the features from the PPE to produce multiscale multimodality features. These two progressive features not only bridge the semantic gap but also improve prediction accuracy. Finally, an UpAttention block refines the predicted results by merging the image and text features. This design provides a simple and accurate way to leverage multiscale progressive text prior prompts for medical image segmentation. Compared with using only images, our model achieves high-quality results with low data annotation costs. Moreover, our model not only has excellent reliability and validity on medical images but also performs well on natural images. The experimental results on different image datasets demonstrate that our model is effective and robust for image segmentation.

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

↑