Text Promptable Surgical Instrument Segmentation with Vision-Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Zijian, Alabi, Oluwatosin, Wei, Meng, Vercauteren, Tom, Shi, Miaojing
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909216183156736
author Zhou, Zijian
Alabi, Oluwatosin
Wei, Meng
Vercauteren, Tom
Shi, Miaojing
author_facet Zhou, Zijian
Alabi, Oluwatosin
Wei, Meng
Vercauteren, Tom
Shi, Miaojing
contents In this paper, we propose a novel text promptable surgical instrument segmentation approach to overcome challenges associated with diversity and differentiation of surgical instruments in minimally invasive surgeries. We redefine the task as text promptable, thereby enabling a more nuanced comprehension of surgical instruments and adaptability to new instrument types. Inspired by recent advancements in vision-language models, we leverage pretrained image and text encoders as our model backbone and design a text promptable mask decoder consisting of attention- and convolution-based prompting schemes for surgical instrument segmentation prediction. Our model leverages multiple text prompts for each surgical instrument through a new mixture of prompts mechanism, resulting in enhanced segmentation performance. Additionally, we introduce a hard instrument area reinforcement module to improve image feature comprehension and segmentation precision. Extensive experiments on several surgical instrument segmentation datasets demonstrate our model's superior performance and promising generalization capability. To our knowledge, this is the first implementation of a promptable approach to surgical instrument segmentation, offering significant potential for practical application in the field of robotic-assisted surgery. Code is available at https://github.com/franciszzj/TP-SIS.
format Preprint
id arxiv_https___arxiv_org_abs_2306_09244
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Text Promptable Surgical Instrument Segmentation with Vision-Language Models
Zhou, Zijian
Alabi, Oluwatosin
Wei, Meng
Vercauteren, Tom
Shi, Miaojing
Computer Vision and Pattern Recognition
In this paper, we propose a novel text promptable surgical instrument segmentation approach to overcome challenges associated with diversity and differentiation of surgical instruments in minimally invasive surgeries. We redefine the task as text promptable, thereby enabling a more nuanced comprehension of surgical instruments and adaptability to new instrument types. Inspired by recent advancements in vision-language models, we leverage pretrained image and text encoders as our model backbone and design a text promptable mask decoder consisting of attention- and convolution-based prompting schemes for surgical instrument segmentation prediction. Our model leverages multiple text prompts for each surgical instrument through a new mixture of prompts mechanism, resulting in enhanced segmentation performance. Additionally, we introduce a hard instrument area reinforcement module to improve image feature comprehension and segmentation precision. Extensive experiments on several surgical instrument segmentation datasets demonstrate our model's superior performance and promising generalization capability. To our knowledge, this is the first implementation of a promptable approach to surgical instrument segmentation, offering significant potential for practical application in the field of robotic-assisted surgery. Code is available at https://github.com/franciszzj/TP-SIS.
title Text Promptable Surgical Instrument Segmentation with Vision-Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2306.09244