Spectral-Spatial Features Integrated Convolution Neural Network for Breast Cancer Classification

Sensors (Basel). 2020 Aug 22;20(17):4747. doi: 10.3390/s20174747.

Abstract

Cancer identification and classification from histopathological images of the breast depends greatly on experts, and computer-aided diagnosis can play an important role in disagreement of experts. This automatic process has increased the accuracy of the classification at a reduced cost. The advancement in Convolution Neural Network (CNN) structure has outperformed the traditional approaches in biomedical imaging applications. One of the limiting factors of CNN is it uses spatial image features only for classification. The spectral features from the transform domain have equivalent importance in the complex image classification algorithm. This paper proposes a new CNN structure to classify the histopathological cancer images based on integrating the spectral features obtained using a multi-resolution wavelet transform with the spatial features of CNN. In addition, batch normalization process is used after every layer in the convolution network to improve the poor convergence problem of CNN and the deep layers of CNN are trained with spectral-spatial features. The proposed structure is tested on malignant histology images of the breast for both binary and multi-class classification of tissue using the BreaKHis Dataset and the Breast Cancer Classification Challenge 2015 Datasest. Experimental results show that the combination of spectral-spatial features improves classification accuracy of the CNN network and requires less training parameters in comparison with the well known models (i.e., VGG16 and ALEXNET). The proposed structure achieves an average accuracy of 97.58% and 97.45% with 7.6 million training parameters on both datasets, respectively.

Keywords: biomedical imaging; breast cancer classification; convolutional neural network; deep learning; wavelet transform.

MeSH terms

  • Algorithms
  • Breast
  • Breast Neoplasms* / classification
  • Breast Neoplasms* / diagnostic imaging
  • Female
  • Humans
  • Neural Networks, Computer*
  • Wavelet Analysis