A deep CNN based transfer learning method for false positive reduction

作者：Shi, Zhenghao^*; Hao, Huan; Zhao, Minghua; Feng, Yaning; He, Lifeng^*; Wang, Yinghui; Suzuki, Kenji^*

来源：Multimedia Tools and Applications, 2019, 78(1): 1017-1033.

DOI：10.1007/s11042-018-6082-6

• False positive reduction
• Nodule detection
• Deep convolutional network
• Support vector machine

A low false positive (FP) rate is of great importance for the use of a Computer Aided Detection (CAD) system to detect pulmonary nodules in thoracic Computed Tomography (CT). However, due to the variations of nodules in appear and size, it is still a very challenging task to obtain a low FP rate. In this paper, we propose a deep Convolutional Neural Network (CNN) based transfer learning method for FP reduction in pulmonary nodule detection on CT slices. We utilized one of the state-of-the-art CNN models, VGG-16 [4], as a feature extractor to obtain nodule features, and used a support vector machine (SVM) for nodule classification. Firstly we transferred all the layers from a pre-trained VGG-16 model in ImageNet to our target networks. Then, we tuned the last fully connected layers to adjust the computer-vision-task-trained CNN model to pulmonary nodule classification task. The initial CNN filter weights were then optimized using the training data, i.e., the pulmonary nodule patch images and corresponding labels through back-propagation so that they better reflected the modalities in the pulmonary nodule image dataset. Finally, features learned in the fine-tuned CNN were used to train a SVM classifier. The output of the trained SVM was used for final classification. Experimental results show that the overall sensitivity of the proposed method was 87.2% with 0.39 FPs per scan, which is higher than 85.4% with 4 FPs per scan obtained by other state of art method.

• 出版日期2019-1
• 单位西安理工大学