Multi-scale pulmonary nodule classification with deep feature fusion via residual network

The early stage detection of benign and malignant pulmonary nodules plays an important role in clinical diagnosis. The malignancy risk assessment is usually used to guide the doctor in identifying the cancer stage and making follow-up prognosis plan. However, due to the variance of nodules on size, shape, and location, it has been a big challenge to classify the nodules in computer aided diagnosis system. In this paper, we design a novel model based on convolution neural network to achieve automatic pulmonary nodule malignancy classification. By using our model, the multi-scale features are extracted through the multi-convolution process, and the structure of residual blocks allows the network to capture more high-level and semantic information. Moreover, a strategy is proposed to fuse the features from the last avg-pooling layer and the ones from the last residual block to further enhance the performance of our model. Experimental results on the public Lung Image Database Consortium dataset demonstrate that our model can achieve a lung nodule classification accuracy of $$87.5\%$$87.5% which outperforms state-of-the-art methods.

[1]  K. Doi,et al.  Quantitative computerized analysis of diffuse lung disease in high-resolution computed tomography. , 2003, Medical physics.

[2]  Lubomir M. Hadjiiski,et al.  Computer-aided diagnosis of pulmonary nodules on CT scans: segmentation and classification using 3D active contours. , 2006, Medical physics.

[3]  M. Gould,et al.  A clinical model to estimate the pretest probability of lung cancer in patients with solitary pulmonary nodules. , 2007, Chest.

[4]  Temesguen Messay,et al.  A new computationally efficient CAD system for pulmonary nodule detection in CT imagery , 2010, Medical Image Anal..

[5]  Jean Ponce,et al.  A Theoretical Analysis of Feature Pooling in Visual Recognition , 2010, ICML.

[6]  Richard C. Pais,et al.  The Lung Image Database Consortium (LIDC) and Image Database Resource Initiative (IDRI): a completed reference database of lung nodules on CT scans. , 2011, Medical physics.

[7]  A. Jemal,et al.  Cancer statistics, 2011 , 2011, CA: a cancer journal for clinicians.

[8]  Honglak Lee,et al.  Learning hierarchical representations for face verification with convolutional deep belief networks , 2012, 2012 IEEE Conference on Computer Vision and Pattern Recognition.

[9]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[10]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[11]  Hong Zhao,et al.  Texture Feature Analysis for Computer-Aided Diagnosis on Pulmonary Nodules , 2015, Journal of Digital Imaging.

[12]  Niranjan Khandelwal,et al.  A Combination of Shape and Texture Features for Classification of Pulmonary Nodules in Lung CT Images , 2016, Journal of Digital Imaging.

[13]  Bram van Ginneken,et al.  Automatic classification of pulmonary peri-fissural nodules in computed tomography using an ensemble of 2D views and a convolutional neural network out-of-the-box , 2015, Medical Image Anal..

[14]  Bai Ying Lei,et al.  Accurate Segmentation of Cervical Cytoplasm and Nuclei Based on Multiscale Convolutional Network and Graph Partitioning , 2015, IEEE Transactions on Biomedical Engineering.

[15]  Alexander Wong,et al.  Lung Nodule Classification Using Deep Features in CT Images , 2015, 2015 12th Conference on Computer and Robot Vision.

[16]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[17]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[18]  Bram van Ginneken,et al.  Pulmonary Nodule Detection in CT Images: False Positive Reduction Using Multi-View Convolutional Networks , 2016, IEEE Transactions on Medical Imaging.

[19]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[20]  Hengyong Yu,et al.  Deep Learning for the Classification of Lung Nodules , 2016, ArXiv.

[21]  Luc Van Gool,et al.  Deep Retinal Image Understanding , 2016, MICCAI.

[22]  Hao Chen,et al.  Multilevel Contextual 3-D CNNs for False Positive Reduction in Pulmonary Nodule Detection , 2017, IEEE Transactions on Biomedical Engineering.

[23]  Jie-Zhi Cheng,et al.  Automatic Scoring of Multiple Semantic Attributes With Multi-Task Feature Leverage: A Study on Pulmonary Nodules in CT Images. , 2017, IEEE transactions on medical imaging.

[24]  Konstantinos Kamnitsas,et al.  Efficient multi‐scale 3D CNN with fully connected CRF for accurate brain lesion segmentation , 2016, Medical Image Anal..

[25]  Sergey Ioffe,et al.  Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning , 2016, AAAI.

[26]  Wei Shen,et al.  Multi-crop Convolutional Neural Networks for lung nodule malignancy suspiciousness classification , 2017, Pattern Recognit..

[27]  Bai Ying Lei,et al.  Automatic Scoring of Multiple Semantic Attributes With Multi-Task Feature Leverage: A Study on Pulmonary Nodules in CT Images , 2017, IEEE Transactions on Medical Imaging.

[28]  Dennis Wollersheim,et al.  Pulmonary nodule classification with deep residual networks , 2017, International Journal of Computer Assisted Radiology and Surgery.

[29]  Ulas Bagci,et al.  Risk Stratification of Lung Nodules Using 3D CNN-Based Multi-task Learning , 2017, IPMI.

[30]  Yuan Li,et al.  A Hybrid Model: DGnet-SVM for the Classification of Pulmonary Nodules , 2017, ICONIP.

[31]  Shutao Li,et al.  Hyperspectral Image Classification With Deep Feature Fusion Network , 2018, IEEE Transactions on Geoscience and Remote Sensing.

[32]  Shihui Ying,et al.  Multimodal Neuroimaging Feature Learning With Multimodal Stacked Deep Polynomial Networks for Diagnosis of Alzheimer's Disease , 2018, IEEE Journal of Biomedical and Health Informatics.

[33]  Kui Jia,et al.  Canonical Correlation Analysis Regularization: An Effective Deep Multiview Learning Baseline for RGB-D Object Recognition , 2019, IEEE Transactions on Cognitive and Developmental Systems.

[34]  Dan Liu,et al.  Deep learning based smart radar vision system for object recognition , 2019, J. Ambient Intell. Humaniz. Comput..

[35]  Yongming Huang,et al.  Feature fusion methods research based on deep belief networks for speech emotion recognition under noise condition , 2017, Journal of Ambient Intelligence and Humanized Computing.