A brain tumor segmentation enhancement in MRI images using U-Net and transfer learning.

Journal: BMC medical imaging
Published Date:

Abstract

This paper presents a novel transfer learning approach for segmenting brain tumors in Magnetic Resonance Imaging (MRI) images. Using Fluid-Attenuated Inversion Recovery (FLAIR) abnormality segmentation masks and MRI scans from The Cancer Genome Atlas's (TCGA's) lower-grade glioma collection, our proposed approach uses a VGG19-based U-Net architecture with fixed pretrained weights. The experimental findings, which show an Area Under the Curve (AUC) of 0.9957, F1-Score of 0.9679, Dice Coefficient of 0.9679, Precision of 0.9541, Recall of 0.9821, and Intersection-over-Union (IoU) of 0.9378, show how effective the proposed framework is. According to these metrics, the VGG19-powered U-Net outperforms not only the conventional U-Net model but also other variants that were compared and used different pre-trained backbones in the U-Net encoder.Clinical trial registrationNot applicable as this study utilized existing publicly available dataset and did not involve a clinical trial.

Authors

  • Amin Pourmahboubi
    Department of Computer Science, Faculty of Mathematics, Statistics, and Computer Science, University of Tabriz, Tabriz, East Azerbaijan, Iran.
  • Nazanin Arsalani Saeed
    Department of Biology, Faculty of Natural Science, University of Tabriz, Tabriz, East Azerbaijan, Iran.
  • Hamed Tabrizchi
    Department of Computer Science, Faculty of Mathematics, Statistics, and Computer Science, University of Tabriz, Tabriz, East Azerbaijan, Iran. hamed.tabrizchi@gmail.com.