让AI看懂医生写的报告,自动分割CT影像中的器官和肿瘤
Towards Universal Text-driven CT Image Segmentation
- 用医疗报告自动生成细粒度标注,训练能理解任意文本描述的3D CT分割模型
- 在9个公开数据集上表现优于现有方法,对未见过的文本提示也能准确响应
- 适合临床科研人员快速构建通用医学图像分析工具
计算机断层扫描(CT)广泛用于精确可视化与分割器官及病灶。尽管卷积神经网络(CNN)和视觉变换器(ViT)显著提升了CT图像分析能力,但其性能在多样化的真实临床数据上常下降。虽然基础模型提供了更广泛的适应性,但大规模体素级医学图像标注难题限制了其潜力。为此,基于提示的模型应运而生。视觉提示方法如通用图像分割模型(SAM)仍需大量人工输入,在临床场景中易产生歧义。相比之下,使用文本提示的基础模型更具灵活性与临床相关性。然而,当前文本提示模型(如CLIP驱动的通用模型)仅能处理训练时见过的文本,难以应对真实临床中复杂多变的描述。为此,我们提出OpenVocabCT——一个在大规模3D CT图像上预训练的视觉语言模型,支持通用文本驱动分割。利用大规模CT-RATE数据集,我们通过大语言模型将诊断报告分解为细粒度的器官级描述,实现多粒度对比学习。我们在九个公共数据集上的器官与肿瘤分割任务中评估该模型,结果表明其性能显著优于现有方法。所有代码、数据集与模型将公开发布于https://github.com/ricklisz/OpenVocabCT。
原文摘要 · Abstract (English)
Computed tomography (CT) is extensively used for accurate visualization and segmentation of organs and lesions. While deep learning models such as convolutional neural networks (CNNs) and vision transformers (ViTs) have significantly improved CT image analysis, their performance often declines when applied to diverse, real-world clinical data. Although foundation models offer a broader and more adaptable solution, their potential is limited due to the challenge of obtaining large-scale, voxel-level annotations for medical images. In response to these challenges, prompting-based models using visual or text prompts have emerged. Visual-prompting methods, such as the Segment Anything Model (SAM), still require significant manual input and can introduce ambiguity when applied to clinical scenarios. Instead, foundation models that use text prompts offer a more versatile and clinically relevant approach. Notably, current text-prompt models, such as the CLIP-Driven Universal Model, are limited to text prompts already encountered during training and struggle to process the complex and diverse scenarios of real-world clinical applications. Instead of fine-tuning models trained from natural imaging, we propose OpenVocabCT, a vision-language model pretrained on large-scale 3D CT images for universal text-driven segmentation. Using the large-scale CT-RATE dataset, we decompose the diagnostic reports into fine-grained, organ-level descriptions using large language models for multi-granular contrastive learning. We evaluate our OpenVocabCT on downstream segmentation tasks across nine public datasets for organ and tumor segmentation, demonstrating the superior performance of our model compared to existing methods. All code, datasets, and models will be publicly released at https://github.com/ricklisz/OpenVocabCT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。