SAS通过模拟尺度与纹理增强数据,提升超声小器官分割精度。
SAS: Segment Anything Small for Ultrasound -- A Non-Generative Data Augmentation Technique for Robust Deep Learning in Ultrasound Imaging

- 用缩放嵌入和噪声注入生成逼真训练数据,避免伪影
- 小结构分割Dice提升最高0.35,平均增0.16(95%置信区间)
- 仅需两点即可达到框提示效果,适合资源有限场景
超声图像中小器官的精确分割因噪声和成像条件差异(如探头位置、患者解剖、组织特性及病理)而困难。为此,我们提出一种名为SAS(Segment Anything Small)的简单高效的数据增强技术,专为提升深度学习模型在超声图像中分割小解剖结构的能力而设计。SAS采用双重变换策略:(1) 通过缩放并嵌入器官缩略图至黑色背景来模拟不同器官尺度;(2) 向感兴趣区域注入噪声以模拟不同组织纹理。该方法生成真实且多样化的训练数据,不引入幻觉或伪影,显著提升模型对噪声和变异性的鲁棒性。我们在一个受控的器官特异性医学影像数据集上微调了一个可提示的基础模型,并在1个内部和5个外部数据集上评估性能。实验结果表明,分割性能显著提升,Dice分数最高提高0.35,平均提升0.16(95% CI 0.132, 0.188)。此外,迭代点提示提供精确控制与自适应优化,仅用两点即实现与边界框提示相当的效果。SAS增强了模型在多种解剖结构和成像条件下,尤其是小结构上的鲁棒性和泛化能力,同时不影响大结构的准确性。该方法计算高效,无需大量人工标注,是推动医疗影像分析的重要工具,尤其适用于资源受限环境。
原文摘要 · Abstract (English)
Accurate segmentation of anatomical structures in ultrasound (US) images, particularly small ones, is challenging due to noise and variability in imaging conditions (e.g., probe position, patient anatomy, tissue characteristics and pathology). To address this, we introduce Segment Anything Small (SAS), a simple yet effective scale- and texture-aware data augmentation technique designed to enhance the performance of deep learning models for segmenting small anatomical structures in ultrasound images. SAS employs a dual transformation strategy: (1) simulating diverse organ scales by resizing and embedding organ thumbnails into a black background, and (2) injecting noise into regions of interest to simulate varying tissue textures. These transformations generate realistic and diverse training data without introducing hallucinations or artifacts, improving the model's robustness to noise and variability. We fine-tuned a promptable foundation model on a controlled organ-specific medical imaging dataset and evaluated its performance on one internal and five external datasets. Experimental results demonstrate significant improvements in segmentation performance, with Dice score gains of up to 0.35 and an average improvement of 0.16 [95% CI 0.132,0.188]. Additionally, our iterative point prompts provide precise control and adaptive refinement, achieving performance comparable to bounding box prompts with just two points. SAS enhances model robustness and generalizability across diverse anatomical structures and imaging conditions, particularly for small structures, without compromising the accuracy of larger ones. By offering a computationally efficient solution that eliminates the need for extensive human labeling efforts, SAS emerges as a powerful tool for advancing medical image analysis, particularly in resource-constrained settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。