AUTOMATED CLASSIFICATION OF SOLITARY PULMONARY NODULES USING CONVOLUTIONAL NEURAL NETWORK BASED ON TRANSFER LEARNING STRATEGY

Abstract

This study aimed to propose an effective malignant solitary pulmonary nodule classification method based on improved Faster R-CNN and transfer learning strategy. In practice, the existing solitary pulmonary nodule classification methods divide the lung cancer images into two categories only: normal and cancerous. This study proposed the deep convolution neural network to classify the computed tomography (CT) images of lung cancer into four categories: lung adenocarcinoma, lung squamous cell carcinoma, metastatic lung cancer, and normal types of lung cancer. Some high-resolution lung CT images have unnecessary characters such as a large number of high-density continuity features, small-size lung nodule targets, CT image background complexity, and so forth. In this study, the CT image sub-block preprocessing strategy was used to extract nodule features for enhancement and alleviate the aforementioned problems. The experimental results showed that the proposed system was effective in resolving issues such as high false-positive rate and long classification time cost based on the original Faster R-CNN detection method. Meanwhile, the transfer learning strategy was used to improve the classification efficiency so as to avoid the overfitting problem caused by a few labeled samples of lung cancer datasets. The classification results were integrated using the majority vote algorithm. The classification results of the lung CT imaging showed that the proposed method had an average detection accuracy of 89.7% and reduced the rate of misdiagnosis to meet the clinical needs.