Improving Medical Image Segmentation Using Test-Time Augmentation with MedSAM

Wasfieh Nazzal, Karl Thurnhofer-Hemsi and Ezequiel López-Rubio ()
Additional contact information
Wasfieh Nazzal: Department of Computer Languages and Computer Science, University of Málaga, Bulevar Louis Pasteur, 35, 29071 Málaga, Spain
Karl Thurnhofer-Hemsi: Department of Computer Languages and Computer Science, University of Málaga, Bulevar Louis Pasteur, 35, 29071 Málaga, Spain
Ezequiel López-Rubio: Department of Computer Languages and Computer Science, University of Málaga, Bulevar Louis Pasteur, 35, 29071 Málaga, Spain

Mathematics, 2024, vol. 12, issue 24, 1-22

Abstract: Medical image segmentation is crucial for diagnostics and treatment planning, yet traditional methods often struggle with the variability of real-world clinical data. Deep learning models, like the Segment Anything Model (SAM), have been proposed as a powerful tool that helps to delimit regions using a prompt. This work proposes a methodology to improve the quality of the segmentation by integrating test-time augmentation (TTA) with the SAM for medical applications (MedSAM) by using random circular shifts, addressing challenges such as misalignments and imaging variability. The method generates several input variations during inference that are combined after, improving robustness and segmentation accuracy without requiring retraining. Evaluated across diverse computed tomography (CT) datasets, including Medical Segmentation Decathlon (MSD), KiTS, and COVID-19-20, the proposed method demonstrated consistent improvements in Dice Similarity Coefficient (DSC) and Normalized Surface Dice (NSD) metrics. The highest performances were 93.6% DSC and 97% NSD. Notably, it achieved superior boundary precision and surface alignment in complex regions like the pancreas and colon, outperforming baseline models, including MedSAM and DeepLabv3+. The approach is computationally feasible, leveraging a balance of augmentation intensity and segmentation accuracy.

Keywords: medical image segmentation; deep learning; test-time augmentation; random circular shifts (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/24/4003/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/24/4003/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:24:p:4003-:d:1548550

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().