Next Article in Journal
Accuracy of Intra-Oral Radiography and Cone Beam Computed Tomography in the Diagnosis of Buccal Bone Loss
Previous Article in Journal
Thangka Image Captioning Based on Semantic Concept Prompt and Multimodal Feature Optimization
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

MRI-Based Effective Ensemble Frameworks for Predicting Human Brain Tumor

1
Glocal School of Science and Technology, Glocal University, Delhi-Yamunotri Marg (State Highway 57), Mirzapur Pole 247121, India
2
Department of Management Information Systems, College of Business Administration, King Faisal University, Al-Ahsa 31982, Saudi Arabia
3
College of Computer Science and Information Technology, Jazan University, Jazan 45142, Saudi Arabia
4
Department of Software Engineering, Faculty of Engineering and Technology, University of Sindh, Jamshoro 76080, Pakistan
*
Authors to whom correspondence should be addressed.
J. Imaging 2023, 9(8), 163; https://doi.org/10.3390/jimaging9080163
Submission received: 9 July 2023 / Revised: 12 August 2023 / Accepted: 15 August 2023 / Published: 16 August 2023
(This article belongs to the Special Issue Multimodal Imaging for Radiotherapy: Latest Advances and Challenges)

Abstract

:
The diagnosis of brain tumors at an early stage is an exigent task for radiologists. Untreated patients rarely survive more than six months. It is a potential cause of mortality that can occur very quickly. Because of this, the early and effective diagnosis of brain tumors requires the use of an automated method. This study aims at the early detection of brain tumors using brain magnetic resonance imaging (MRI) data and efficient learning paradigms. In visual feature extraction, convolutional neural networks (CNN) have achieved significant breakthroughs. The study involves features extraction by deep convolutional layers for the efficient classification of brain tumor victims from the normal group. The deep convolutional neural network was implemented to extract features that represent the image more comprehensively for model training. Using deep convolutional features helps to increase the precision of tumor and non-tumor patient classifications. In this paper, we experimented with five machine learnings (ML) to heighten the understanding and enhance the scope and significance of brain tumor classification. Further, we proposed an ensemble of three high-performing individual ML models, namely Extreme Gradient Boosting, Ada-Boost, and Random Forest (XG-Ada-RF), to derive binary class classification output for detecting brain tumors in images. The proposed voting classifier, along with convoluted features, produced results that showed the highest accuracy of 95.9% for tumor and 94.9% for normal. Compared to individual methods, the proposed ensemble approach demonstrated improved accuracy and outperformed the individual methods.

1. Introduction

