Ensemble framework based on attributes and deep features for benign-malignant classification of lung nodule

• Published in: Biomedical signal processing and control Vol. 79; p. 104217
• Main Authors: Qiao, Jianping, Fan, Yanling, Zhang, Mowen, Fang, Kunlun, Li, Dengwang, Wang, Zhishun
• Format: Journal Article
• Language: English
• Published: Elsevier Ltd 01.01.2023
• Subjects: Benign-malignant classification; Data fusion; Ensemble learning; Lung nodule; Data fusion; Benign-malignant classification; Ensemble learning; Lung nodule
• ISSN: 1746-8094; 1746-8108
• DOI: 10.1016/j.bspc.2022.104217

•An ensemble learning model (F-LSTM-CNN) for lung nodule classification is proposed.•The F-CNN module applies attribute features to lung nodule classification.•The F-LSTM module focuses on the important features by integrating and mapping attributes into the deep features.•Statistical analysis and comparative experiments demonstrate the superiority of the proposed model. Early detection and identification of malignant lung nodules improve the survival of lung cancer patients. The visual attributes such as subtlety, spiculation, and calcification of lung nodules play an important role in the diagnosis of malignancy. However, the gap between attributes and computation features is the main factor that restricts the performance of computer-aided diagnosis (CAD). Therefore, we propose a Fuse-Long Short-Term Memory-Convolutional Neural Network (F-LSTM-CNN) ensemble learning algorithm which incorporates visual attributes and deep features to classify benign and malignant nodules. First, the attribute features are obtained from clinical information while the deep features of nodules are extracted from the preprocessed computed tomography (CT) images. Second, the Fuse-Convolutional Neural Network (F-CNN) model is proposed for highlighting the essential role of attributes in the classification processing which integrates deep features and attribute features mapped through the transposed convolution. Meanwhile, the Fuse-Long Short-Term Memory (F-LSTM) model is proposed to focus on the specific deep features for classification via the affine transformation of attribute features. Finally, early identification of malignant lung nodules is conducted by fusing the prediction scores of the F-LSTM and F-CNN models. The experiments were conducted on the public lung nodule dataset (LIDC-IDRI) and achieved accuracy, sensitivity, and specificity of 0.955, 1, and 0.937 with an Area under the ROC Curve (AUC) of 0.995 for lung nodule classification. The experiment results show that the proposed F-LSTM-CNN ensemble learning model facilitates the interpretation of diagnostic data and helps radiologists to make decisions in clinical practice.