No More Training: SAM’s Zero-Shot Transfer Capabilities for Cost-Efficient Medical Image Segmentation

Semantic segmentation of medical images presents an enormous potential for diagnosis and surgery. However, achieving precise results involves designing and training complex Deep Learning (DL) models specifically for this task, which is only available to some. SAM is a model developed by Meta capable...

Descripción completa

Detalles Bibliográficos
Autores: Gutiérrez Gallardo, Juan Diego, Rodriguez-Echeverria, Roberto, Delgado, Emilio, Suero-Rodrigo, Miguel Ángel, Sánchez-Figueroa, Fernando
Tipo de recurso: artículo
Fecha de publicación:2024
País:España
Institución:Universidad de Santiago de Compostela (USC)
Repositorio:Minerva. Repositorio Institucional de la Universidad de Santiago de Compostela
Idioma:inglés
OAI Identifier:oai:minerva.usc.gal:10347/41028
Acceso en línea:https://hdl.handle.net/10347/41028
Access Level:acceso abierto
Palabra clave:Image segmentation
Lung
X-ray imaging
Computed tomography
Medical diagnostic imaging
Training
Task analysis
Deep learning
Zero-shot learning
Medical imaging
Semantic segmentation
Descripción
Sumario:Semantic segmentation of medical images presents an enormous potential for diagnosis and surgery. However, achieving precise results involves designing and training complex Deep Learning (DL) models specifically for this task, which is only available to some. SAM is a model developed by Meta capable of segmenting objects present in virtually any type of image. This paper showcases SAM’s robustness and exceptional performance in medical image segmentation, even in the absence of direct training on these image types (lung Computed Tomographies (CTs) and chest X-rays, in particular). Additionally, it achieves this impressive outcome while requiring minimal user intervention. Although the dataset used to train SAM does not contain a single sample of both medical image types, processing a popular dataset comprised of 20 volumes with a total of 3520 slices using the ViT-L version of the model yields an average Jaccard index of 91.45% and an average Dice score of 94.95% . The same version of the model achieves a 93.19% Dice score and a 87.45% Jaccard index when segmenting a frequently-used chest X-ray dataset. The values obtained are above the 70% mark recommended in the literature, and close to state-of-the art models developed specifically for medical segmentation. These results are achieved without user interaction by providing the model with positive prompts based on the masks of the dataset used and a negative prompt located in the center of bounding box that contains the masks.