无需标注,自动分割医学影像,准确率提升69%
AutoMiSeg: Automatic Medical Image Segmentation via Test-Time Adaptation of Foundation Models
- 用视觉语言模型生成初始框,再优化提示信息进行分割
- 在7个数据集上实现Dice分数71.81,相对前人提升69%
- 适合无标注、多任务场景的自动化医学图像分析
医学图像分割对临床诊断至关重要,但现有深度学习方法通常需要大量专家工作,包括标注大规模训练数据或为每个新病例提供推理时的提示。本文提出一种零样本、全自动的分割流程,结合现成的视觉-语言和分割基础模型。给定医学图像和任务定义(如“分割眼底图像中的视盘”),该方法先通过定位模型生成初始边界框,再经视觉提示增强模块优化提示,由可提示分割模型生成最终掩码。为解决领域差异与结果验证问题,引入测试时自适应框架,包含一组可学习适配器,将医学输入对齐至基础模型表征;其超参数通过贝叶斯优化调整,由代理验证模型引导,无需真实标签。该流程在七个多样化医学影像数据集上评估,表现优异:通过合理分解与测试时自适应,全自动管道不仅显著超越此前最优方法(Dice分数从42.53提升至71.81,相对提高69%),且性能媲美弱提示交互式基础模型。
原文摘要 · Abstract (English)
Medical image segmentation is vital for clinical diagnosis, yet current deep learning methods often demand extensive expert effort, i.e., either through annotating large training datasets or providing prompts at inference time for each new case. This paper introduces a zero-shot and automatic segmentation pipeline that combines off-the-shelf vision-language and segmentation foundation models. Given a medical image and a task definition (e.g., "segment the optic disc in an eye fundus image"), our method uses a grounding model to generate an initial bounding box, followed by a visual prompt boosting module that enhance the prompts, which are then processed by a promptable segmentation model to produce the final mask. To address the challenges of domain gap and result verification, we introduce a test-time adaptation framework featuring a set of learnable adaptors that align the medical inputs with foundation model representations. Its hyperparameters are optimized via Bayesian Optimization, guided by a proxy validation model without requiring ground-truth labels. Our pipeline offers an annotation-efficient and scalable solution for zero-shot medical image segmentation across diverse tasks. Our pipeline is evaluated on seven diverse medical imaging datasets and shows promising results. By proper decomposition and test-time adaptation, our fully automatic pipeline not only substantially surpasses the previously best-performing method, yielding a 69\% relative improvement in accuracy (Dice Score from 42.53 to 71.81), but also performs competitively with weakly-prompted interactive foundation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。