Affiliations: [a] College of Computer Science and Engineering, Shandong University of Science and Technology, Qingdao, China
| [b] Shandong Province Key Laboratory of Wisdom Mining Information Technology, Shandong University of Science and Technology, Qingdao, China
Correspondence:
[*]
Corresponding author. Tsu-Yang Wu, College of Computer Science and Engineering, Shandong University of Science and Technology, Qingdao, China. E-mail: [email protected].
Abstract: Transfer learning technique is popularly employed for a lot of medical image classification tasks. Here based on convolutional neural network (CNN) and sparse coding process, we present a new deep transfer learning architecture for false positive reduction in lymph node detection task. We first convert the linear combination of the deep transferred features to the pre-trained filter banks. Next, a new point-wise filter based CNN branch is introduced to automatically integrate transfer features for the false and positive image classification purpose. To lower the scale of the proposed architecture, we bring sparse coding process to the fixed transferred convolution filter banks. On this basis, a two-stage training strategy with grouped sparse connection is presented to train the model efficiently. The model validity is tested on lymph node dataset for false positive reduction and our approach indicates encouraging performances compared to prior approaches. Our method reaches sensitivities of 71% /85% at 3 FP/vol. and 82% /91% at 6 FP/vol. in abdomen and mediastinum respectively, which compare competitively to previous approaches.
Keywords: Deep learning, lymph node false
positive reduction, transfer learning method, sparse coding
algorithm
