Training-Free Prompt Placement by Propagation for SAM Predictions in Bone CT Scans

Abstract

The Segment Anything Model (SAM) is an interactive foundation segmentation model, showing impressive results for 2D natural images using prompts such as points and boxes. Transferring these results to medical image segmentation is challenging due to the 3D nature of medical images and the high demand of manual interaction. As a 2D architecture, SAM is applied slice-per-slice to a 3D medical scan. This hinders the application of SAM for volumetric medical scans since at least one prompt per class for each single slice is needed. In our work, the applicability is improve by reducing the number of necessary user-generated prompts. We introduce and evaluate multiple training-free strategies to automatically place box prompts in bone CT volumes, given only one initial box prompt per class. The average performance of our methods ranges from 54.22% Dice to 88.26% Dice. At the same time, the number of annotated pixels is reduced significantly from a few millions to two pixels per class. These promising results underline the potential of foundation models in medical image segmentation, paving the way for annotation-efficient, general approaches.

Cite

Text

Magg et al. "Training-Free Prompt Placement by Propagation for SAM Predictions in Bone CT Scans." Proceedings of MIDL 2024, 2024.

Markdown

[Magg et al. "Training-Free Prompt Placement by Propagation for SAM Predictions in Bone CT Scans." Proceedings of MIDL 2024, 2024.](https://mlanthology.org/midl/2024/magg2024midl-trainingfree/)

BibTeX

@inproceedings{magg2024midl-trainingfree,
  title     = {{Training-Free Prompt Placement by Propagation for SAM Predictions in Bone CT Scans}},
  author    = {Magg, Caroline and Verweij, Lukas P.E. and Wee, Maaike A. and Buijs, George S. and Dobbe, Johannes G.G. and Streekstra, Geert J. and Blankevoort, Leendert and Sánchez, Clara I.},
  booktitle = {Proceedings of MIDL 2024},
  year      = {2024},
  pages     = {964-985},
  volume    = {250},
  url       = {https://mlanthology.org/midl/2024/magg2024midl-trainingfree/}
}