Robust brain tumor classification by fusion of deep learning and channel-wise attention mode approach

BMC Med Imaging. 2024 Jun 17;24(1):147. doi: 10.1186/s12880-024-01323-3.

Abstract

Diagnosing brain tumors is a complex and time-consuming process that relies heavily on radiologists' expertise and interpretive skills. However, the advent of deep learning methodologies has revolutionized the field, offering more accurate and efficient assessments. Attention-based models have emerged as promising tools, focusing on salient features within complex medical imaging data. However, the precise impact of different attention mechanisms, such as channel-wise, spatial, or combined attention within the Channel-wise Attention Mode (CWAM), for brain tumor classification remains relatively unexplored. This study aims to address this gap by leveraging the power of ResNet101 coupled with CWAM (ResNet101-CWAM) for brain tumor classification. The results show that ResNet101-CWAM surpassed conventional deep learning classification methods like ConvNet, achieving exceptional performance metrics of 99.83% accuracy, 99.21% recall, 99.01% precision, 99.27% F1-score and 99.16% AUC on the same dataset. This enhanced capability holds significant implications for clinical decision-making, as accurate and efficient brain tumor classification is crucial for guiding treatment strategies and improving patient outcomes. Integrating ResNet101-CWAM into existing brain classification software platforms is a crucial step towards enhancing diagnostic accuracy and streamlining clinical workflows for physicians.

Keywords: Attention mechanism; Brain tumor; CWAM; Deep learning; ResNet101.

MeSH terms

  • Brain Neoplasms* / classification
  • Brain Neoplasms* / diagnostic imaging
  • Brain Neoplasms* / pathology
  • Deep Learning*
  • Humans
  • Image Interpretation, Computer-Assisted / methods
  • Magnetic Resonance Imaging / methods