ScribblePrompt: Fast and Flexible Interactive Segmentation for Any Biomedical Image

TL;DR

ScribblePrompt is an interactive biomedical image segmentation tool that enables rapid, accurate annotations using minimal user input, outperforming previous methods in accuracy and efficiency across diverse datasets.

Contribution

The paper introduces ScribblePrompt, a novel neural network-based interactive segmentation system that improves accuracy and reduces annotation time for biomedical images using diverse training and innovative interaction algorithms.

Findings

Outperforms previous methods in accuracy on unseen datasets.

Reduces annotation time by 28% in user studies.

Improves Dice score by 15% over existing tools.

Abstract

Biomedical image segmentation 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 biomedical image segmentation tasks. However, manually segmenting images to create training data is highly labor intensive and requires domain expertise. We present \emph{ScribblePrompt}, a flexible neural network based interactive segmentation tool for biomedical imaging that enables human annotators to segment previously unseen structures using scribbles, clicks, and bounding boxes. Through rigorous quantitative experiments, we demonstrate that given comparable amounts of interaction, ScribblePrompt produces more accurate segmentations than previous methods on datasets unseen during training. In a user study with domain experts, ScribblePrompt reduced annotation time by 28% while improving Dice by 15% compared to the next best method. ScribblePrompt's success rests on a set of careful design decisions. These include a training strategy that incorporates both a highly diverse set of images and tasks, novel algorithms for simulated user interactions and labels, and a network that enables fast inference. We showcase ScribblePrompt in an interactive demo, provide code, and release a dataset of scribble annotations at https://scribbleprompt.csail.mit.edu