학술논문

Modality Specific CBAM-VGGNet Model for the Classification of Breast Histopathology Images via Transfer Learning
Document Type
Periodical
Source
IEEE Access Access, IEEE. 11:15750-15762 2023
Subject
Aerospace
Bioengineering
Communication, Networking and Broadcast Technologies
Components, Circuits, Devices and Systems
Computing and Processing
Engineered Materials, Dielectrics and Plasmas
Engineering Profession
Fields, Waves and Electromagnetics
General Topics for Engineers
Geoscience
Nuclear Engineering
Photonics and Electrooptics
Power, Energy and Industry Applications
Robotics and Control Systems
Signal Processing and Analysis
Transportation
Histopathology
Feature extraction
Transfer learning
Breast cancer
Convolutional neural networks
Computational modeling
Deep learning
CBAM
CLAHE
classification
deep learning
histopathology
transfer learning
Language
ISSN
2169-3536
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.