Deep-Learning Model of ResNet Combined with CBAM for Malignant-Benign Pulmonary Nodules Classification on Computed Tomography Images.
Yanfei ZhangWei FengZhiyuan WuWeiming LiLixin TaoXiangtong LiuFeng ZhangYan GaoJian HuangXiu-Hua GuoPublished in: Medicina (Kaunas, Lithuania) (2023)
Background and Objectives : Lung cancer remains a leading cause of cancer mortality worldwide. Accurately classifying benign pulmonary nodules and malignant ones is crucial for early diagnosis and improved patient outcomes. The purpose of this study is to explore the deep-learning model of ResNet combined with a convolutional block attention module (CBAM) for the differentiation between benign and malignant lung cancer, based on computed tomography (CT) images, morphological features, and clinical information. Methods and materials : In this study, 8241 CT slices containing pulmonary nodules were retrospectively included. A random sample comprising 20% ( n = 1647) of the images was used as the test set, and the remaining data were used as the training set. ResNet combined CBAM (ResNet-CBAM) was used to establish classifiers on the basis of images, morphological features, and clinical information. Nonsubsampled dual-tree complex contourlet transform (NSDTCT) combined with SVM classifier (NSDTCT-SVM) was used as a comparative model. Results : The AUC and the accuracy of the CBAM-ResNet model were 0.940 and 0.867, respectively, in test set when there were only images as inputs. By combining the morphological features and clinical information, CBAM-ResNet shows better performance (AUC: 0.957, accuracy: 0.898). In comparison, a radiomic analysis using NSDTCT-SVM achieved AUC and accuracy values of 0.807 and 0.779, respectively. Conclusions : Our findings demonstrate that deep-learning models, combined with additional information, can enhance the classification performance of pulmonary nodules. This model can assist clinicians in accurately diagnosing pulmonary nodules in clinical practice.
Keyphrases
- deep learning
- computed tomography
- convolutional neural network
- artificial intelligence
- pulmonary hypertension
- machine learning
- positron emission tomography
- clinical practice
- image quality
- magnetic resonance imaging
- health information
- dual energy
- healthcare
- optical coherence tomography
- type diabetes
- cardiovascular disease
- palliative care
- squamous cell carcinoma
- magnetic resonance
- coronary artery disease
- risk factors
- social media
- squamous cell
- lymph node metastasis
- clinical evaluation