arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.07085cs.CV

CAT:协调解剖-文本提示用于多器官与肿瘤分割

CAT: Coordinating Anatomical-Textual Prompts for Multi-Organ and Tumor Segmentation

Qing Yuan Research Institute, Shanghai Jiao Tong University(上海交通大学清源研究院);Shanghai AI Laboratory(上海人工智能实验室)

Zhongzhen Huang, Yankai Jiang, Rongzhao Zhang, Shaoting Zhang, Xiaofan Zhang

AI总结:

针对医学图像分割中单一提示难以应对肿瘤等异常的问题,提出协调解剖与文本双提示的CAT模型,通过ShareRefiner协同两类提示,在多数据集上实现多器官与肿瘤分割的优越性能。

摘要:

医学成像领域现有的可提示分割方法主要采用文本或视觉提示来分割相关目标,但在处理医学图像中的异常(如肿瘤)时往往效果不佳,这类异常在形状、大小和外观上可能存在极大差异。考虑到医学场景的复杂性以及文本或视觉提示的局限性,我们提出一种新颖的双提示模式,利用视觉与文本提示的互补优势来分割各类器官和肿瘤。具体而言,我们推出CAT这一创新模型,它能协调从3D裁剪图像中提取的解剖提示与经医学领域知识增强的文本提示。该模型架构采用基于通用查询的设计,其中提示查询为掩码预测提供分割查询的支撑。为在统一框架内协同两类提示,我们实现了ShareRefiner模块,该模块在解耦两类提示的同时,对分割查询与提示查询均进行优化。在由10个公开CT数据集组成的联合体上训练后,CAT在多项分割任务中展现出优越性能。在专门的内部数据集上的进一步验证显示,其具备跨多个癌症阶段分割肿瘤的卓越能力。该方法证实,协调多模态提示是解决医学领域复杂场景的一条极具前景的路径。

英文摘要:

Existing promptable segmentation methods in the medical imaging field primarily consider either textual or visual prompts to segment relevant objects, yet they often fall short when addressing anomalies in medical images, like tumors, which may vary greatly in shape, size, and appearance. Recognizing the complexity of medical scenarios and the limitations of textual or visual prompts, we propose a novel dual-prompt schema that leverages the complementary strengths of visual and textual prompts for segmenting various organs and tumors. Specifically, we introduce CAT, an innovative model that Coordinates Anatomical prompts derived from 3D cropped images with Textual prompts enriched by medical domain knowledge. The model architecture adopts a general query-based design, where prompt queries facilitate segmentation queries for mask prediction. To synergize two types of prompts within a unified framework, we implement a ShareRefiner, which refines both segmentation and prompt queries while disentangling the two types of prompts. Trained on a consortium of 10 public CT datasets, CAT demonstrates superior performance in multiple segmentation tasks. Further validation on a specialized in-house dataset reveals the remarkable capacity of segmenting tumors across multiple cancer stages. This approach confirms that coordinating multimodal prompts is a promising avenue for addressing complex scenarios in the medical domain.

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

↑