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

即时改进 Segment Anything:用于医学图像分割的辅助在线学习与自适应融合

Improving Segment Anything on the Fly: Auxiliary Online Learning and Adaptive Fusion for Medical Image Segmentation

School of Computer Science and Engineering, Nanjing University of Science and Technology(南京理工大学计算机科学与工程学院);Digital Medical Research Center, School of Basic Medical Sciences, Fudan University(复旦大学基础医学院数字医学研究中心);Shanghai Jiaotong University(上海交通大学);the Chinese University of Hong Kong(香港中文大学)

Tianyu Huang, Tao Zhou, Weidi Xie, Shuo Wang, Qi Dou, Yizhe Zhang

AI总结:

针对SAM在医学图像分割精度不足的问题,提出AuxOL方法,利用专家修正标注进行在线学习,通过小型辅助模型与自适应分割融合机制,有效提升了大规模视觉模型在下游医学分割任务中的表现。

摘要:

Segment Anything Model (SAM) 的当前变体(包括原始 SAM 和 Medical SAM)仍然缺乏为医学图像生成足够准确分割的能力。在医学成像场景中,人类专家在 SAM 生成其分割预测后,对特定测试样本的分割进行修正并不罕见。这些修正通常需要使用最先进的标注工具进行手动或半手动校正。受此过程启发,我们引入了一种新方法,利用在线机器学习的优势在测试阶段增强 Segment Anything (SA)。我们使用修正后的标注进行在线学习,旨在提高 SA 在医学图像上的分割质量。为了提高与 SAM 等大规模视觉模型集成时在线学习的有效性和效率,我们提出了一种名为 Auxiliary Online Learning (AuxOL) 的新方法。AuxOL 创建并应用一个小型辅助模型(专家)与 SAM(通才)协同工作,包含自适应在线-批处理和自适应分割融合。在涵盖四种医学成像模态的八个数据集上进行的实验验证了所提方法的有效性。我们的工作提出并验证了一种新的、实用的、有效的方法,用于增强 SA 在下游分割任务(例如医学图像分割)上的表现。

英文摘要:

The current variants of the Segment Anything Model (SAM), which include the original SAM and Medical SAM, still lack the capability to produce sufficiently accurate segmentation for medical images. In medical imaging contexts, it is not uncommon for human experts to rectify segmentations of specific test samples after SAM generates its segmentation predictions. These rectifications typically entail manual or semi-manual corrections employing state-of-the-art annotation tools. Motivated by this process, we introduce a novel approach that leverages the advantages of online machine learning to enhance Segment Anything (SA) during test time. We employ rectified annotations to perform online learning, with the aim of improving the segmentation quality of SA on medical images. To improve the effectiveness and efficiency of online learning when integrated with large-scale vision models like SAM, we propose a new method called Auxiliary Online Learning (AuxOL). AuxOL creates and applies a small auxiliary model (specialist) in conjunction with SAM (generalist), entails adaptive online-batch and adaptive segmentation fusion. Experiments conducted on eight datasets covering four medical imaging modalities validate the effectiveness of the proposed method. Our work proposes and validates a new, practical, and effective approach for enhancing SA on downstream segmentation tasks (e.g., medical image segmentation).

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

↑