Whenever cells in the brain begin to multiply uncontrolled, a mass-like formation forms, known as a brain tumor [1]. Patients with brain tumors have varying survival rates, depending on the tumor’s size and stage of development. According to their point of origin, brain tumors can be identified as either primary or secondary. Unlike secondary tumors, which form outside, primary brain tumors begin inside the brain. Brain tumors are a type of cancer that can lead to severe neurological and physical symptoms and ultimately, death if not detected and treated early. The detection of brain tumors is a challenging task that requires the expertise of highly trained medical professionals. However, the current methods used for brain tumor detection are time-consuming, costly, and can be inaccurate in certain cases [2]. There are several medical imaging modalities that are commonly used to locate and identify a brain tumor. These include magnetic resonance imaging (MRI), magnetic resonance spectroscopy (MRS), and computed tomography (CT). There is an assortment of aspects, such as the suspected tumor kind and location, that should be taken into account when determining which imaging modality to use, based on the patient’s medical history and the availability of imaging equipment. In some cases, multiple imaging modalities may be used to obtain the accurate diagnosis. Therefore, there is a need for a more efficient and accurate method for brain tumor detection.
Artificial intelligence (AI) and machine learning (ML) [3,4,5] have shown tremendous potential in various medical applications [6], including the detection of brain tumors. AI and ML algorithms can analyze large amounts of medical data, such as CT and MRI scans, and accurately identify the presence of brain tumors [7]. They can also provide additional information, such as the type of tumor and its location, which can aid in the treatment planning process. This research has resulted in experimenting with various algorithms and models that can accurately detect brain tumors. However, there are still numerous challenges that need to be addressed, such as improving the accuracy of these algorithms and ensuring their reliability in clinical settings.
Deep neural networks (DNNs) have deep learning, which has revolutionized different areas, such as agriculture [8,9,10,11,12], education [13], finance [14], healthcare [15] and more. Deep learning networks are effective in brain tumor detection and diagnosis because they can automatically learn and extract features from large amounts of brain medical imaging data [16]. DNNs are a particular type of ML algorithm that are intended to replicate the brain’s framework and the function of the human brain. To learn and extract more sophisticated characteristics from the input data, they are built of progressively more complicated layers of interconnected nodes [17,18].
In the context of brain tumor detection and diagnosis, DNNs can analyze large volumes of medical imaging data and automatically identify patterns and features that are indicative of brain tumors. These features may include the shape, size, and location of the tumor, as well as its relationship to surrounding brain tissue. One of the key advantages of DNNs is their ability to learn from vast amounts of data. By training on large datasets that contain both tumor and non-tumor images, DNNs can learn to identify subtle differences between the two and accurately distinguish between them. This ability to learn from large amounts of data also makes DNNs more robust to variations in imaging conditions, such as differences in scanner hardware or patient positioning. Another advantage of DNNs is their ability to incorporate multiple imaging modalities into their analysis. For example, a DNN can integrate information from medical images to improve its accuracy in detecting and diagnosing brain tumors. Overall, DNNs are effective in brain tumor detection and diagnosis because they can automatically learn and extract features from large volumes of brain medical imaging data, and they can integrate information from multiple imaging modalities to improve their accuracy.
CNNs are designed to process images and can automatically learn and extract features from brain imaging data to enhance their accuracy in detecting and diagnosing brain tumors [19]. The features in brain imaging data are analyzed in CNNs through a series of convolutional layers [20,21,22]. Each convolutional layer has a collection of filters that convolutionally float across the input image. These filters detect edges, corners, and textures. Each convolutional layer produces feature maps that show where these features are in the input image. To learn higher-level characteristics more pertinent to the goal of brain tumor identification and diagnosis, the feature maps are then sent through one or more fully connected layers after the convolutional layers. These fully linked layers employ convolutional layer characteristics to predict brain tumor presence or absence. During the training phase, the CNN learns to identify patterns in the brain imaging data that are indicative of brain tumors. This is performed by adjusting the weights of the filters in the convolutional layers and the fully connected layers through a process called back-propagation. By adjusting the weights of the network in this way, the CNN learns to identify patterns in the brain imaging data that are associated with brain tumors. These patterns may include the shape, size, and location of the tumor, as well as its relationship to surrounding brain tissue. Overall, CNNs use convolutional layers to analyze the features in brain imaging data and learn patterns that are indicative of brain tumors. By adjusting the weights of the network during training, CNNs can learn to accurately detect and diagnose brain tumors in medical images.
There have been many attempts carried out by researchers to improve the accuracy of identifying brain tumors. In assessing the clinical importance of each MR sequence in brain tumor prediction and detection, the prior studies are highly helpful and are a great resource. Imaging modalities, such as MRI and CT, were utilized by the researchers in order to discern between tumors and other abnormalities. MR spectroscopy has been employed for the categorization of brain tumors in a wide variety of studies. In particular, spectroscopic and magnetic resonance imaging were used in a study; Wang et al. [23] tried to differentiate benign from malignant brain neoplasms by applying a decision tree algorithm. In contrast, spectroscopic and perfusion magnetic resonance imaging were used in the study by Weber et al. [24] to evaluate the inherent heterogeneity of brain neoplasms by defining four regions of interest in the tumoral and peritumoral regions. The SVM classifier was used by Qin et al. [25] to detect brain tumors. In the first step, characteristics of MR images of brain tumors were extracted using the HOG algorithm and then compared to those extracted using wavelet transform. As a second classification method, they employed SVM with the a-norm loss function. Since it was sparse, it could be detected much more quickly. Several learning algorithms, including the Gaussian mixture model (GMM), the Random Forests classifier (RFC), and the K-means clustering and its derivatives, were presented for the diagnosis of brain tumors by Gopal et al. [26]. The support vector machine was tried and tested by Mathew et al. [27] and Amin et al. [28]. The above approach is a well-solved problem for identifying brain tumors, although it can be made more computationally efficient. Parallel to this, deep learning models have proven to be more effective. Due to the nature of the deep network model architecture, deep learning techniques such as convolutional neural networks (CNNs) need a substantial amount of training data consisting of images of brain tumors [29].
Devnath et al. [30] proposed an ensemble approach with deep features from CheXNet-121, which enhanced pneumoconiosis detection in chest X-rays (CXRs), achieving 92.68% accuracy, 85.66% MCC, and 0.9302 PR AUC, surpassing single model and other techniques; Grad-CAM visualization confirmed over 90% detection accuracy. Devnath et al. [31] introduced ensemble techniques for automating pneumoconiosis detection in coal miners through chest X-rays, achieving a robust accuracy of 91.50% by combining deep learning models, surpassing existing methods. Saeedi et al. [32] trained many ML models to classify brain tumors and achieved accuracies from 95.63% to 96.47%. Sobhaninia et al. [33] proposed a model which simultaneously segmented and classified the brain tumor. Their model achieved an accuracy of around 96.27 for classifying brain tumors, whereas it achieved 97.98% accuracy for segmentation. Zulfiqar et al. [34] proposed a model based on EfficientNet for classifying brain tumors based on MR images. They incorporated transfer learning to improve the accuracy of the proposed model. They further added some layers to enhance the model in terms of accuracy and achieved 98.86% accuracy. In another study [35], authors tried to predict the age of the patients with malignant brain tumors. They first performed the segmentation and then performed feature extraction. They used MRI brain images and improved the accuracy by 33%, compared to existing research works.
This study introduces a novel system that offers several advantages in the context of brain cancer forecasting. The main contributions and characteristics of the proposed system are as follows:
  • Ensemble model with convolutional features: The research proposes an ensemble model that combines convolutional features extracted from a specialized convolutional neural network (CNN). The ensemble model incorporates a voting mechanism, employing logistic regression and a stochastic gradient descent classifier to generate a final prediction.
  • Comparison of convolutional features: The effectiveness of models utilizing convolutional features is compared to the impact of using the original characteristics of the data. This analysis provides insights into the benefits and performance improvements offered by the convolutional features in the context of brain cancer forecasting.
  • Evaluation of multiple ML models: The study evaluates the performances of various machine learning (ML) models, including Random Forest (RF), K-Nearest Neighbor (k-NN), decision tree (DT), and Extreme Gradient Boost (XG-Boost). The results of these models are compared to assess their effectiveness in the classification task.
  • Ensemble model experimentation: Among the evaluated ML models, the three best-performing models are selected for further experimentation as an ensemble model. This ensemble approach aims to leverage the strengths of multiple models to improve the classification accuracy of the system.
  • Performance comparison with other ML methods: The proposed framework’s performance is assessed in terms of accuracy, precision, memory usage, and the F1-score. The evaluation includes a comparison with other ML methods to determine the superiority of the proposed framework in the context of brain cancer forecasting.
By employing an ensemble model with convolutional features and conducting a comprehensive evaluation of various ML models, this work provides insights into the effectiveness and performance of different approaches for brain cancer forecasting. The proposed framework’s performance is thoroughly assessed and compared to other ML methods, providing a comprehensive analysis of its capabilities.

2. Materials and Methods

