YOLO-UNet Architecture for Detecting and Segmenting the Localized MRI Brain Tumor Image

Nur Iriawan, Anindya A. Pravitasari*, Ulfa S. Nuraini, Nur I. Nirmalasari, Taufik Azmi, Muhammad Nasrudin, Adam F. Fandisyah, Kartika Fithriasari, Santi W. Purnami, Irhamah, Widiana Ferriastuti

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Brain tumor detection and segmentation are the main issues in biomedical engineering research fields, and it is always challenging due to its heterogeneous shape and location in MRI. The quality of the MR images also plays an important role in providing a clear sight of the shape and boundary of the tumor. The clear shape and boundary of the tumor will increase the probability of safe medical surgery. Analysis of this different scope of image types requires refined computerized quantification and visualization tools. This paper employed deep learning to detect and segment brain tumor MRI images by combining the convolutional neural network (CNN) and fully convolutional network (FCN) methodology in serial. The fundamental finding is to detect and localize the tumor area with YOLO-CNN and segment it with the FCN-UNet architecture. This analysis provided automatic detection and segmentation as well as the location of the tumor. The segmentation using the UNet is run under four scenarios, and the best one is chosen by the minimum loss and maximum accuracy value. In this research, we used 277 images for training, 69 images for validation, and 14 images for testing. The validation is carried out by comparing the segmentation results with the medical ground truth to provide the correct classification ratio (CCR). This study succeeded in the detection of brain tumors and provided a clear area of the brain tumor with a high CCR of about 97%.

Original languageEnglish
Article number3819801
JournalApplied Computational Intelligence and Soft Computing
Volume2024
DOIs
Publication statusPublished - 2024

Fingerprint

Dive into the research topics of 'YOLO-UNet Architecture for Detecting and Segmenting the Localized MRI Brain Tumor Image'. Together they form a unique fingerprint.

Cite this