改进DenseNet的乳腺癌病理图像八分类研究

doi:10.3778/j.issn.1002-8331.2111-0081

摘要/Abstract

摘要： 目前，在医学图像领域存在乳腺癌组织病理图像自动分类难以应用于临床诊断的现象，究其根源是当前没有大型公开的数据集或数据集数据不均衡。针对上述问题，提出一种结合密集卷积神经网络（dense convolutional network，DenseNet）、注意力机制（attention mecheanism）和焦点损失函数（Focal loss）的乳腺癌组织病理图像的多分类模型，即DAFLNet。DAFLNet在乳腺癌组织病理图像数据集BreaKHis上进行训练、验证与测试，最终实验结果显示，该模型对良恶性二分类的识别准确率达到99.1%，对乳腺亚型八分类的识别准确率达到95.5%。证明在数据不均衡的条件下，DAFLNet模型能够准确地对乳腺组织病理图像进行八分类。

关键词: 乳腺癌病理图像, DenseNet, 八分类, 注意力机制, Focal loss

Abstract: At present, in the field of medical image, the automatic classification of breast cancer histopathology is difficult to be applied to clinical diagnosis. The root cause is that there is no large-scale open dataset or data set is unbalanced. In view of the above problems, this paper proposes a multi classification model of breast cancer histopathological images combined with dense convolutional network（DenseNet）, attention mechanism and Focal loss, that is DAFLNet. DAFLNet is trained, validated and tested on the breast cancer histopathological image dataset BreaKHis. The final experimental results show that the recognition accuracy of the model for benign and malignant secondary classification is 99.1%, and the recognition accuracy of eight classification of breast subtypes is 95.5%. It is proved that DAFLNet model can accurately classify breast histopathological images under the condition of unbalanced data.

Key words: breast cancer pathological image, dense convolutional network（DenseNet）, eight categories, attention mecha-nism, Focal loss

赵晓平, 王荣发, 孙中波, 魏旭全. 改进DenseNet的乳腺癌病理图像八分类研究[J]. 计算机工程与应用, 2023, 59(5): 213-221.

ZHAO Xiaoping, WANG Rongfa, SUN Zhongbo, WEI Xuquan. Research on Eight Classifications of Breast Cancer Pathological Images Based on Improved DenseNet[J]. Computer Engineering and Applications, 2023, 59(5): 213-221.

参考文献