In this section, the proposed framework for brain tumor prediction is discussed using the MRI dataset, as shown in Figure 1. The ML classifiers utilized in this work are also briefly described in this section.
The convolutional operations of CNN are initially utilized in order to extract the features. In addition, a dimensionality reduction method is then used to pick a limited collection of effective features for classification in order to increase the generalization ability and performance of the classifier. This is performed in order to make the classifier more efficient. The dataset and the data will be discussed in this part of the article.
After that, the pre-processing of the data, feature extraction, and dimensionality reduction are detailed, and then the classification algorithms that were employed are presented.

2.1. Dataset Description

This study made use of the “Brain tumor” dataset, which is publicly available on figshare [36]. The dataset contained 3762 instances. The target features contained two classes: tumors and non-tumor, of which 2079 belonged to the non-tumor class (Healthy), and 1683 belonged to the tumor class. Class-wise distributions of the sample are given in Table 1. Machine learning and deep learning models require a good amount of data to train them [37]. From Table 1, it can be seen that the dataset was balanced. A few sample images of brain tumors and normal images are displayed in Figure 2. Input data from the MRI scans were transformed into NumPy arrays and used by the model. A NumPy array is an n-dimensional array that can be expressed by a grid of values with non-negative number tuples.

2.2. Convolutional Neural Network for Feature Engineering

Conventional ML algorithms can be effective in detecting brain tumors from imaging data by learning patterns from large datasets. Feature extraction and selection are critical steps in developing a successful machine learning model, and various approaches can be used to identify relevant features for training the model [38]. Algorithms are trained to differentiate between images of a healthy brain and a brain with a tumor, based on various features, such as shape, +size, texture, and intensity [39]. Feature extraction and selection are crucial steps in developing a machine learning model for detecting brain tumors. In this context, feature extraction refers to the process of identifying relevant features from the imaging data, while feature selection involves choosing the most informative subset of features that can help the model perform well [40].

2.3. Principal Component Analysis (PCA)

There are various approaches to feature selection in ML algorithms for brain imaging. One common approach, such as PCA, is to reduce the dimensionality of the data and extract relevant features [41]. Another approach is to use domain-specific knowledge to select features that are known to be relevant for tumor detection, such as the size and shape of the tumor or the intensity values of the image. Once the features have been extracted or selected, they are used as inputs to train a machine learning model. During the training process, the model adjusts its parameters to learn the relationship between the input features and the corresponding output (i.e., tumor or no tumor).
After pre-processing, the number of features obtained was more than 23,000 for a set of 3712 brain MRI images. Therefore, optimization was performed using PCA to overcome the curse of dimensionality, removing data inconsistencies and redundant data and finding correlations among different features to create a new data space with fewer features that retain most relevant features. After using PCA, 12,000 features were selected to reduce the dimension for better performance. The stepwise functionality of PCA is represented in Figure 3. PCA transformed the data into a set of significantly lower dimensions without the loss of relevant information that accounted to final output.

2.4. Methods

The following machine learning models were explored as part of this work to perform analyses for improved brain tumor prediction using MRI data.

2.4.1. Adaptive (Ada-Boost) Boosting

Ada-Boost is one of the earliest ensemble-boosting algorithms [42]. The weighting instances of the dataset serve as Ada-Boost’s framework. It begins each observation with the same weights. The method is trained using the observation’s weights, and a feeble classifier is produced. Using the coefficient, we evaluated the performance of the limb. When data were incorrectly categorized, weights were increased, and when data were appropriately categorized, weights were decreased. This procedure was repeated to generate a classifier for weighted data using the weak learning method.

2.4.2. Extreme Gradient (XG-Boost) Boosting

XG-Boost is an extremely effective and adaptable method for distributed gradient boosting. XG-Boost is a framework for gradient boosting that integrates machine learning classifiers [43]. It employs a parallel tree-boosting technique to manage a variety of machine learning issues efficiently and effectively. During training and testing, the XG-Boost classifier is extremely fast. In addition, the regularization parameter effectively reduces variance, thereby improving model performance.

2.4.3. The K-Nearest Neighbor (K-NN)

Altman first introduced the non-parametric technique for data classification in 1991 [44]. Initially, the K-NN method was designed to address both classification and regression issues [45]. Similarity measure, also known as closeness, proximity, or distance, was the guiding principle behind the development of K-NN.

2.4.4. Decision Tree (DT)

In the category of non-parametric supervised learning is the decision tree classifier. The decision node and the root node are the two branches that emerge from the node to serve the inner node [27]. The leaf nodes represent pure subsets that are generated in accordance with the decision rule.

2.4.5. Random Forest (RF)

Friedl et al. [45] created the Random Forest classifier in 1997 by combining multiple decision trees. This technique of ensemble learning employs the bootstrap bagging or aggregation technique to trees [46]. Using arbitrarily selected data (a subset of data), the RF classifier generates a cluster of trees by aggregating the results of several trees, or decision trees. The final classification of the input is determined by combining the results of multiple decision trees and aggregating the majority ballots from various decision trees.

3. Proposed Novel Ensemble Approach: XG-Boost_Ada-Boost Random-Forest (XG-Ada-RF)

