U-Net Model-Based Classification and Description of Brain Tumor in MRI Images

Abstract

In this paper, we discuss the classification of the brain tumor in Magnetic Resonance Imaging (MRI) images using the U-Net model, then evaluate parameters that indicate the performance of the model. We also discuss the extraction of the tumor region from brain image and description of the tumor regarding its position and size. Here, we consider the case of Gliomas, one of the types of brain tumors, which occur in common and can be fatal depending on their position and growth. U-Net is a model of Convolutional Neural Network (CNN) which has U-shaped architecture. MRI employs a non-invasive technique and can very well provide soft-tissue contrast and hence, for the detection and description of the brain tumor, this imaging method can be beneficial. Manual delineation of tumors from brain MRI is laborious, time-consuming and can vary from expert to expert. Our work forms a computer aided technique which is relatively faster and reproducible, and the accuracy is very much on par with ground truth. The results of the work can be used for treatment planning and further processing related to storage or transmission of images.