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

探索基于视觉语言模型的医学图像分割中的迁移学习

Exploring Transfer Learning in Medical Image Segmentation using Vision-Language Models

Naamii(纳米机构)

Kanchan Poudel, Manish Dhakal, Prasiddha Bhandari, Rabin Adhikari, Safal Thapaliya, Bishesh Khanal

AI总结:

该研究首次系统探索视觉语言分割模型向2D医学图像的迁移,通过11个多模态数据集实验发现其微调后性能与纯图像模型相当,但未充分利用语言提示信息,需新方法挖掘语言辅助信息价值。

摘要:

医学图像分割可量化目标结构的尺寸与形状,为疾病诊断、预后、手术规划及医学认知提供辅助。基于从自然图像-文本对训练的基础视觉语言模型(Vision-Language Models, VLMs)的最新进展,已有多项研究提出将其适配为视觉语言分割模型(Vision-Language Segmentation Models, VLSMs),这类模型可将语言文本作为分割模型的额外输入。推理阶段通过文本引入辅助信息并结合人在环提示机制,为开放词汇分割、提升模型对分布外数据的鲁棒性等带来了独特机遇。尽管针对仅图像输入的分割模型,学界已探索了从自然图像到医学图像的迁移学习,但分割任务中视觉-语言的联合表征仍有待深入研究。本研究首次系统探究了VLSMs向2D医学图像的迁移,使用了经精心筛选的11个涵盖多种模态的数据集,并设计了具有参考价值的语言提示与实验。研究结果表明,在有限医学图像数据集上微调后,VLSMs的分割性能与仅图像输入的模型相当,但并非所有VLSMs都能有效利用语言提示的额外信息,图像特征仍占据主导地位。与传统分割模型相比,VLSMs在处理包含多种模态的混合数据集时性能更优,且在域迁移场景下展现出潜在的鲁棒性,但结果同时显示,仍需提出新方法以帮助VLSMs充分利用语言提示提供的各类辅助信息。代码与数据集可在https://github.com/naamiinepal/medvlsm 获取。

英文摘要:

Medical image segmentation allows quantifying target structure size and shape, aiding in disease diagnosis, prognosis, surgery planning, and comprehension.Building upon recent advancements in foundation Vision-Language Models (VLMs) from natural image-text pairs, several studies have proposed adapting them to Vision-Language Segmentation Models (VLSMs) that allow using language text as an additional input to segmentation models. Introducing auxiliary information via text with human-in-the-loop prompting during inference opens up unique opportunities, such as open vocabulary segmentation and potentially more robust segmentation models against out-of-distribution data. Although transfer learning from natural to medical images has been explored for image-only segmentation models, the joint representation of vision-language in segmentation problems remains underexplored. This study introduces the first systematic study on transferring VLSMs to 2D medical images, using carefully curated $11$ datasets encompassing diverse modalities and insightful language prompts and experiments. Our findings demonstrate that although VLSMs show competitive performance compared to image-only models for segmentation after finetuning in limited medical image datasets, not all VLSMs utilize the additional information from language prompts, with image features playing a dominant role. While VLSMs exhibit enhanced performance in handling pooled datasets with diverse modalities and show potential robustness to domain shifts compared to conventional segmentation models, our results suggest that novel approaches are required to enable VLSMs to leverage the various auxiliary information available through language prompts. The code and datasets are available at https://github.com/naamiinepal/medvlsm.

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

↑