Deconv-transformer (DecT): A histopathological image classification model for breast cancer based on color deconvolution and transformer architecture

被引:108
作者
He, Zhu [1 ]
Lin, Mingwei [1 ]
Xu, Zeshui [3 ]
Yao, Zhiqiang [1 ]
Chen, Hong [2 ]
Alhudhaif, Adi [4 ]
Alenezi, Fayadh [5 ]
机构
[1] Fujian Normal Univ, Coll Comp & Cyber Secur, Fuzhou 350117, Fujian, Peoples R China
[2] Fujian Normal Univ, Sch Math & Stat, Fuzhou 350117, Fujian, Peoples R China
[3] Sichuan Univ, Business Sch, Chengdu 610064, Sichuan, Peoples R China
[4] Prince Sattam Bin Abdulaziz Univ, Coll Comp Engn & Sci Al Kharj, Dept Comp Sci, POB 151, Al Kharj 11942, Saudi Arabia
[5] Jouf Univ, Coll Engn, Dept Elect Engn, Sakaka, Saudi Arabia
关键词
Histopathological image; Breast cancer; Deep learning; Color deconvolution; Color space; RECOGNITION;
D O I
10.1016/j.ins.2022.06.091
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Histopathological image recognition of breast cancer is an onerous task. Although many deep learning models have achieved good classification results on histopathological image classification tasks, these models do not take full advantage of the staining properties of histopathological images. In this paper, we propose a novel Deconv-Transformer (DecT) network model, which incorporates the color deconvolution in the form of convolution layers. This model uses a self-attention mechanism to match the independent properties of the HED channel information obtained by the color deconvolution. It also uses a method similar to the residual connection to fuse the information of both RGB and HED color space images, which can compensate for the information loss in the process of transferring RGB images to HED images. The training process of the DecT model is divided into two stages so that the parameters of the deconvolution layer can be better adapted to different types of histopathological images. We use the color jitter in the image data augmentation process to reduce the overfitting in the model training process. The DecT model achieves an average accuracy of 93.02% and F1-score of 0.9389 on BreakHis dataset, and an average accuracy of 79.06% and 81.36% on BACH and UC datasets. (C) 2022 Elsevier Inc. All rights reserved.
引用
收藏
页码:1093 / 1112
页数:20
相关论文
共 47 条
[1]   BARF: A new direct and cross-based binary residual feature fusion with uncertainty-aware module for medical image classification [J].
Abdar, Moloud ;
Fahami, Mohammad Amin ;
Chakrabarti, Satarupa ;
Khosravi, Abbas ;
Plawiak, Pawel ;
Acharya, U. Rajendra ;
Tadeusiewicz, Ryszard ;
Nahavandi, Saeid .
INFORMATION SCIENCES, 2021, 577 (577) :353-378
[2]  
Albashish Dheeb, 2021, 2021 International Conference on Information Technology (ICIT), P805, DOI 10.1109/ICIT52682.2021.9491631
[3]  
American Cancer Society, 2019, Am Cancer Soc 1-120
[4]  
[Anonymous], 2016, VERY DEEP CONVOLUTIO
[5]   BACH: Grand challenge on breast cancer histology images [J].
Aresta, Guilherme ;
Araujo, Teresa ;
Kwok, Scotty ;
Chennamsetty, Sai Saketh ;
Safwan, Mohammed ;
Alex, Varghese ;
Marami, Bahram ;
Prastawa, Marcel ;
Chan, Monica ;
Donovan, Michael ;
Fernandez, Gerardo ;
Zeineh, Jack ;
Kohl, Matthias ;
Walz, Christoph ;
Ludwig, Florian ;
Braunewell, Stefan ;
Baust, Maximilian ;
Quoc Dang Vu ;
Minh Nguyen Nhat To ;
Kim, Eal ;
Kwak, Jin Tae ;
Galal, Sameh ;
Sanchez-Freire, Veronica ;
Brancati, Nadia ;
Frucci, Maria ;
Riccio, Daniel ;
Wang, Yaqi ;
Sun, Lingling ;
Ma, Kaiqiang ;
Fang, Jiannan ;
Kone, Ismael ;
Boulmane, Lahsen ;
Campilho, Aurelio ;
Eloy, Catarina ;
Polonia, Antonio ;
Aguiar, Paulo .
MEDICAL IMAGE ANALYSIS, 2019, 56 :122-139
[6]   Experimental Assessment of Color Deconvolution and Color Normalization for Automated Classification of Histology Images Stained with Hematoxylin and Eosin [J].
Bianconi, Francesco ;
Kather, Jakob N. ;
Reyes-Aldasoro, Constantino Carlos .
CANCERS, 2020, 12 (11) :1-20
[7]   Computer-aided diagnosis in the era of deep learning [J].
Chan, Heang-Ping ;
Hadjiiski, Lubomir M. ;
Samala, Ravi K. .
MEDICAL PHYSICS, 2020, 47 (05) :E218-E227
[8]   Transformer Interpretability Beyond Attention Visualization [J].
Chefer, Hila ;
Gur, Shir ;
Wolf, Lior .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :782-791
[9]  
Chen HY, 2022, Arxiv, DOI arXiv:2104.14528
[10]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807