Modality Specific CBAM-VGGNet Model for the Classification of Breast Histopathology Images via Transfer Learning
Date
Journal Title
Journal ISSN
Volume Title
Publisher
Abstract
Histopathology images are very distinctive, one image may contain thousands of objects. Transferring features from natural images to histopathology images may not provide impressive outcomes. In this study, we have proposed a novel modality specific CBAM-VGGNet model for classifying H and E stained breast histopathology images. Instead of using pre-trained models on ImageNet, we have trained VGG16 and VGG19 models on the same domain cancerous histopathology datasets, which are then used as fixed feature extractors. We have added the GAP layer and Convolutional block attention module (CBAM) after the first convolutional layer of convolutional blocks. CBAM is an effective module for neural networks to focus on relevant features. We have implemented the VGG16 and VGG19 in a novel way following the configuration of state-of-the-art models with our own concatenated layers. The addition of the GAP layer in VGGNet has reduced the number of parameters, requiring less computational power. Both models are ensembled using the averaging ensemble technique. Features are extracted from the final ensembled model and then passed to the feed-forward neural network. A hybrid pre-processing technique is proposed that first uses a median filter and then contrasts limited adaptive histogram equalization (CLAHE). The median filter removes the highly significant noise and is directly related to image quality. CLAHE improves the local contrast present in an image and boosts the weak boundary edges in each image pixel. The proposed CBAM ensemble model has outperformed state-of-the-art models with an accuracy of 98.96% and 97.95% F1-score on 400X data of the BreakHis dataset.