[1] SUNG H，FERLAY J，SIEGEL R L，et al.Global cancer statistics 2020：GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries[J].CA：a Cancer Journal for Clinicians，2021，71（3）：209-249.
[2] 何雪英，韩忠义，魏本征.基于深度学习的乳腺癌病理图像自动分类[J].计算机工程与应用，2018，54（12）：121-125.
HE X Y，HAN Z Y，WEI B Z.Breast cancer histopathological image auto-classification using deep learning[J].Computer Engineering and Applications，2018，54（12）：121-125.
[3] ZHANG Y，ZHANG B，LU W.Breast cancer histological image classification with multiple features and random subspace classifier ensemble[J].Studies in Computational Intelligence，2013，450（2）：27-42.
[4] QU A P，SHEN J M，WANG L W，et al.Segmentation of Hematoxylin-Eosin stained breast cancer histopathological images based on pixel-wise SVM classifier[J].Science China Information Sciences，2015，58（9）：1-13.
[5] PAUL A，DEY A，MUKHERJEE D P，et al.Regenerative random forest with automatic feature selection to detect mitosis in histopathological breast cancer images[C]//International Conference on Medical Image Computer-Assisted Intervention.Charm：Springer，2015.
[6] ZHANG Y，ZHANG B，COENEN F，et al.One-class kernel subspace ensemble for medical image classification[J].EURASIP Journal on Advances in Signal Processing，2014（1）：1-13.
[7] SPANHOL F A，OLIVEIRA L S，PETITJEAN C，et al.A dataset for breast cancer histopathological image classi-fication[J].IEEE Transactions on Biomedical Engineering，2015，63（7）：1455-1462.
[8] BELSARE A D，MUSHRIF M M，PANGARKAR M A，et al.Classification of breast cancer histopathology images using texture feature analysis[C]//TENCON 2015-2015 IEEE Region 10 Conference，2016.
[9] WANG P，HU X，LI Y，et al.Automatic cell nuclei segmentation and classification of breast cancer histopathology images[J].Signal Processing，2016，122：1-13.
[10] LITJENS G，KOOI T，BEJNORDI B E，et al.A survey on deep learning in medical image analysis[J].Medical Image Analysis，2017，42（9）：60-88.
[11] SPANHOL F A，OLIVEIRA L S，PETITJEAN C，et al.Breast cancer histopathological image classification using convolutional neural networks[C]//International Joint Conference on Neural Networks（IJCNN 2016），2016.
[12] BAYRAMOGLU N，KANNALA J，HEIKKILA J.Deep learning for magnification independent breast cancer histopathology image classification[C]//International Conference on Pattern Recognition，2017.
[13] 郑群花，段慧芳，沈尧，等.基于卷积神经网络和迁移学习的乳腺癌病理图像分类[J].计算机应用与软件，2018，35（7）：237-242.
ZHENG Q H，DUAN H F，SHEN Y，et al.Breast cancer pathological image classification based on convolution neural network and transfer learning[J].Computer Applications and Software，2018，35（7）：237-242.
[14] 王恒，李霞，刘晓芳，等.基于ResNet50网络的乳腺癌病理图像分类研究[J].中国计量大学学报，2019，30（1）：72-77.
WANG H，LI X，LIU X F，et al.Classification of breast cancer pathological images based on ResNet50 network[J].Journal of China Jiliang University，2019，30（1）：72-77.
[15] HUANG G，LIU Z，LAURENS V，et al.Densely connected convolutional networks[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），2017：2261-2269.
[16] HE K，ZHANG X，REN S，et al.Deep residual learning for image recognition[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），2016：770-778.
[17] SZEGEDY C，LIU W，JIA Y，et al.Going deeper with convolutions[C]//2015 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），2015：1-9.
[18] FEI W，JIANG M，CHEN Q，et al.Residual attention network for image classification[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），2017：6450-6458.
[19] FU J，LIU J，TIAN H，et al.Dual attention network for scene segmentation[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），2019：3141-3149.
[20] LIN T Y，GOYAL P，GIRSHICK R，et al.Focal loss for dense object detection[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2020，42（2）：318-327.
[21] 程年，俞晨，宁静艳.基于深度学习网络的乳腺癌图片分类研究[J].软件导刊，2019，18（8）：26-28.
CHENG N，YU C，NING J Y.Classification of breast couldcer images based on deep learning network[J].Software Guide，2019，18（8）：26-28.
[22] NGUYEN P T，NGUYEN T T，NGUYEN N C，et al.Multiclass breast cancer classification using convolutional neural network[C]//2019 International Symposium on Electrical and Electronics Engineering（ISEE），2019.
[23] TO?A?AR M，?ZKURT K B，ERGEN B，et al.BreastNet：a novel convolutional neural network model through histopathological images for the diagnosis of breast cancer[J].Physica A：Statistical Mechanics and Its Applications，2020，545：123592.
[24] 刘靖雯，黄理灿.基于Inception-ResNet-V2的乳腺癌病理图像识别研究[J].软件导刊，2020，19（5）：225-229.
LIU J W，HUANG L C.Pathological image recognition of breast cancer based on Inception-ResNet-V2[J].Software Guide，2020，19（5）：225-229.
[25] HAN Z，WEI B，ZHENG Y，et al.Breast cancer multi-classification from histopathological images with structured deep learning model[J].Scientific Reports，2017，7（1）：4172.
[26] BARDOU D，ZHANG K，AHMAD S M.Classification of breast cancer based on histology images using convolutional neural networks[J].IEEE Access，2018，16：24680-24693.
[27] 李雨倩.基于深度学习的乳腺组织病理类型的多分类方法研究[D].合肥：中国科学技术大学，2019.
LI Y Q.Research on multi-classification of breast histopathological patterns based on deep learning[D].Hefei：University of Science and Technology of China，2019.
[28] 明涛，王丹，郭继昌，等.基于多尺度通道重校准的乳腺癌病理图像分类[J].浙江大学学报（工学版），2020，54（7）：1289-1297.
MING T，WANG D，GUO J C，et al.Breast cancer histopathological image classification using multi-scale channel squeeze-and-excitation model[J].Journal of Zhejiang University（Engineering Edition），2020，54（7）：1289-1297.
[29] JIANG Y，CHEN L，ZHANG H，et al.Breast cancer histopathological image classification using convolutional neural networks with small SE-ResNet module[J].Plos One，2019，14（3）：e0214587.
[30] 刘巧利，闫航，贺鹏飞，等.基于改进DenseNet网络的乳腺癌自动诊断[J].计算机与数字工程，2019，47（10）：2496-2502.
LIU Q L，YAN H，HE P F，et al.Automatic diagnosis of breast cancer based on improved DenseNet network[J].Computer and Digital Engineering，2019，47（10）：2496-2502.
[31] 于凌涛，夏永强，王鹏程，等.基于多任务模型的乳腺癌病理图像分类[J].华中科技大学学报（自然科学版），2021，49（8）：53-57.
YU L T，XIA Y Q，WANG P C，et al.Breast cancer pathological image classification based on multi-task model[J].Journal of Huazhong University of Science and Technology（Natural Science Edition），2021，49（8）：53-57.