多尺度融合注意力机制的胆囊癌显微高光谱图像分类
A micro-hyperspectral image classification method of gallbladder cancer based on multi-scale fusion attention mechanism
- 2023年28卷第4期 页码:1173-1185
纸质出版日期: 2023-04-16
DOI: 10.11834/jig.211201
移动端阅览
浏览全部资源
扫码关注微信
纸质出版日期: 2023-04-16 ,
移动端阅览
高红民, 朱敏, 曹雪莹, 李臣明, 刘芹, 许佩佩. 2023. 多尺度融合注意力机制的胆囊癌显微高光谱图像分类. 中国图象图形学报, 28(04):1173-1185
Gao Hongmin, Zhu Min, Cao Xueying, Li Chenming, Liu Qin, Xu Peipei. 2023. A micro-hyperspectral image classification method of gallbladder cancer based on multi-scale fusion attention mechanism. Journal of Image and Graphics, 28(04):1173-1185
目的
2
胆囊癌作为胆道系统中一种恶性程度极高的肿瘤,早期诊断困难、预后极差,因此准确鉴别胆囊病变对早期发现胆囊癌具有重要意义。目前胆囊癌的诊断主要依赖于超声、CT(computed tomography)等传统影像学方法,但准确性较低。显微高光谱能够在获取生物组织图像信息的同时从生化角度对生物组织进行分析,从而实现对胆囊癌的早期诊断,相比于传统医学图像更具优势。因此,本文基于胆囊癌显微高光谱图像设计了一种基于多尺度融合注意力机制的网络模型,以提高分类准确率。
方法
2
提出多尺度融合注意力模块(multiscale squeeze-and-excitation-residual, MSE-Res)。MSE-Res模块引入改进的多尺度特征提取模块实现通道维上特征的融合,用一个最大池化层和一个上采样层代替1 × 1的卷积层来提取图像的显著特征。为了弥补池化层丢失的局部信息,在跳跃连接中加入一个1 × 1的卷积层。在多尺度特征提取模块后,引入注意力机制来学习不同通道间特征的相关性,实现通道间特征的融合,并通过残差连接使网络在提取图像深层特征的同时避免出现过拟合现象。
结果
2
在胆囊癌高光谱数据集上进行实验,本文模型的总体分类精度、平均分类精度和Kappa系数分别为99.599%、99.546%和0.990,性能优于SE-ResNet(squeeze-and-excitation-residual network)和Inception-SE-ResNet(inception-squeeze-and-excitation-residual network)。
结论
2
本文提出的MSE-ResNet能够有效利用高光谱图像的空间信息和光谱信息,提高胆囊癌分类准确率,在对胆囊癌的医学诊断方面具有一定的研究价值和现实意义。
Objective
2
Gallbladder carcinoma is recognized as one of the most malignant tumors in relevant to biliary system. Its prognosis is extremely poor, and only 6 months of overall average. It is challenged for missed diagnose because of the lack of typical clinical manifestations in early stage of gallbladder cancer. To clarify gallbladder lesions for early detection of gallbladder carcinoma accurately, current gallbladder cancer-related diagnosis is mainly focused on the interpretation of digital pathological section images (such as b-ultrasound, computed tomography (CT), magnetic resonance imaging (MRI), etc.) in terms of the computer-aided diagnosis (CAD). However, the accuracy is quite lower because the molecular level information of diseased organs cannot be obtained. Micro-hyperspectral technology can be incorporated the features of spectral analysis and optical imaging, and it can obtain the chemical composition and physical features for biological tissue samples at the same time. The changes of physical attributes of cancerous tissue may not be clear in the early stage, but the changes of chemical factors like its composition, structure and content can be reflected by spectral information. Therefore, micro hyperspectral imaging has its potentials to achieve the early diagnosis of cancer more accurately. Micro-hyperspectral technology, as a special optical diagnosis technology, can provide an effective auxiliary diagnosis method for clinical research. However, it can provide richer spectral information but large amount of data and information redundancy are increased. To develop an improved accuracy detection method and use the rich spatial and hyperspectral information effectively, we design a multi-scale fusion attention mechanism-relevant network model for gallbladder cancer-oriented classification accuracy optimization.
Method
2
The multiscale squeeze-and-excitation-residual (MSE-Res) can be used to realize the fusion of multiscale features between channel dimensions. First, an improved multi-scale feature extraction module is employed to extract features of different scales in channel dimension. To extract the salient features of the image a maximum pooled layer, an upper sampling layer is used beyond convolution layer of 1 × 1. To compensate for the missing local information in the pooled layer, a 1 × 1 convolution layer is added to the jump link. Next, the attention mechanism is introduced to learn the correlation of features between different channels, and the fusion of features is realized between channels. Finally, the residual link is used to alleviate over fitting problem while the deep features of the image are extracted. Our gallbladder cancer-based micro-hyperspectral images dataset is derived from the multidimensional common bile duct database produced by Professor Li Qingli’s team of East China Normal University. The database is composed of 880 multi-dimensional image scenes captured from common bile duct tissues of 174 patients. Each micro-hyperspectral image size is 1 280 × 1 024 × 60. The spectral resolution is 2~5 nm and the spectral range is 550~1 000 nm. All images are labeled by expertise. These micro-hyperspectral images of gallbladder carcinoma consists of three different samples: 1) background, 2) distorted region, and 3) normal region. The background part is organized of cells or blank areas-secreted fat mucus, which can be removed during model training and excluded in the training process. To facilitate the follow-up experiments, the image size is cut to 640 × 512 × 60, and four different hyperspectral image datasets are involved in. At the beginning, the spectral validation and principal component analysis (PCA) are used to preprocess the micro-hyperspectral images in order to reduce the interference of the stability of the light source and the noise in the micro-hyperspectral imaging system to the spectral curves of different tissues in the micro-hyperspectral imaging system. Then, the MSE-ResNet is used to classify the microscopic hyperspectral images of gallbladder carcinoma-relevant pathological sections. Our configuration is equipped with python3.5.6 and Keras2.1.6 on NVIDIA GeForce RTX 2080 Ti GPU, Intel (R) Xeon (R) CPU E5-2678 v3 CPU. The learning rate is 0.001, batch size is 16, and dropout rate is 0.3, as well as the optimization strategy is based on stochastic gradient descent (SGD). To alleviate over fitting problem of the network and improve the generalization ability of the model, three kind of regularization methods are used in MSE-ResNet, which are 1) batch normalization, 2) L2 regularization, and 3) dropout regularization.
Result
2
The comparison and ablation-related experiments on micro-hyperspectral datasets of gallbladder carcinoma are carried out. Initially, we use several evaluation metrics to evaluate the performance of the MSE-ResNet. The overall classification accuracy, average classification accuracy and kappa coefficient of this model are reached to 99.619%, 99.581% and 0.990 of each, which is better than SE-ResNet and Inception-SE-ResNet. Second, we compare the MSE-ResNet model to other related deep learning and machine learning methods, such as 1D-CNN, ResNet, DenseNet, support vector machine (SVM), and K-nearest neighbor (KNN). The results show that our MSE-Res module can extract the spatial and channel features of micro-hyperspectral images effectively, and classification results can be achieved with less computational cost and better robustness. Our MSE-ResNet model can be used to learn the features of hyperspectral images automatically and optimize the network parameters through back propagation in comparison with the traditional machine learning methods, which is more beneficial for the classification of micro-hyperspectral images. At last, we compare the micro-hyperspectral image to the traditional RGB image. The experimental results show that the richer band information of the micro-hyperspectral image can improve the model classification results effectively.
Conclusion
2
To improve the accuracy of classification of gallbladder cancer, our MSE-ResNet can be focused on the spatial and spectral information of hyperspectral images effectively. It has its potentials for gallbladder cancer-oriented medical diagnosis.
胆囊癌高光谱图像多尺度特征融合残差网络图像分类SE模块
hyperspectral image of gallbladder carcinomamulti-scale feature fusionresidual networkimages classificationsqueeze and excitation(SE) module
Akbari H, Halig L V, Schuster D M, Osunkoya A, Master V, Nieh P T, Chen G Z and Fei B W. 2012. Hyperspectral imaging and quantitative analysis for prostate cancer detection. Journal of Biomedical Optics, 17(7): #076005 [DOI: 10.1117/1.JBO.17.7.076005http://dx.doi.org/10.1117/1.JBO.17.7.076005]
Chang L. 2018. The Classification of Hyperspectral Microscopic Image about the Human Body Blood. Beijing: Beijing University of Chemical Technology (常岚. 2018. 人体血细胞高光谱显微图像分类. 北京: 北京化工大学)
Du J. 2018. Research of Tumor Tissue Classification Based on Medical Hyperspectral Imaging Analysis. Xi’an: Xi’an Institute of Optics and Precision Mechanics, Chinese Academy of Sciences
杜剑. 2018. 基于医学高光谱影像分析的肿瘤组织分类方法研究. 西安: 中国科学院西安光学精密机械研究所
Goetze T O. 2015. Gallbladder carcinoma: prognostic factors and therapeutic options. World Journal of Gastroenterology, 21(43): 12211-12217 [DOI: 10.3748/wjg.v21.i43.12211http://dx.doi.org/10.3748/wjg.v21.i43.12211]
He K M, Zhang X Y, Ren S Q and Sun J. 2016. Deep residual learning for image recognition//Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas, USA: IEEE: 770-778 [DOI: 10.1109/CVPR.2016.90http://dx.doi.org/10.1109/CVPR.2016.90]
Hu J, Shen L, Albanie S, Sun G and Wu E H. 2020. Squeeze-and-excitation networks. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42(8): 2011-2023 [DOI: 10.1109/TPAMI.2-019.2913372http://dx.doi.org/10.1109/TPAMI.2-019.2913372]
Huang G, Liu Z, van der Maaten L and Weinberger K Q. 2017. Densely connected convolutional networks//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition. Honolulu, USA: IEEE: 2261-2269 [DOI: 10.1109/CVPR.2017.243http://dx.doi.org/10.1109/CVPR.2017.243]
Huang Q. 2019. Classification of Hyperspectral Medical Microscopic Imagery Based on Convolutional Neural Network. Beijing: Beijing University of Chemical Technology
黄前. 2019. 基于卷积神经网络的高光谱医学显微图像分类研究. 北京: 北京化工大学
Huang Y. 2018. The Research on the Identification of Skin Melanoma Based on Microscopic Hyperspectral Imaging. Shanghai: East China Normal University
黄怡. 2018. 基于显微高光谱成像的皮肤黑色素瘤识别方法研究. 上海: 华东师范大学
Hundal R and Shaffer E A. 2014. Gallbladder cancer: epidemiology and outcome. Clinical Epidemiology, 6: 99-109 [DOI: 10.2147/CLEP.S37357http://dx.doi.org/10.2147/CLEP.S37357]
Ioffe S and Szegedy C. 2015. Batch normalization: accelerating deep network training by reducing internal covariate shift [EB/OL]. [2015-03-02]. https://arxiv.org/pdf/1502.03167.pdfhttps://arxiv.org/pdf/1502.03167.pdf
Jia G J. 2016. Study on the Application of Microscopic Hyperspectral Imaging in Blood Cell Recognition. Shanghai: East China Normal University
贾高杰. 2016. 基于显微高光谱成像的血液细胞识别研究与应用. 上海: 华东师范大学
Jiang H M, Mei J, Hu X X and Yi W S. 2013. Discussions of hyperspectral imaging for medical diagnostics. Chinese Journal of Medical Physics, 30(3): 4148-4152, 4158
江厚敏, 梅洁, 胡响祥, 易伟松. 2013. 高光谱成像医学诊断的探讨. 中国医学物理学杂志, 30(3): 4148-4152, 4158
Kukačka J, Golkov V and Cremers D. 2017. Regularization for deep learning: a taxonomy. [EB/OL]. [2017-10-29]. https://arxiv.org/pdf/1710.10686.pdfhttps://arxiv.org/pdf/1710.10686.pdf
Lu G L, Halig L, Wang D S, Chen Z G and Fei B W. 2014. Spectral-spatial classification using tensor modeling for cancer detection with hyperspectral imaging//Proceedings Volume 9034, Medical Imaging 2014: Image Processing. San Diego, USA: SPIE [DOI: 10.1117/12.2043796http://dx.doi.org/10.1117/12.2043796]
Pike R, Patton S K, Lu G L, Halig L V, Wang D S, Chen Z G and Fei B W. 2014. A minimum spanning forest based hyperspectral image classification method for cancerous tissue detection//Proceedings Volume 9034, Medical Imaging 2014: Image Processing. San Diego, USA: SPIE [DOI: 10.1117/12.2043848http://dx.doi.org/10.1117/12.2043848]
Song J. 2020. Study on lung cancer histopathological analysis based on microscopic hyperspectral imaging and deep neural network. Shanghai: East China Normal University
宋璟. 2020. 基于显微高光谱成像和深度神经网络的肺癌组织病理分析方法研究. 上海: 华东师范大学
Srivastava N, Hinton G, Krizhevsky A, Sutskever I and Salakhutdinov R. 2014. Dropout: a simple way to prevent neural networks from overfitting. The Journal of Machine Learning Research, 15(1): 1929-1958
Szegedy C, Ioffe S, Vanhoucke V and Alemi A. 2016. Inception-v4, inception-ResNet and the impact of residual connections on learning [EB/OL]. [2016-08-23]. https://arxiv.org/pdf/1602.07261.pdfhttps://arxiv.org/pdf/1602.07261.pdf
Wang Q, Wang J B, Zhou M, Li Q L and Wang Y T. 2017. Spectral-spatial feature-based neural network method for acute lymphoblastic leukemia cell identification via microscopic hyperspectral imaging technology. Biomedical Optics Express, 8(6): 3017-3028 [DOI: 10.1364/BOE.8.003017http://dx.doi.org/10.1364/BOE.8.003017]
Wang R D, He Y D, Yao C P, Wang S J, Xue Y, Zhang Z X, Wang J and Liu X L. 2020. Classification and segmentation of hyperspectral data of hepatocellular carcinoma samples using 1-D convolutional neural network. Cytometry Part A, 97(1): 31-38 [DOI: 10.1002/cyto.a.23871http://dx.doi.org/10.1002/cyto.a.23871]
Zhang Q, Li Q L, Yu G Z, Sun L, Zhou M and Chu J H. 2019. A multidimensional choledoch database and benchmarks for cholangiocarcinoma diagnosis. IEEE Access, 7: 149414-149421 [DOI: 10.1109/ACCESS.2019.2947470http://dx.doi.org/10.1109/ACCESS.2019.2947470]
Zheng S J, Qiu S, Li Q L, Zhou M, Hu M H and Yu G Z. 2021. Fourier transform channel attention network for cholangiocarcinoma hyperspectral image segmentation. Journal of Image and Graphics, 26(8): 1836-1846
郑少佳, 邱崧, 李庆利, 周梅, 胡孟晗, 于观贞. 2021. 傅里叶变换通道注意力网络的胆管癌高光谱图像分割. 中国图象图形学报, 26(8): 1836-1846[DOI: 10.11834/jig.210207http://dx.doi.org/10.11834/jig.210207]
相关作者
相关机构