TensorX
返回文献探索

Paper · arXiv 2411.12814

Interactive Medical Image Segmentation: A Benchmark Dataset and Baseline

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

23 upvotesNovember 19, 2024arXiv 预印本
AI 摘要

The IMed-361M benchmark dataset, featuring over 6.4 million medical images and 361 million masks, enhances interactive medical image segmentation through a foundation model and diverse, high-density annotations.

interactive medical image segmentationIMed-361Mvision foundational modelautomatic mask generationquality controlinteractive inputsclicksbounding boxestext promptsmedical image segmentationhigh-quality mask generationscalability

Abstract

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.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Interactive Medical Image Segmentation: A Benchmark Dataset and Baseline | TensorX