Ensemble is the idea of wrapping up the basic models to make the average prediction, which is more reliable in comparison with a single prediction with single models [47]. It works based on the majority voting. The ensemble voting scheme has made it so that error due to the independent classifier is overcome [48,49]. The ensemble technique offers improved accuracy by using intelligent computational models, as illustrated in the Figure 4. Each model contributes its expertise, and the ensemble aims to leverage the strengths of each individual model while compensating for their weaknesses.
This study applied an ensemble Ada-Boost, XG-Boost, and Random Forest (XG-Ada-RF), based on their individual performances for the binary classification of brain MRIs for predicting brain tumors. The detailed architecture of the proposed ensemble XG-Ada-RF model is shown in Figure 5. The pseudocode for the proposed study is given in Algorithm 1. The ensemble approach increases the performance by combining the advantages of more than one classifier.
The first step involves training three individual machine learning models: XG-Boost, Ada-Boost, and Random Forest. Each model is trained on the same brain tumor classification dataset, using a combination of labeled brain MRI images and corresponding tumor labels. Once the individual models are trained, they are used to make predictions on the test data. Each model generates its set of predictions, indicating whether each test image is classified as a tumor or normal. The ensemble combines the predictions from the three individual models using a weighted voting approach. Each model’s prediction is assigned a weight based on its performance and reliability during the training phase. The models with higher accuracy or lower error rates may receive higher weights, indicating their greater influence on the ensemble’s final decision. The ensemble aggregates the weighted predictions from the individual models. This can be performed using a simple majority vote, where the class with the most votes is considered the final prediction. The aggregated prediction is considered the ensemble’s final decision for each test image. This decision is based on the combined expertise of the individual models, taking into account their diverse decision-making processes and generalization capabilities.
Algorithm 1: Procedure Ensemble Model ( X G A d a R F )
Input: The training dataset ( d t r a i n ) , the test dataset ( d t e s t ) , the input shape ( i n p )
Output: The output is classified in two categories: normal and brain tumor, and the model will return the results based on a c c u r a c y ,   p r e c i s i o n ,   r e c a l l ,   f b e t a _ s c o r e

1: D a t a _ P r e p r o c e s s i n g , Class labelling, Data-resampling
2: F e a t u r e E x t r a c t i o n C N N C o n v o l u t i o n L a y e r s
3: m o d e l _ c n n . a d d _ M a x P o o l i n g 2 D ( p o o l _ s i z e ,   s t r i d e )
4: m o d e l _ c n n . a d d ( D r o p o u t ( r a t e ) )
5: m o d e l _ c n n . a d d ( F l a t t e n ( ) )
6: m o d e l s _ ( A d a B o o s t , X G B o o s t , D T , K N N , R F )
7: P r o p o s e d _ E n s e m b l e _ m o d e l _ X G A d a R F ( )
8:   a c c u r a c y , p r e c i s i o n ,   r e c a l l ,   f b e t a _ s c o r e , a u c m o d e l _ E v a l u a t e ( d t e s t )
9: Return a c c u r a c y , p r e c i s i o n ,   r e c a l l ,   f b e t a _ s c o r e ,   a u c

Experimental Environment Settings and Performance Evaluation Metrics

The experimentation setup required a 64 GB RAM Intel i7 11th Generation processor and an 8 GB Nvidia RTX3070 GPU. For parallel computing, 16 cores were utilized using threaded gradient descent. Tensorflow 3.1 and Keras were used as the backbones, providing the modeling libraries. Aside from the hardware requirements, Anaconda, Python, and VS Code were used to boost additional processing power. After the preprocessing phase was completed, the image volumetric blob was passed to the sparse volume convolution to increase the channeling output.
Several metrics were employed to evaluate the performances of the models, including accuracy, precision, recall, and F1-score, which are frequently used indicators.
A c c u r a c y = T r u e   P o s i t i v e + T r u e   N e g a t i v e T r u e   P o s i t i v e + T r u e   N e g a t i v e + F a l s e   P o s i t i v e + F a l s e   N e g a t i v e
R e c a l l = T r u e   P o s i t i v e T r u e   P o s i t i v e + F a l s e   N e g a t i v e
P r e c i s i o n = T P T r u e   P o s i t i v e + F a l s e   P o s i t i v e
F 1 S c o r e = 2 × R e c a l l × P r e c i s i o n R e c a l l + P r e c i s i o n

4. Results and Discussions

The objective of this study is to identify the effectiveness of different algorithms for accurate classification and with minimum log loss. Accuracy represents the degree to which a measurement corresponds to its actual value. Sensitivity mathematically represents the values by calculating the number of correct positive predictions divided by the total number of positive values. Specificity mathematically represents the value by calculating the number of correct negative predictions divided by the total number of negative values. Sensitivity and specificity are statistical measures of the performance of the test results.
Table 2 displays the classification results, including accuracy, sensitivity, specificity, and loss, for the various classification algorithms reported in this study. Different classification accuracy performances on testset were observed for various algorithms, based on the observed results. Several machine learning and proposed ensemble models were applied on the brain MRI dataset for the classification, but out of them all, the proposed approach stood out from other models. Upon comparing the performance evaluations of ML and EL techniques, Ada-Boost yielded a great accuracy of 93.12% among all models and also attained great results in terms of precision, recall, and F1-score, with the lowest loss compared to other models. The three best-performing learners were than combined in an ensemble approach, such as Ada-Boost, XG-Boost, and Random Forest, which performed great on the dataset, in contrast to machine learning models. The ensemble proposed model made better predictions. The graphical representation for comparing the performance of each model is shown in Figure 6.
The efficacy of the proposed XG-Ada-RF model was evaluated using a 2 × 2 confusion matrix, where 2 is the total number of target classes. As shown in Figure 7, the matrix compared the actual target values to those predicted by the XG-Ada-RF model. The receiver operating characteristic curve (ROC) plots are given in Figure 8.
To detect brain tumors early, the study employed brain MRI data and effective learning paradigms. CNNs have made tremendous advances in visual feature extraction tasks, making them suitable for our goal. By utilizing deep convolutional features, we aimed to enhance the precision of brain tumor and non-tumor patient classifications. These learned features conveyed critical visual information, allowing the model to detect tiny brain tumor patterns and fluctuations. Authors experimented with five ML models to broaden our understanding of their capabilities and significance in brain tumor classification. The models evaluated Extreme Gradient Boosting, Ada-Boost, and Random Forest, among others. We compared their individual performances to identify the most promising candidates for our proposed ensemble approach. An ensemble of three high-performing individual ML models, Extreme Gradient Boosting, Ada-Boost, and Random Forest (XG-Ada-RF), was proposed. The ensemble approach aimed to leverage the strengths of these individual models while mitigating their weaknesses. The ensemble took advantage of the diverse behaviors and decision-making processes of the constituent models, resulting in more robust and accurate predictions. The proposed ensemble approach achieved a high level of accuracy in binary class classification for detecting brain tumors, and the results validated the significance of the ensemble approach.
Furthermore, we compared our study with the techniques used in previous studies, as shown in Table 3. The table encapsulates a compilation of studies concerning brain tumor classification via MRI scans. Each study employed distinct methodologies to attain accurate results. One study employed a neural network with a back-propagation technique, achieving a high accuracy rate. Another study adopted a Naïve Bayes classifier approach, obtaining a notable accuracy score. Ensemble learning was harnessed in another study, demonstrating competitive accuracy. The support vector machine (SVM) technique was used in a study, resulting in a remarkable accuracy rate. Additionally, a study combined a saliency map and deep learning strategies, achieving a robust accuracy. Lastly, the proposed model introduced a novel ensemble XG-Ada-RF approach, yielding a commendable accuracy rate. These studies collectively underscored the diverse techniques employed in advancing the field of brain tumor classification through MRI data analysis, with potential implications for medical diagnosis and treatment strategies.

