We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

eess.IV

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Electrical Engineering and Systems Science > Image and Video Processing

Title: One-Prompt to Segment All Medical Images

Abstract: Large foundation models, known for their strong zero-shot generalization, have excelled in visual and language applications. However, applying them to medical image segmentation, a domain with diverse imaging types and target labels, remains an open challenge. Current approaches, such as adapting interactive segmentation models like Segment Anything Model (SAM), require user prompts for each sample during inference. Alternatively, transfer learning methods like few/one-shot models demand labeled samples, leading to high costs. This paper introduces a new paradigm toward the universal medical image segmentation, termed 'One-Prompt Segmentation.' One-Prompt Segmentation combines the strengths of one-shot and interactive methods. In the inference stage, with just \textbf{one prompted sample}, it can adeptly handle the unseen task in a single forward pass. We train One-Prompt Model on 64 open-source medical datasets, accompanied by the collection of over 3,000 clinician-labeled prompts. Tested on 14 previously unseen datasets, the One-Prompt Model showcases superior zero-shot segmentation capabilities, outperforming a wide range of related methods. The code and data is released as this https URL
Comments: arXiv admin note: text overlap with arXiv:2304.12620
Subjects: Image and Video Processing (eess.IV); Computer Vision and Pattern Recognition (cs.CV)
Cite as: arXiv:2305.10300 [eess.IV]
  (or arXiv:2305.10300v5 [eess.IV] for this version)

Submission history

From: Junde Wu [view email]
[v1] Wed, 17 May 2023 15:37:47 GMT (12777kb,D)
[v2] Mon, 27 Nov 2023 17:16:58 GMT (37522kb,D)
[v3] Wed, 20 Dec 2023 03:01:18 GMT (37522kb,D)
[v4] Thu, 11 Apr 2024 00:35:04 GMT (37485kb,D)
[v5] Wed, 17 Apr 2024 11:04:57 GMT (20580kb,D)

Link back to: arXiv, form interface, contact.