融合注意力机制和空洞卷积的3D ELD_MobileNetV2在肝结节分类中的应用
收稿日期: 2024-08-12
网络出版日期: 2024-11-13
Application of 3D ELD_MobileNetV2 Incorporating Attention Mechanism and Dilated Convolution in Hepatic Nodules Classification
Received date: 2024-08-12
Online published: 2024-11-13
我国肝癌的发病率和死亡率都不容乐观,早诊断、早治疗成为改变这一现状的迫切手段.因此,本文提出一种融合注意力机制和空洞卷积的3D ELD_MobileNetV2肝结节分类模型,用于腹部动态增强磁共振图像四分类.首先,将二维网络结构扩展为三维,避免对磁共振图像进行特征提取时出现空间特征损失的现象;其次,基于局部跨通道交互策略,设计了融合局部特征和全局特征的高效通道注意力机制,将其嵌入到MobileNetV2的瓶颈结构中,增强网络对关键特征的提取能力;然后,在深度卷积中引入三维空洞结构,提高卷积核的感受野;同时,使用Leaky ReLU6激活函数替换原始激活函数,提升模型鲁棒性.该模型在120名患者中(肝炎、硬化结节、异型增生、肝细胞癌各30例)进行了测试和验证.实验结果表明,所提出的模型相较于原始MobileNetV2,准确率提高了0.083.与AlexNet、VggNet16、ResNet50、ConvNeXt等网络相比,3D ELD_MobileNetV2表现最佳,准确率为0.792,F1分数(F1_Score)为0.688、0.750、0.848、0.872,微平均曲线下面积(AUC)为0.954,宏平均AUC为0.948.该研究所提出的模型能够较好对不同时期的肝结节进行分类,有望为肝癌早期诊断提供帮助.
关键词: 3D MobileNetV2; 高效通道注意力; 空洞卷积; 肝结节分类; 动态增强磁共振成像
孙灏芸 , 王丽嘉 . 融合注意力机制和空洞卷积的3D ELD_MobileNetV2在肝结节分类中的应用[J]. 波谱学杂志, 2025 , 42(2) : 130 -142 . DOI: 10.11938/cjmr20243128
The morbidity and mortality rates of liver cancer in China remain concerning, emphasizing the urgent need for early diagnosis and treatment to improve the situation. In response to this challenge, we propose a novel 3D ELD_MobileNetV2 hepatic nodule classification model that incorporates attention mechanism and dilated convolution. This model is specifically designed to classify abdominal dynamic enhanced magnetic resonance images. Firstly, the two-dimensional network structure was extended to three dimensions to avoid the loss of spatial information during feature extraction of magnetic resonance imaging (MRI). Secondly, based on the local cross-channel interaction strategy, an efficient channel attention mechanism combining local features and global features was embedded in the bottleneck structure of MobileNetV2 network to enhance the key feature extraction capability. Then, 3D dilated structure was introduced into depthwise convolution to improve the receptive field of the convolution kernel. Meanwhile, the original activation function was replaced with Leaky ReLU6 activation function to improve the model's robustness. The model was tested and validated on a dataset comprising 120 patients (30 cases each of hepatitis, cirrhotic nodules, dysplastic nodules, and hepatocellular carcinoma). Experimental results demonstrate significant improvements over the original MobileNetV2, with an accuracy increase of 0.083. Compared to other networks, including AlexNet, VggNet16, ResNet50, ConvNeXt, the 3D ELD_MobileNetV2 achieves superior performance, with the accuracy of 0.792, F1_Score of 0.688, 0.750, 0.848, 0.872, micro-average AUC of 0.954, and macro-average AUC of 0.948. The findings highlight the effectiveness of the proposed model in classifying liver nodules across different stages. This advancement is expected to facilitate early diagnosis of liver cancer and improve clinical outcomes.
| [1] | HAN B, ZHENG R, ZENG H, et al. Cancer incidence and mortality in China, 2022[J]. J Natl Cancer Cent, 2024, 4(1): 47-53. |
| [2] | RUMGAY H, ARNOLD M, FERLAY J, et al. Global burden of primary liver cancer in 2020 and predictions to 2040[J]. J Hepatol, 2022, 77(6): 1598-1606. |
| [3] | BRAY F, LAVERSANNE M, SUNG H, et al. Global cancer statistics 2022: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries[J]. CA Cancer J Clin, 2024, 74(3): 229-263. |
| [4] | VOGEL A, MEYER T, SAPISOCHIN G, et al. Hepatocellular carcinoma[J]. Lancet, 2022, 400(10360): 1345-1362. |
| [5] | RENZULLI M, BRACCISCHI L, DERRICO A, et al. State-of-the-art review on the correlations between pathological and magnetic resonance features of cirrhotic nodules[J]. Histol Histopathol, 2022, 37(12): 1151-1165. |
| [6] | YOKOO T, MASAKI N, PARIKH N D, et al. Multicenter validation of abbreviated MRI for detecting early-stage hepatocellular carcinoma[J]. Radiology, 2023, 307(2): e220917. |
| [7] | BR?KER M E E, TAIMR P, DE VRIES M, et al. Performance of contrast-enhanced sonography versus MRI with a liver-specific contrast agent for diagnosis of hepatocellular adenoma and focal nodular hyperplasia[J]. AJR Am J Roentgenol, 2020, 214(1): 81-89. |
| [8] | RELA M, RAO S N, PATIL R R. Performance analysis of liver tumor classification using machine learning algorithms[J]. Int J Adv Tec Engin Explor, 2022, 9(86): 143-154. |
| [9] | LI D, HUO L, WAN M Y, et al. Application of radiomics based on new support vector machine in the classification of hepatic nodules[J]. Chinese J Magn Reson, 2022, 39(3): 278-290. |
| 李笛, 霍雷, 万梦云, 等. 基于新型支持向量LIDI机的影像组学在肝脏结节分类中的应用[J]. 波谱学杂志, 2022, 39(3): 278-290. | |
| [10] | XING Y H, HUANG Z J, HOU Y H. Application of enhanced CT-based radiomics in differentiating small hepatocellular carcinoma from hepatic dysplastic nodules[J]. Chinese Journal of CT and MRI, 2023, 21(7): 107-109. |
| 邢艳虎, 黄忠江, 侯跃宏. 基于增强CT影像组学鉴别小肝癌与肝不典型增生结节的应用研究[J]. 中国CT和MRI杂志, 2023, 21(7): 107-109. | |
| [11] | SAHA ROY S, ROY S, MUKHERJEE P, et al. An automated liver tumour segmentation and classification model by deep learning based approaches[J]. Comput Methods Biomech Biomed Engin, 2023, 11(3): 638-650. |
| [12] | CHENG N, REN Y, ZHOU J, et al. Deep learning-based classification of hepatocellular nodular lesions on whole-slide histopathologic images[J]. Gastroenterology, 2022, 162(7): 1948-1961.e7. |
| [13] | TRIVIZAKIS E, MANIKIS G C, NIKIFORAKI K, et al. Extending 2-D convolutional neural networks to 3-D for advancing deep learning cancer classification with application to MRI liver tumor differentiation[J]. IEEE J Biomed Health Inform, 2018, 23(3): 923-930. |
| [14] | ZHANG X, JIA N, WANG Y. Multi-input dense convolutional network for classification of hepatocellular carcinoma and intrahepatic cholangiocarcinoma[J]. Biomed Signal Proces, 2023, 80: 104226. |
| [15] | ZHOU J, WANG W, LEI B, et al. Automatic detection and classification of focal liver lesions based on deep convolutional neural networks: a preliminary study[J]. Front Oncol, 2021, 10: 581210. |
| [16] | KRIZHEVSKY A, SUTSKEVER I, HINTON G E. ImageNet classification with deep convolutional neural networks[J]. Communications of the ACM, 2017, 60(6): 84-90. |
| [17] | SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[J]. arXiv preprint arXiv:1409.1556, 2014. |
| [18] | ABDULSAHIB FI, AL-KHATEEB B, KóCZY LT, et al. A transfer learning approach for the classification of liver cancer[J]. J Intell Syst, 2023, 32(1): 1-14. |
| [19] | LIU Z, MAO H, WU C Y, et al. A convnet for the 2020s[C]// Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2022: 11976-11986. |
| [20] | SANDLER M, HOWARD A, ZHU M, et al. Mobilenetv2: Inverted residuals and linear bottlenecks[C]// Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 4510-4520. |
| [21] | HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition[C]// Proceedings of the IEEE conference on computer vision and pattern recognition. 2016: 770-778. |
| [22] | TRIPATHI A, SINGH T, NAIR R R, et al. Improving early detection and classification of lung diseases with innovative MobileNetV2 framework[J]. IEEE Access, 2024: 116202-116217. |
| [23] | WANG H, QI Q, SUN W, et al. Classification of skin lesions with generative adversarial networks and improved MobileNetV2[J]. Int J Imaging Syst Technol, 2023, 33(5): 1561-1576. |
| [24] | XU L, MOHAMMADI M. Brain tumor diagnosis from MRI based on Mobilenetv2 optimized by contracted fox optimization algorithm[J]. Heliyon, 2023, 10(1): e23866. |
| [25] | WANG Q, WU B, ZHU P, et al. ECA-Net: Efficient channel attention for deep convolutional neural networks[C]// Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2020: 11534-11542. |
| [26] | TU C, LIU W, JIANG W, et al. Hyperspectral image classification based on residual dense and dilated convolution[J]. Infrared Phys Techn, 2023, 131: 104706. |
| [27] | DAUBECHIES I, DEVORE R, FOUCART S, et al. Nonlinear approximation and (deep) ReLU networks[J]. Constr Approx, 2022, 55(1): 127-172. |
| [28] | ZHAO L, WANG N, ZHU X, et al. Establishment and validation of an artificial intelligence-based model for real-time detection and classification of colorectal adenoma[J]. Sci Rep, 2024, 14(1): 10750. |
| [29] | XU J, LI Z, DU B, et al. Reluplex made more practical: Leaky ReLU[C]// 2020 IEEE Symposium on Computers and communications (ISCC). IEEE, 2020: 1-7. |
| [30] | GARCEA F, SERRA A, LAMBERTI F, et al. Data augmentation for medical imaging: A systematic literature review[J]. Comput Biol Med, 2023, 152: 106391. |
| [31] | CARTER J V, PAN J, RAI S N, et al. ROC-ing along: Evaluation and interpretation of receiver operating characteristic curves[J]. Surgery, 2016, 159(6): 1638-1645. |
| [32] | TIAN H, WU J, BIAN Y, et al. Classification of pancreatic cystic tumors based on DenseNet and transfer learning[J]. Chinese J Magn Reson, 2023, 40(3): 270-279. |
| 田慧, 武杰, 边云, 等. 基于DenseNet结合迁移学习的胰腺囊性肿瘤分类方法[J]. 波谱学杂志, 2023, 40(3): 270-279. |
/
| 〈 |
|
〉 |