5. Conclusions

This paper provides a comprehensive review of the research on the detection and classification of brain tumor MRI images using ML algorithms, specifically focusing on the use of an ensemble of the three best-performing models. Through a critical analysis of conventional ML models, this study offers a detailed examination of their strengths and limitations. The proposed ensemble model, XG-Ada-RF, demonstrates superior predictive performance, with an accuracy of 94.9% for the healthy class and 95.9% for the tumor class. This outcome highlights the effectiveness of the ensemble approach in accurately predicting and classifying brain tumors from medical image data. Building upon this research, a comprehensive framework based on an ensemble of three classifiers was developed to achieve accurate predictions and classifications. This framework serves as a valuable tool for enhancing our understanding and expanding the scope and significance of brain tumor classification.
Furthermore, an experimental comparative performance analysis was conducted to assess the robustness of the various algorithms and the ensemble model. Key evaluation metrics, such as accuracy, sensitivity, specificity, and log loss, were considered to provide a comprehensive evaluation. The diverse results obtained by implementing the aforementioned algorithms on the given dataset further supported the objectives of this study.
Looking ahead, future research directions could focus on refining and optimizing the ensemble model, exploring additional ML algorithms, and incorporating advanced techniques, such as deep learning. Additionally, further investigation into the interpretability and explaining ability of the ensemble model could provide valuable insights for clinical decision-making. Overall, the results of this study emphasize the superiority of the ensemble approach in brain tumor classification, laying a foundation for further advancements in this field.

Author Contributions

Conceptualization, F.K., S.A., Y.G., M.M., F.A.R., M.S.M., A.B.S. and O.E.; data curation, M.M.; formal analysis, Y.G.; funding acquisition, Y.G.; investigation, Y.G.; methodology, F.K., S.A., Y.G., M.M., F.A.R., M.S.M., A.B.S. and O.E.; project administration, Y.G.; resources, Y.G.; software, S.A. and Y.G.; supervision, S.A., Y.G. and M.M.; validation, F.K., S.A. and Y.G.; visualization, Y.G.; writing—original draft, F.K., S.A., Y.G. and M.M.; writing—review and editing, Y.G., F.A.R., M.S.M., A.B.S. and O.E. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported by the Deanship of Scientific Research, Vice Presidency for Graduate Studies and Scientific Research, King Faisal University, Saudi Arabia, under the Project GRANT3,912.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

