Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SegVol: Universal and Interactive Volumetric Medical Image Segmentation

About

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.

Yuxin Du, Fan Bai, Tiejun Huang, Bo Zhao• 2023

Related benchmarks

TaskDatasetResultRank
SegmentationPancreas-CT (test)
Dice58.83
44
Aortic Dissection SegmentationImageTBAD (test)
True Lumen DSC34.41
33
Organ SegmentationWORD
Overall DICE81.65
20
Pan-cancer SegmentationInternal datasets
Lung Tumor DSC68.9
14
Medical Image SegmentationD1 Pediatric Organs in CT
DSC73.95
14
Medical Image SegmentationD2
DSC85.63
14
Medical Image SegmentationToothfairy D3
DSC74.19
14
Medical Image SegmentationD4
DSC51.1
14
Medical Image SegmentationD5 Pancreatic Tumor in MR
DSC60.24
14
Medical Image SegmentationD6
DSC89.1
14
Showing 10 of 31 rows

Other info

Code

Follow for update