27 August 2022 Multi-task learning architecture for brain tumor detection and segmentation in MRI images
Maria Nazir, Muhammad Junaid Ali, Hafiz Zahid Tufail, Ahmad Raza Shahid, Basit Raza, Sadia Shakil, Khurram Khurshid
Author Affiliations +
Abstract

Brain tumors have existed for a long time, but their rate is increasing among all age groups. Early detection is essential for timely prognosis as it is a dangerous and life-threatening disease. Since the advent of artificial intelligence and machine learning, different algorithms have been proposed to classify and segment brain tumors. However, all have their limitations when it comes to local deployment. The main drawback is designing multiple architectures for multiple tasks, which increases the time and computational complexity and affects performance. This paper addresses this drawback by proposing a multi-task learning (MTL) model that can take 2D-magnetic resonance imaging (MRI) images as input and gives predictions for multiple outputs such as detection and segmentation. It gives two outputs from one architecture: the system’s efficiency. Brain Tumor Segmentation (BRaTs) 2019 and BRaTs 2020 dataset has been used to evaluate the proposed architecture. Experimental results show that the best model shows 98% accuracy for detecting MRI images as either normal/abnormal, whereas an overall Dice score of 92% for multi-class segmentation of high-grade glioma gliomas into the whole tumor, enhancing tumor and core tumor. The overall performance of the proposed architecture proves that it can be the best-suited framework for clinical setup.

© 2022 SPIE and IS&T
Maria Nazir, Muhammad Junaid Ali, Hafiz Zahid Tufail, Ahmad Raza Shahid, Basit Raza, Sadia Shakil, and Khurram Khurshid "Multi-task learning architecture for brain tumor detection and segmentation in MRI images," Journal of Electronic Imaging 31(5), 051606 (27 August 2022). https://doi.org/10.1117/1.JEI.31.5.051606
Received: 30 January 2022; Accepted: 10 August 2022; Published: 27 August 2022
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image segmentation

Tumors

Brain

Magnetic resonance imaging

Neuroimaging

Data modeling

Computer programming

Back to Top