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

Interactive Medical Image Segmentation: A Benchmark Dataset and Baseline

About

Interactive Medical Image Segmentation (IMIS) has long been constrained by the limited availability of large-scale, diverse, and densely annotated datasets, which hinders model generalization and consistent evaluation across different models. In this paper, we introduce the IMed-361M benchmark dataset, a significant advancement in general IMIS research. First, we collect and standardize over 6.4 million medical images and their corresponding ground truth masks from multiple data sources. Then, leveraging the strong object recognition capabilities of a vision foundational model, we automatically generated dense interactive masks for each image and ensured their quality through rigorous quality control and granularity management. Unlike previous datasets, which are limited by specific modalities or sparse annotations, IMed-361M spans 14 modalities and 204 segmentation targets, totaling 361 million masks-an average of 56 masks per image. Finally, we developed an IMIS baseline network on this dataset that supports high-quality mask generation through interactive inputs, including clicks, bounding boxes, text prompts, and their combinations. We evaluate its performance on medical image segmentation tasks from multiple perspectives, demonstrating superior accuracy and scalability compared to existing interactive segmentation models. To facilitate research on foundational models in medical computer vision, we release the IMed-361M and model at https://github.com/uni-medical/IMIS-Bench.

Junlong Cheng, Bin Fu, Jin Ye, Guoan Wang, Tianbin Li, Haoyu Wang, Ruoyu Li, He Yao, Junren Chen, Jingwen Li, Yanzhou Su, Min Zhu, Junjun He• 2024

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationISIC 2018
Dice Score88.93
92
Medical Image SegmentationCOVID-CT
Dice (%)56.05
32
Interactive Medical Image SegmentationMRI (Magnetic Resonance Imaging)
Dice0.849
16
Interactive Medical Image SegmentationFundus
Dice0.841
16
Interactive Medical Image SegmentationCT (Computed Tomography)
Dice Coefficient80.5
16
Interactive Medical Image SegmentationEndoscopy
Dice Coefficient90.5
16
Interactive Medical Image SegmentationAverage across 6 medical imaging modalities
Dice77.9
16
Interactive Medical Image SegmentationX-Ray
Dice0.706
16
Interactive Medical Image SegmentationUltrasound
Dice56.9
16
Medical Image SegmentationACDC
Average IoU61.89
16
Showing 10 of 30 rows

Other info

Follow for update