Beyond Adapting SAM: Towards End-to-End Ultrasound Image Segmentation via Auto Prompting

TL;DR

This paper introduces SAMUS and AutoSAMUS, novel models for ultrasound image segmentation that enhance generalization and enable end-to-end automatic segmentation, outperforming existing models on a large ultrasound dataset.

Contribution

The work presents a universal ultrasound segmentation model with an auto prompt generator, reducing manual intervention and improving performance over prior SAM adaptations.

Findings

SAMUS outperforms state-of-the-art models on ultrasound segmentation

AutoSAMUS achieves fully automatic end-to-end segmentation

Models demonstrate strong generalization across object categories

Abstract

End-to-end medical image segmentation is of great value for computer-aided diagnosis dominated by task-specific models, usually suffering from poor generalization. With recent breakthroughs brought by the segment anything model (SAM) for universal image segmentation, extensive efforts have been made to adapt SAM for medical imaging but still encounter two major issues: 1) severe performance degradation and limited generalization without proper adaptation, and 2) semi-automatic segmentation relying on accurate manual prompts for interaction. In this work, we propose SAMUS as a universal model tailored for ultrasound image segmentation and further enable it to work in an end-to-end manner denoted as AutoSAMUS. Specifically, in SAMUS, a parallel CNN branch is introduced to supplement local information through cross-branch attention, and a feature adapter and a position adapter are jointly used to adapt SAM from natural to ultrasound domains while reducing training complexity. AutoSAMUS is realized by introducing an auto prompt generator (APG) to replace the manual prompt encoder of SAMUS to automatically generate prompt embeddings. A comprehensive ultrasound dataset, comprising about 30k images and 69k masks and covering six object categories, is collected for verification. Extensive comparison experiments demonstrate the superiority of SAMUS and AutoSAMUS against the state-of-the-art task-specific and SAM-based foundation models. We believe the auto-prompted SAM-based model has the potential to become a new paradigm for end-to-end medical image segmentation and deserves more exploration. Code and data are available at https://github.com/xianlin7/SAMUS.