arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.09592eess.IVq-bio.QM

Segment Any Tumour:一种用于全身分析的不确定性感知视觉基础模型

Segment Any Tumour: An Uncertainty-Aware Vision Foundation Model for Whole-Body Analysis

Himashi Peiris, Sizhe Wang, Gary Egan, Mehrtash Harandi, Meng Law, Zhaolin Chen

AI总结:

针对提示驱动视觉基础模型在医学成像中因组织异质等导致分割欠佳的问题,提出轻量级体积模型SAT3D,通过移位窗口视觉Transformer和不确定性感知训练提升肿瘤分割稳健性与泛化性,在多模态数据集上表现优异。

摘要:

提示驱动的视觉基础模型(如Segment Anything Model)最近在计算机视觉中展现出显著的适应性。然而,由于异质组织结构、成像伪影和低对比度边界,它们直接应用于医学成像仍具挑战性,尤其是在肿瘤和癌症原发灶中,导致模糊或重叠病变区域的分割效果欠佳。在此,我们提出Segment Any Tumour 3D(SAT3D),这是一种轻量级体积基础模型,旨在实现跨多种医学成像模态的稳健且可泛化的肿瘤分割。SAT3D集成了移位窗口视觉Transformer以获取分层体积表示,并结合不确定性感知训练流程,该流程明确将不确定性估计作为提示,指导低对比度区域的可靠边界预测。对抗学习进一步提升了模型在模糊病理区域的性能。我们在11个公开数据集上对SAT3D与3个近期视觉基础模型及nnUNet进行基准测试,这些数据集包含3884个肿瘤和癌症病例用于训练,694个病例用于分布内评估。在跨多种模态和癌症原发灶的17075个3D体积-掩码对上训练后,SAT3D展现出强大的泛化性和稳健性。为促进实际应用和临床转化,我们开发了3D Slicer插件,支持使用训练后的SAT3D模型进行交互式、提示驱动的分割和可视化。大量实验凸显了其在挑战性和分布外场景中提升分割精度的有效性,强调了其作为医学图像分析可扩展基础模型的潜力。

英文摘要:

Prompt-driven vision foundation models, such as the Segment Anything Model, have recently demonstrated remarkable adaptability in computer vision. However, their direct application to medical imaging remains challenging due to heterogeneous tissue structures, imaging artefacts, and low-contrast boundaries, particularly in tumours and cancer primaries leading to suboptimal segmentation in ambiguous or overlapping lesion regions. Here, we present Segment Any Tumour 3D (SAT3D), a lightweight volumetric foundation model designed to enable robust and generalisable tumour segmentation across diverse medical imaging modalities. SAT3D integrates a shifted-window vision transformer for hierarchical volumetric representation with an uncertainty-aware training pipeline that explicitly incorporates uncertainty estimates as prompts to guide reliable boundary prediction in low-contrast regions. Adversarial learning further enhances model performance for the ambiguous pathological regions. We benchmark SAT3D against three recent vision foundation models and nnUNet across 11 publicly available datasets, encompassing 3,884 tumour and cancer cases for training and 694 cases for in-distribution evaluation. Trained on 17,075 3D volume-mask pairs across multiple modalities and cancer primaries, SAT3D demonstrates strong generalisation and robustness. To facilitate practical use and clinical translation, we developed a 3D Slicer plugin that enables interactive, prompt-driven segmentation and visualisation using the trained SAT3D model. Extensive experiments highlight its effectiveness in improving segmentation accuracy under challenging and out-of-distribution scenarios, underscoring its potential as a scalable foundation model for medical image analysis.

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

↑