ScribblePrompt

Fast and Flexible Interactive Segmentation for Any Biomedical Image

Hallee E. Wong
MIT CSAIL & MGH
Marianne Rakic
MIT CSAIL & Broad Institute
John Guttag
MIT CSAIL
Adrian V. Dalca
MIT CSAIL & HMS, MGH

ScribblePrompt

Fast and Flexible Interactive Segmentation for Any Biomedical Image

Hallee E. Wong
MIT CSAIL & MGH
Marianne Rakic
MIT CSAIL & MGH
John Guttag
MIT CSAIL
Adrian V. Dalca
MIT CSAIL & HMS, MGH

tl;dr: ScribblePrompt enables users to segment unseen structures in (bio)medical images using scribbles, clicks, and bounding boxes

CT
Microscopy
Photograph
Ultrasound
X-Ray
MRI

Abstract


Semantic segmentation of medical images is a crucial part of both scientific research and clinical care. With enough labelled data, deep learning models can be trained to accurately automate specific medical image segmentation tasks. However, manually segmenting images to create training data is highly labor intensive.

We present ScribblePrompt, an interactive segmentation framework for medical imaging that enables human annotators to segment unseen structures using scribbles, clicks, and bounding boxes.

Scribbles are an intuitive form of user interaction for complex tasks, however most existing methods focus on click-based interactions. We introduce algorithms for simulating realistic scribbles that enable training models that are amenable to multiple types of interaction.

To achieve generalization to new tasks, we train on a diverse collection of 65 open-access biomedical datasets -- using both real and synthetic labels.

We test ScribblePrompt with multiple network architectures on 12 unseen datasets, and demonstrate that it can be used in real-time on a single CPU. We evaluate ScribblePrompt using manually-collected scribbles, simulated interactions, and a user study. In the user study, ScribblePrompt reduced annotation time by 28% while improving Dice by 15% compared to the Segment Anything Model (SAM).

ScribblePrompt outperforms existing methods in all our evaluations.

Citation


If you find our work or any of our materials useful, please cite our paper:

@article{wong2024scribbleprompt,
  title={ScribblePrompt: Fast and Flexible Interactive Segmentation for Any Biomedical Image},
  author={Hallee E. Wong and Marianne Rakic and John Guttag and Adrian V. Dalca},
  journal={arXiv:2312.07381},
  year={2024},
}