A public dataset was used in this research work [36].

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Bondy, M.L.; Scheurer, M.E.; Malmer, B.; Barnholtz-Sloan, J.S.; Davis, F.G.; Il’yasova, D.; Kruchko, C.; McCarthy, B.J.; Rajaraman, P.; Schwartzbaum, J.A.; et al. Brain Tumor Epidemiology: Consensus from the Brain Tumor Epidemiology Consortium. Cancer 2008, 113, 1953–1968. [Google Scholar] [CrossRef]
  2. Mabray, M.C.; Barajas, R.F.; Cha, S. Modern Brain Tumor Imaging. Brain Tumor Res. Treat. 2015, 3, 8–23. [Google Scholar] [CrossRef]
  3. Ayoub, S.; Behera, N.R.; Raju, M.N.; Singh, P.; Praveena, S.; Ravikiran, K. Hyperparameter Tuned Deep Learning Model for Healthcare Monitoring System in Big Data. In Proceedings of the IDCIoT 2023—International Conference on Intelligent Data Communication Technologies and Internet of Things, Proceedings, Bengaluru, India, 5–7 January 2023. [Google Scholar]
  4. Ayoub, S.; Khan, M.A.; Jadhav, V.P.; Anandaram, H.; Anil Kumar, T.C.; Reegu, F.A.; Motwani, D.; Shrivastava, A.K.; Berhane, R. Minimized Computations of Deep Learning Technique for Early Diagnosis of Diabetic Retinopathy Using IoT-Based Medical Devices. Comput. Intell. Neurosci. 2022, 2022, 7040141. [Google Scholar] [CrossRef]
  5. Londhe, V.Y.; Bhasin, B. Artificial Intelligence and Its Potential in Oncology. Drug Discov. Today 2019, 24, 228–232. [Google Scholar] [CrossRef]
  6. Alam, S.; Raja, P.; Gulzar, Y. Investigation of Machine Learning Methods for Early Prediction of Neurodevelopmental Disorders in Children. Wirel. Commun. Mob. Comput. 2022, 2022, 5766386. [Google Scholar] [CrossRef]
  7. Borole, V.Y.; Nimbhore, S.S.; Kawthekar, D.S.S. Image Processing Techniques for Brain Tumor Detection: A Review. Int. J. Emerg. Trends Technol. Comput. Sci. 2015, 4, 1–28. [Google Scholar]
  8. Gulzar, Y. Fruit Image Classification Model Based on MobileNetV2 with Deep Transfer Learning Technique. Sustainability 2023, 15, 1906. [Google Scholar] [CrossRef]
  9. Gulzar, Y.; Hamid, Y.; Soomro, A.B.; Alwan, A.A.; Journaux, L. A Convolution Neural Network-Based Seed Classification System. Symmetry 2020, 12, 2018. [Google Scholar] [CrossRef]
  10. Albarrak, K.; Gulzar, Y.; Hamid, Y.; Mehmood, A.; Soomro, A.B. A Deep Learning-Based Model for Date Fruit Classification. Sustainability 2022, 14, 6339. [Google Scholar] [CrossRef]
  11. Mamat, N.; Othman, M.F.; Abdulghafor, R.; Alwan, A.A.; Gulzar, Y. Enhancing Image Annotation Technique of Fruit Classification Using a Deep Learning Approach. Sustainability 2023, 15, 901. [Google Scholar] [CrossRef]
  12. Dhiman, P.; Kaur, A.; Balasaraswathi, V.R.; Gulzar, Y.; Alwan, A.A.; Hamid, Y. Image Acquisition, Preprocessing and Classification of Citrus Fruit Diseases: A Systematic Literature Review. Sustainability 2023, 15, 9643. [Google Scholar] [CrossRef]
  13. Sahlan, F.; Hamidi, F.; Misrat, M.Z.; Adli, M.H.; Wani, S.; Gulzar, Y. Prediction of Mental Health Among University Students. Int. J. Perceptive Cogn. Comput. 2021, 7, 85–91. [Google Scholar]
  14. Gulzar, Y.; Alwan, A.A.; Abdullah, R.M.; Abualkishik, A.Z.; Oumrani, M. OCA: Ordered Clustering-Based Algorithm for E-Commerce Recommendation System. Sustainability 2023, 15, 2947. [Google Scholar] [CrossRef]
  15. Mehmood, A.; Gulzar, Y.; Ilyas, Q.M.; Jabbari, A.; Ahmad, M.; Iqbal, S. SBXception: A Shallower and Broader Xception Architecture for Efficient Classification of Skin Lesions. Cancers 2023, 15, 3604. [Google Scholar] [CrossRef]
  16. Siar, M.; Teshnehlab, M. Brain Tumor Detection Using Deep Neural Network and Machine Learning Algorithm. In Proceedings of the 2019 9th International Conference on Computer and Knowledge Engineering, ICCKE 2019, Mashhad, Iran, 24–25 October 2019. [Google Scholar]
  17. Seetha, J.; Raja, S.S. Brain Tumor Classification Using Convolutional Neural Networks. Biomed. Pharmacol. J. 2018, 11, 1457–1461. [Google Scholar] [CrossRef]
  18. Gulzar, Y.; Ünal, Z.; Akta¸s, H.A.; Mir, M.S. Harnessing the Power of Transfer Learning in Sunflower Disease Detection: A Comparative Study. Agriculture 2023, 13, 1479. [Google Scholar] [CrossRef]
  19. Choudhury, C.L.; Mahanty, C.; Kumar, R.; Mishra, B.K. Brain Tumor Detection and Classification Using Convolutional Neural Network and Deep Neural Network. In Proceedings of the 2020 International Conference on Computer Science, Engineering and Applications, ICCSEA 2020, Gunupur, India, 13–14 March 2020. [Google Scholar]
  20. Gulzar, Y.; Khan, S.A. Skin Lesion Segmentation Based on Vision Transformers and Convolutional Neural Networks— A Comparative Study. Appl. Sci. 2022, 12, 5990. [Google Scholar] [CrossRef]
  21. Ayoub, S.; Gulzar, Y.; Reegu, F.A.; Turaev, S. Generating Image Captions Using Bahdanau Attention Mechanism and Transfer Learning. Symmetry 2022, 14, 2681. [Google Scholar] [CrossRef]
  22. Khan, S.A.; Gulzar, Y.; Turaev, S.; Peng, Y.S. A Modified HSIFT Descriptor for Medical Image Classification of Anatomy Objects. Symmetry 2021, 13, 1987. [Google Scholar] [CrossRef]
  23. Wang, Q.; Liacouras, E.K.; Miranda, E.; Kanamalla, U.S.; Megalooikonomou, V. Classification of Brain Tumors Using MRI and MRS Data. In Proceedings of the Medical Imaging 2007: Computer-Aided Diagnosis, San Diego, CA, USA, 29 March 2007; Volume 6514. [Google Scholar]
  24. Weber, M.A.; Zoubaa, S.; Schlieter, M.; Jüttler, E.; Huttner, H.B.; Geletneky, K.; Ittrich, C.; Lichy, M.P.; Kroll, A.; Debus, J.; et al. Diagnostic Performance of Spectroscopic and Perfusion MRI for Distinction of Brain Tumors. Neurology 2006, 66, 1899–1906. [Google Scholar] [CrossRef]
  25. Qin, C.; Li, B.; Han, B. Fast Brain Tumor Detection Using Adaptive Stochastic Gradient Descent on Shared-Memory Parallel Environment. Eng. Appl. Artif. Intell. 2023, 120, 105816. [Google Scholar] [CrossRef]
  26. Gopal, N.N.; Karnan, M. Diagnose Brain Tumor through MRI Using Image Processing Clustering Algorithms Such as Fuzzy C Means along with Intelligent Optimization Techniques. In Proceedings of the 2010 IEEE International Conference on Computational Intelligence and Computing Research, ICCIC 2010, Coimbatore, India, 28–29 December 2010. [Google Scholar]
  27. Mathew, A.R.; Anto, P.B. Tumor Detection and Classification of MRI Brain Image Using Wavelet Transform and SVM. In Proceedings of the Proceedings of IEEE International Conference on Signal Processing and Communication, ICSPC 2017, Coimbatore, India, 28–29 July 2017; Volume 2018. [Google Scholar]
  28. Amin, J.; Sharif, M.; Gul, N.; Yasmin, M.; Shad, S.A. Brain Tumor Classification Based on DWT Fusion of MRI Sequences Using Convolutional Neural Network. Pattern Recognit. Lett. 2020, 129, 115–122. [Google Scholar] [CrossRef]
  29. Anand, V.; Gupta, S.; Gupta, D.; Gulzar, Y.; Xin, Q.; Juneja, S.; Shah, A.; Shaikh, A. Weighted Average Ensemble Deep Learning Model for Stratification of Brain Tumor in MRI Images. Diagnostics 2023, 13, 1320. [Google Scholar] [CrossRef]
  30. Devnath, L.; Fan, Z.; Luo, S.; Summons, P.; Wang, D. Detection and Visualisation of Pneumoconiosis Using an Ensemble of Multi-Dimensional Deep Features Learned from Chest X-Rays. Int. J. Environ. Res. Public Health 2022, 19, 11193. [Google Scholar] [CrossRef] [PubMed]
  31. Devnath, L.; Luo, S.; Summons, P.; Wang, D.; Shaukat, K.; Hameed, I.A.; Alrayes, F.S. Deep Ensemble Learning for the Automatic Detection of Pneumoconiosis in Coal Worker’s Chest X-Ray Radiography. J. Clin. Med. 2022, 11, 5342. [Google Scholar] [CrossRef]
  32. Saeedi, S.; Rezayi, S.; Keshavarz, H.; Niakan Kalhori, S.R. MRI-Based Brain Tumor Detection Using Convolutional Deep Learning Methods and Chosen Machine Learning Techniques. BMC Med. Inform. Decis. Mak. 2023, 23, 16. [Google Scholar] [CrossRef]
  33. Sobhaninia, Z.; Karimi, N.; Khadivi, P.; Samavi, S. Brain Tumor Segmentation by Cascaded Multiscale Multitask Learning Framework Based on Feature Aggregation; Elsevier: Amsterdam, The Netherlands, 2023. [Google Scholar]
  34. Zulfiqar, F.; Ijaz Bajwa, U.; Mehmood, Y. Multi-Class Classification of Brain Tumor Types from MR Images Using EfficientNets. Biomed. Signal Process. Control 2023, 84, 104777. [Google Scholar] [CrossRef]
  35. Rajput, S.; Kapdi, R.A.; Raval, M.S.; Roy, M. Interpretable Machine Learning Model to Predict Survival Days of Malignant Brain Tumor Patients. Mach. Learn. Sci. Technol. 2023, 4, 025025. [Google Scholar] [CrossRef]
  36. Brain Tumor Dataset. Available online: https://figshare.com/articles/dataset/brain_tumor_dataset/1512427 (accessed on 7 July 2023).
  37. Ayoub, S.; Gulzar, Y.; Rustamov, J.; Jabbari, A.; Reegu, F.A.; Turaev, S. Adversarial Approaches to Tackle Imbalanced Data in Machine Learning. Sustainability 2023, 15, 7097. [Google Scholar] [CrossRef]
  38. Wiatowski, T.; Bolcskei, H. A Mathematical Theory of Deep Convolutional Neural Networks for Feature Extraction. IEEE Trans. Inf. Theory 2018, 64, 1845–1866. [Google Scholar] [CrossRef]
  39. Namatēvs, I. Deep Convolutional Neural Networks: Structure, Feature Extraction and Training. Inf. Technol. Manag. Sci. 2018, 20, 40–47. [Google Scholar] [CrossRef]
  40. Weimer, D.; Scholz-Reiter, B.; Shpitalni, M. Design of Deep Convolutional Neural Network Architectures for Automated Feature Extraction in Industrial Inspection. CIRP Ann. Manuf. Technol. 2016, 65, 417–420. [Google Scholar] [CrossRef]
  41. Zebari, R.; Abdulazeez, A.; Zeebaree, D.; Zebari, D.; Saeed, J. A Comprehensive Review of Dimensionality Reduction Techniques for Feature Selection and Feature Extraction. J. Appl. Sci. Technol. Trends 2020, 1, 56–70. [Google Scholar] [CrossRef]
  42. Abolhasanzadeh, B. Nonlinear Dimensionality Reduction for Intrusion Detection Using Auto-Encoder Bottleneck Features. In Proceedings of the 2015 7th Conference on Information and Knowledge Technology, IKT 2015, Urmia, Iran, 26–28 May 2015. [Google Scholar]
  43. Jadhav, D.A. An Enhanced and Secured Predictive Model of Ada-Boost and Random-Forest Techniques in HCV Detections. Proc. Mater. Today Proc. 2021, 51, 186–195. [Google Scholar] [CrossRef]
  44. Chen, T.; He, T. R Lecture, Version 1.7.5.1. Xgboost: Extreme Gradient Boosting. 2014.
  45. Hughes, M.; Cooper, S.-M.; Nevill, A. Analysis Procedures for Non-Parametric Data from Performance Analysis. Int. J. Perform. Anal. Sport 2002, 2, 6–20. [Google Scholar] [CrossRef]
  46. Rodriguez-Galiano, V.F.; Ghimire, B.; Rogan, J.; Chica-Olmo, M.; Rigol-Sanchez, J.P. An Assessment of the Effectiveness of a Random Forest Classifier for Land-Cover Classification. ISPRS J. Photogramm. Remote Sens. 2012, 67, 93–104. [Google Scholar] [CrossRef]
  47. Zhang, Y.; Zhang, B.; Coenen, F.; Xiao, J.; Lu, W. One-Class Kernel Subspace Ensemble for Medical Image Classification. EURASIP J. Adv. Signal Process. 2014, 2014, 17. [Google Scholar] [CrossRef]
  48. Dogan, A.; Birant, D. A Weighted Majority Voting Ensemble Approach for Classification. In Proceedings of the UBMK 2019—Proceedings, 4th International Conference on Computer Science and Engineering, Samsun, Turkey, 11–15 September 2019. [Google Scholar]
  49. Khan, Y.F.; Kaushik, B.; Chowdhary, C.L.; Srivastava, G. Ensemble Model for Diagnostic Classification of Alzheimer’s Disease Based on Brain Anatomical Magnetic Resonance Imaging. Diagnostics 2022, 12, 3193. [Google Scholar] [CrossRef]
  50. Ismael, M.R.; Abdel-Qader, I. Brain Tumor Classification via Statistical Features and Back-Propagation Neural Network. In Proceedings of the IEEE International Conference on Electro Information Technology, Rochester, MI, USA, 3–5 May 2018; Volume 2018. [Google Scholar]
  51. Zaw, H.T.; Maneerat, N.; Win, K.Y. Brain Tumor Detection Based on Naïve Bayes Classification. In Proceedings of the Proceeding—5th International Conference on Engineering, Applied Sciences and Technology, ICEAST 2019, Luang Prabang, Laos, 2–5 July 2019. [Google Scholar]
  52. Kang, J.; Ullah, Z.; Gwak, J. MRI-Based Brain Tumor Classification Using Ensemble of Deep Features and Machine Learning Classifiers. Sensors 2021, 21, 2222. [Google Scholar] [CrossRef]
  53. Deepak, S.; Ameer, P.M. Automated Categorization of Brain Tumor from MRI Using CNN Features and SVM. J. Ambient Intell. Humaniz. Comput. 2021, 12, 8357–8369. [Google Scholar] [CrossRef]
  54. Khan, M.A.; Khan, A.; Alhaisoni, M.; Alqahtani, A.; Alsubai, S.; Alharbi, M.; Malik, N.A.; Damaševičius, R. Multimodal Brain Tumor Detection and Classification Using Deep Saliency Map and Improved Dragonfly Optimization Algorithm. Int. J. Imaging Syst. Technol. 2023, 33, 572–587. [Google Scholar] [CrossRef]
