arrow
Return

SAM-UNETR: Clinically Significant Prostate Cancer Segmentation Using Transfer Learning From Large Model

delete2023-01-01
delete4
delete
OA
AI
J
Jesús Alejandro Alzate-Grisales *
A
Alejandro Mora-Rubio
F
Francisco García‐García
R
Reinel Tabares-Soto
M
María de la Iglesia-Vayá *
DOI:10.1109/ACCESS.2023.3326882delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Prostate cancer (PCa) is one of the leading causes of cancer-related mortality among men worldwide. Accurate and efficient segmentation of clinically significant prostate cancer (csPCa) regions from magnetic resonance imaging (MRI) plays a crucial role in diagnosis, treatment planning, and monitoring of the disease, however, this is a challenging task even for the specialized clinicians. This study presents SAM-UNETR, a novel model for segmenting csPCa regions from MRI images. SAM-UNETR combines a transformer-encoder from the Segment Anything Model (SAM), a versatile segmentation model trained on 11 million images, with a residual-convolution decoder inspired by UNETR. The model uses multiple image modalities and applies prostate zone segmentation, normalization, and data augmentation as preprocessing steps. The performance of SAM-UNETR is compared with three other models using the same strategy and preprocessing. The results show that SAM-UNETR achieves superior reliability and accuracy in csPCa segmentation, especially when using transfer learning for the image encoder. This demonstrates the adaptability of large-scale models for different tasks. SAM-UNETR attains a Dice Score of 0.467 and an AUROC of 0.77 for csPCa prediction.
Keywords:
Image segmentation
Training
Principal component analysis
Transfer learning
Task analysis
Magnetic resonance imaging
Lesions
Artificial intelligence
deep learning
prostate cancer
semantic segmentation

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

P
prince felipe research center
Scholars:
1.2K
Papers: 880
Citations: 0
U
Universidad Adolfo Ibanez
Scholars:
1.2K
Papers: 1.3K
Citations: 17
U
Universidad Autonoma de Manizales
Scholars:
175
Papers: 91
Citations: 64
researcher View more organizations