Enhancing brain tumor MRI classification with an ensemble of deep learning models and transformer integration

PeerJ Comput Sci. 2024 Nov 27:10:e2425. doi: 10.7717/peerj-cs.2425. eCollection 2024.

Abstract

Brain tumors are widely recognized as the primary cause of cancer-related mortality globally, necessitating precise detection to enhance patient survival rates. The early identification of brain tumor is presented with significant challenges in the healthcare domain, necessitating the implementation of precise and efficient diagnostic methodologies. The manual identification and analysis of extensive MRI data are presented as a challenging and laborious task, compounded by the importance of early tumor detection in reducing mortality rates. Prompt initiation of treatment hinges upon identifying the specific tumor type in patients, emphasizing the urgency for a dependable deep learning methodology for precise diagnosis. In this research, a hybrid model is presented which integrates the strengths of both transfer learning and the transformer encoder mechanism. After the performance evaluation of the efficacy of six pre-existing deep learning model, both individually and in combination, it was determined that an ensemble of three pretrained models achieved the highest accuracy. This ensemble, comprising DenseNet201, GoogleNet (InceptionV3), and InceptionResNetV2, is selected as the feature extraction framework for the transformer encoder network. The transformer encoder module integrates a Shifted Window-based Self-Attention mechanism, sequential Self-Attention, with a multilayer perceptron layer (MLP). These experiments were conducted on three publicly available research datasets for evaluation purposes. The Cheng dataset, BT-large-2c, and BT-large-4c dataset, each designed for various classification tasks with differences in sample number, planes, and contrast. The model gives consistent results on all three datasets and reaches an accuracy of 99.34%, 99.16%, and 98.62%, respectively, which are improved compared to other techniques.

Keywords: Brain tumor detection; Deep learning; Ensemble learning; MRI data analysis; Medical imaging; Multilayer perceptron (MLP); Pretrained models; Shifted window-based self-attention; Transfer learning; Transformer encoder.

Associated data

  • figshare/10.6084/m9.figshare.1512427.v5

Grants and funding

This work was supported by the National Natural Science Foundation of China (No. 42075129), the Hebei Province Natural Science Foundation (No. E2021202179), the Key Research and Development Project from Hebei Province (No. 21351803D) and the Hebei special project for key technology and product R&D (No. SJMYF2022Y06). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.