Figure 1. Workflow diagram of the proposed methodology.
Figure 1. Workflow diagram of the proposed methodology.
Jimaging 09 00163 g001
Figure 2. (a) Sample MRI 2D images for normal brain. (b) Sample MRI 2D images for tumor brain.
Figure 2. (a) Sample MRI 2D images for normal brain. (b) Sample MRI 2D images for tumor brain.
Jimaging 09 00163 g002
Figure 3. Step-by-step procedure of PCA for dimensionality reduction.
Figure 3. Step-by-step procedure of PCA for dimensionality reduction.
Jimaging 09 00163 g003
Figure 4. Ensemble voting.
Figure 4. Ensemble voting.
Jimaging 09 00163 g004
Figure 5. Proposed ensemble model (XG-Ada-RF).
Figure 5. Proposed ensemble model (XG-Ada-RF).
Jimaging 09 00163 g005
Figure 6. Performance comparison for brain tumor classification binary class for each model implemented in this work.
Figure 6. Performance comparison for brain tumor classification binary class for each model implemented in this work.
Jimaging 09 00163 g006
Figure 7. Confusion matrix for the XG-Ada-RF model.
Figure 7. Confusion matrix for the XG-Ada-RF model.
Jimaging 09 00163 g007
Figure 8. ROC for all the implemented models and the proposed XG-Ada-RF model.
Figure 8. ROC for all the implemented models and the proposed XG-Ada-RF model.
Jimaging 09 00163 g008
Table 1. Detailed sample-wise distribution of the dataset.
Table 1. Detailed sample-wise distribution of the dataset.
DatasetNormalBrain Tumor
Brain Images20791683
Table 2. Performance evaluation of classifiers for brain tumors.
Table 2. Performance evaluation of classifiers for brain tumors.
ModelClassAccuracy (%)PrecisionRecallF1-ScoreAUC
Ada-BoostHealthy92.510.88970.88970.88970.9732
Tumor93.120.87230.85560.87640.9723
XG-BoostHealthy92.560.93600.93600.93600.9821
Tumor91.400.87640.93740.92930.9635
K-NNHealthy88.250.93270.86210.93250.8861
Tumor89.980.87340.97480.83570.8767
DTHealthy89.750.72460.71190.79560.8349
Tumor89.460.83450.78970.82340.8645
RFHealthy91.330.94400.94060.94330.9318
Tumor89.990.92340.89450.92340.9344
Proposed Model
(XG-Ada-RF)
Healthy94.9%0.98320.94230.97640.9932
Tumor95.9%0.98880.95340.98230.9878
Table 3. Performance comparison with previous studies.
Table 3. Performance comparison with previous studies.
S. NoStudyMaterialTechniqueResult (%)
1.Ismael et al. [50]Brain MRIBack-propagation neural network91.9
2.Zaw et al. [51]Brain MRINaïve Bayes classifier94
3.Kang et al. [52]Brain MRIensemble learning91.58
4.Deepak et al. [53]Brain MRISVM95.82
5.Khan et al. [54]Brain MRIsaliency map and deep learning feature optimization94.89
6.Proposed ModelBrain MRIEnsemble XG-Ada-RF model95.9
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Khan, F.; Ayoub, S.; Gulzar, Y.; Majid, M.; Reegu, F.A.; Mir, M.S.; Soomro, A.B.; Elwasila, O. MRI-Based Effective Ensemble Frameworks for Predicting Human Brain Tumor. J. Imaging 2023, 9, 163. https://doi.org/10.3390/jimaging9080163

AMA Style

Khan F, Ayoub S, Gulzar Y, Majid M, Reegu FA, Mir MS, Soomro AB, Elwasila O. MRI-Based Effective Ensemble Frameworks for Predicting Human Brain Tumor. Journal of Imaging. 2023; 9(8):163. https://doi.org/10.3390/jimaging9080163

Chicago/Turabian Style

Khan, Farhana, Shahnawaz Ayoub, Yonis Gulzar, Muneer Majid, Faheem Ahmad Reegu, Mohammad Shuaib Mir, Arjumand Bano Soomro, and Osman Elwasila. 2023. "MRI-Based Effective Ensemble Frameworks for Predicting Human Brain Tumor" Journal of Imaging 9, no. 8: 163. https://doi.org/10.3390/jimaging9080163

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop