arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.03412cs.CVphysics.med-ph

TG-LMM:通过文本引导的大型多模态模型提升医学图像分割精度

TG-LMM: Enhancing Medical Image Segmentation Accuracy through Text-Guided Large Multi-Modal Model

Sun Yat-sen University(中山大学);National Cancer Center/National Clinical Research Center for Cancer/Cancer Hospital & Shenzhen Hospital, Chinese Academy of Medical Sciences and Peking Union Medical College(中国医学科学院北京协和医学院国家癌症中心/国家肿瘤临床医学研究中心/肿瘤医院&深圳医院);University of Macau(澳门大学)

Yihao Zhao, Enhao Zhong, Cuiyun Yuan, Yang Li, Man Zhao, Chunxia Li, Jun Hu, Chenbin Liu

AI总结:

TG-LMM通过将器官空间位置的专家文本描述与预训练图像-文本编码器融合,减少训练参数并提升医学图像分割精度,在多个数据集上优于MedSAM、SAM和nnUnet。

摘要:

我们提出TG-LMM(文本引导的大型多模态模型),这是一种利用器官文本描述来提升医学图像分割精度的新方法。现有医学图像分割方法面临若干挑战:当前医学自动分割模型未能有效利用先验知识,例如器官位置的描述;以往的文本-视觉模型侧重于识别目标,而非提升分割精度;先前模型尝试利用先验知识提升精度,但未引入预训练模型。为解决这些问题,TG-LMM将先验知识,特别是专家对器官空间位置的描述,整合到分割过程中。我们的模型利用预训练的图像和文本编码器来减少训练参数数量并加速训练过程。此外,我们设计了一种综合的图像-文本信息融合结构,以确保两种模态数据的充分整合。我们在三个权威医学图像数据集上评估了TG-LMM,涵盖人体多个部位的分割。与MedSAM、SAM和nnUnet等现有方法相比,我们的方法展现出更优越的性能。

英文摘要:

We propose TG-LMM (Text-Guided Large Multi-Modal Model), a novel approach that leverages textual descriptions of organs to enhance segmentation accuracy in medical images. Existing medical image segmentation methods face several challenges: current medical automatic segmentation models do not effectively utilize prior knowledge, such as descriptions of organ locations; previous text-visual models focus on identifying the target rather than improving the segmentation accuracy; prior models attempt to use prior knowledge to enhance accuracy but do not incorporate pre-trained models. To address these issues, TG-LMM integrates prior knowledge, specifically expert descriptions of the spatial locations of organs, into the segmentation process. Our model utilizes pre-trained image and text encoders to reduce the number of training parameters and accelerate the training process. Additionally, we designed a comprehensive image-text information fusion structure to ensure thorough integration of the two modalities of data. We evaluated TG-LMM on three authoritative medical image datasets, encompassing the segmentation of various parts of the human body. Our method demonstrated superior performance compared to existing approaches, such as MedSAM, SAM and nnUnet.

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

↑