SegVol: Universal and Interactive Volumetric Medical Image Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Du, Yuxin, Bai, Fan, Huang, Tiejun, Zhao, Bo
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916609223819264
author Du, Yuxin
Bai, Fan
Huang, Tiejun
Zhao, Bo
author_facet Du, Yuxin
Bai, Fan
Huang, Tiejun
Zhao, Bo
contents Precise image segmentation provides clinical study with instructive information. Despite the remarkable progress achieved in medical image segmentation, there is still an absence of a 3D foundation segmentation model that can segment a wide range of anatomical categories with easy user interaction. In this paper, we propose a 3D foundation segmentation model, named SegVol, supporting universal and interactive volumetric medical image segmentation. By scaling up training data to 90K unlabeled Computed Tomography (CT) volumes and 6K labeled CT volumes, this foundation model supports the segmentation of over 200 anatomical categories using semantic and spatial prompts. To facilitate efficient and precise inference on volumetric images, we design a zoom-out-zoom-in mechanism. Extensive experiments on 22 anatomical segmentation tasks verify that SegVol outperforms the competitors in 19 tasks, with improvements up to 37.24% compared to the runner-up methods. We demonstrate the effectiveness and importance of specific designs by ablation study. We expect this foundation model can promote the development of volumetric medical image analysis. The model and code are publicly available at: https://github.com/BAAI-DCAI/SegVol.
format Preprint
id arxiv_https___arxiv_org_abs_2311_13385
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle SegVol: Universal and Interactive Volumetric Medical Image Segmentation
Du, Yuxin
Bai, Fan
Huang, Tiejun
Zhao, Bo
Computer Vision and Pattern Recognition
Precise image segmentation provides clinical study with instructive information. Despite the remarkable progress achieved in medical image segmentation, there is still an absence of a 3D foundation segmentation model that can segment a wide range of anatomical categories with easy user interaction. In this paper, we propose a 3D foundation segmentation model, named SegVol, supporting universal and interactive volumetric medical image segmentation. By scaling up training data to 90K unlabeled Computed Tomography (CT) volumes and 6K labeled CT volumes, this foundation model supports the segmentation of over 200 anatomical categories using semantic and spatial prompts. To facilitate efficient and precise inference on volumetric images, we design a zoom-out-zoom-in mechanism. Extensive experiments on 22 anatomical segmentation tasks verify that SegVol outperforms the competitors in 19 tasks, with improvements up to 37.24% compared to the runner-up methods. We demonstrate the effectiveness and importance of specific designs by ablation study. We expect this foundation model can promote the development of volumetric medical image analysis. The model and code are publicly available at: https://github.com/BAAI-DCAI/SegVol.
title SegVol: Universal and Interactive Volumetric Medical Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2311.13385