A deep learning approach for multi‐stage classification of brain tumor through magnetic resonance images

Abstract

AbstractBrain tumor is the 10th major cause of death among humans. The detection of brain tumor is a significant process in the medical field. Therefore, the objective of this research work is to propose a fully automated deep learning framework for multistage classification. Besides, this study focuses on to develop an efficient and reliable system using a convolutional neural network (CNN). In this study, the fast bounding box technique is used for segmentation. Moreover, the CNN layers‐based three models are developed for multistage classification through magnetic resonance images on three publicly available datasets. The first dataset is obtained from Kaggle Repository (Dataset‐1), the second dataset is known as Figshare (Dataset‐2), and the third dataset is called REMBRANDT (Dataset‐3) to classify the MR images into different grades. Different augmentation techniques are applied to increase the data size of MR images. In pre‐processing, the proposed models achieved higher Peak Signal‐to‐Noise ratio to remove noise. The first proposed deep CNN framework mentioned as Classification‐1 has obtained 99.40% accuracy, which classified MR images into two classes, that is (i) normal and (ii) abnormal, while the second proposed CNN framework mentioned as Classification‐2 has obtained 97.78% accuracy, which classified brain tumor into three types, which are meningioma, glioma, and pituitary. Similarly, the third developed CNN framework mentioned as Classification‐3 has obtained 98.91% accuracy that further classified MR images of tumors into four different classes as: Grade I, Grade II, Grade III, and Grade IV. The results demonstrate that the proposed models achieved better performance on three large and diverse datasets. The comparison of obtained outcomes shows that the developed models are more efficient and effective than state‐of‐the‐art methods.