Ensemble of Deep Learning Models for Brain Tumor Detection
Suraj Patil, D. K. Kirange
Abstract
In the last two decades, improvement in artificial intelligence and medical imaging technology have made healthcare sector to achieve some remarkable achievements in diseases analysis and prediction. Due to advancement in medical imaging technology the brain images are taken in different modalities, that gives 3D view of different sections of brain for tumor diagnosis. The ability to extract relevant characteristics from magnetic resonance imaging (MRI) scans is a crucial step for brain tumor classifiers. As a result, several studies have proposed various strategies to extract relevant features from different modalities of MRI to predict the growth of abnormal tumor. Most of techniques used conventional techniques of image processing for feature extraction and machine learning for classification. More recently, the use deep learning algorithms in medical imaging has resulted in significant improvements in the classification and diagnosis of brain tumor. Since tumors are located at different regions of brain, the localizing the tumor and classifying it to particular category is challenging task. In this paper, we have solved this problem by designing deep ensemble model. In the proposed approach, first shallow convolutional neural network (SCNN) and VGG16 network were designed with T1C modality MRI image and subsequently loss and accuracy were examined. To improve the performance of model in terms accuracy and loss information, the extracted features from both the deep learning model were fused to improve the classification accuracy of three types of tumors. The obtained results from ensemble deep convolutional neural network model (EDCNN), proved that the fusion of deep learning model improves the accuracy of multiclass classification problem and also tries to address the problem of overfitting of model for imbalance dataset. The proposed model tries to give classification accuracy up to 97.77%. Furthermore, the proposed framework, achieves competitive results when compared with other state of art studies.