Saliency Map and Deep Learning in Binary Classification of Brain Tumours

Sensors (Basel). 2023 May 7;23(9):4543. doi: 10.3390/s23094543.

Abstract

The paper was devoted to the application of saliency analysis methods in the performance analysis of deep neural networks used for the binary classification of brain tumours. We have presented the basic issues related to deep learning techniques. A significant challenge in using deep learning methods is the ability to explain the decision-making process of the network. To ensure accurate results, the deep network being used must undergo extensive training to produce high-quality predictions. There are various network architectures that differ in their properties and number of parameters. Consequently, an intriguing question is how these different networks arrive at similar or distinct decisions based on the same set of prerequisites. Therefore, three widely used deep convolutional networks have been discussed, such as VGG16, ResNet50 and EfficientNetB7, which were used as backbone models. We have customized the output layer of these pre-trained models with a softmax layer. In addition, an additional network has been described that was used to assess the saliency areas obtained. For each of the above networks, many tests have been performed using key metrics, including statistical evaluation of the impact of class activation mapping (CAM) and gradient-weighted class activation mapping (Grad-CAM) on network performance on a publicly available dataset of brain tumour X-ray images.

Keywords: CAM; Grad-CAM; binary brain tumour classification; convolutional neural networks; deep learning.

MeSH terms

  • Brain Neoplasms* / diagnostic imaging
  • Deep Learning*
  • Humans
  • Neural Networks, Computer

Grants and funding

This research was funded by the research subsidy for AGH University of Science and Technology.