Transfer Learning for Classification of 2D Brain MRI Images and Tumor Segmentation

Abstract

The focus of the paper is to classify the images into tumorous and non-tumorous and then locate the tumor. Amongst many medical imaging applications segmentation of Brain Tumors is an important and arduous task as the data acquired is disrupted due to artifacts being produced and acquisition time being very less, so classifying and finding the exact location of tumor is one of the most important jobs. In the paper, deep learning specifically the convolutional neural network is used to demonstrate its potential for image classification task. As the learning from available dataset will be low, so we use transfer learning [4] approach, as it is a developing AI strategy that overwhelms with the best outcomes on several image classification assignments because the pre-trained models have gained good knowledge about the features by training on a large number of images. Since, medical image datasets are hard to collect so transfer learning (Alexnet) [1] is used. Later on, after successful classification the aim is to find the exact location of the tumor and this is achieved using basics of image processing inspired by well-known technique of Mask R-CNN [9].