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

TP-DRSeg:通过显式文本提示辅助的SAM改进糖尿病视网膜病变病灶分割

TP-DRSeg: Improving Diabetic Retinopathy Lesion Segmentation with Explicit Text-Prompts Assisted SAM

Monash-Airdoc Research(莫纳什-鹰瞳研究中心);Monash University(莫纳什大学);Monash Medical AI Group(莫纳什医学人工智能集团);School of Computer Science and Engineering, Sun Yat-sen University(中山大学计算机科学与工程学院)

Wenxue Li, Xinyu Xiong, Peng Xia, Lie Ju, Zongyuan Ge

AI总结:

针对SAM等大基础模型在糖尿病视网膜病变病灶分割中难以识别细微类间差异的问题,提出TP-DRSeg框架,通过显式先验编码器和先验对齐注入器将文本医学先验注入SAM,实现更优的分割效果且训练参数高效。

摘要:

大基础模型(如Segment Anything Model,SAM,即分割一切模型)的最新进展在各类任务中展现出巨大潜力。尽管取得了这些进展,这些模型在专业医学图像分析中仍面临挑战,尤其是在糖尿病视网膜病变(DR)病灶分割中识别细微的类间差异方面。本文提出了一种为文本提示式DR病灶分割定制SAM的新型框架,名为TP-DRSeg。我们的核心思路是利用语言提示将医学先验知识注入仅视觉的分割网络,从而结合不同基础模型的优势,提升分割的可信度。具体而言,为释放视觉语言模型在医学概念识别中的潜力,我们提出了一种显式先验编码器,将隐式医学概念转化为显式先验知识,提供可解释的线索以挖掘与病灶相关的低层特征。此外,我们设计了一个先验对齐注入器,将显式先验注入分割过程,这能够促进多模态特征间的知识共享,并使我们的框架能以参数高效的方式训练。实验结果表明,我们的框架优于其他传统模型和基础模型变体。

英文摘要:

Recent advances in large foundation models, such as the Segment Anything Model (SAM), have demonstrated considerable promise across various tasks. Despite their progress, these models still encounter challenges in specialized medical image analysis, especially in recognizing subtle inter-class differences in Diabetic Retinopathy (DR) lesion segmentation. In this paper, we propose a novel framework that customizes SAM for text-prompted DR lesion segmentation, termed TP-DRSeg. Our core idea involves exploiting language cues to inject medical prior knowledge into the vision-only segmentation network, thereby combining the advantages of different foundation models and enhancing the credibility of segmentation. Specifically, to unleash the potential of vision-language models in the recognition of medical concepts, we propose an explicit prior encoder that transfers implicit medical concepts into explicit prior knowledge, providing explainable clues to excavate low-level features associated with lesions. Furthermore, we design a prior-aligned injector to inject explicit priors into the segmentation process, which can facilitate knowledge sharing across multi-modality features and allow our framework to be trained in a parameter-efficient fashion. Experimental results demonstrate the superiority of our framework over other traditional models and foundation model variants.

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

↑