TY - JOUR
T1 - Deep CNN models for pulmonary nodule classification
T2 - Model modification, model integration, and transfer learning
AU - Zhao, Xinzhuo
AU - Qi, Shouliang
AU - Zhang, Baihua
AU - Ma, He
AU - Qian, Wei
AU - Yao, Yudong
AU - Sun, Jianjun
N1 - Publisher Copyright:
© 2019 - IOS Press and the authors. All rights reserved.
PY - 2019
Y1 - 2019
N2 - BACKGROUND: Deep learning has made spectacular achievements in analysing natural images, but it faces challenges for medical applications partly due to inadequate images. OBJECTIVE: Aiming to classify malignant and benign pulmonary nodules using CT images, we explore different strategies to utilize the state-of-the-art deep convolutional neural networks (CNN). METHODS: Experiments are conducted using the Lung Image Database Consortium image collection (LIDC-IDRI), which is a public database containing 1018 cases. Three strategies are implemented including to 1) modify some state-of-the-art CNN architectures, 2) integrate different CNNs and 3) adopt transfer learning. Totally, 11 deep CNN models are compared using the same dataset. RESULTS: Study demonstrates that, for the model modification scheme, a concise CifarNet performs better than the other modified CNNs with more complex architectures, achieving an area under ROC curve of AUC = 0.90. Integrated CNN models do not significantly improve the classification performance, but the model complexity is reduced. Transfer learning outperforms the other two schemes and ResNet with fine-tuning leads to the best performance with an AUC = 0.94, as well as the sensitivity of 91% and an overall accuracy of 88%. CONCLUSIONS: Model modification, model integration, and transfer learning can play important roles to identify and generate optimal deep CNN models in classifying pulmonary nodules based on CT images efficiently. Transfer learning is preferred when applying deep learning to medical imaging applications.
AB - BACKGROUND: Deep learning has made spectacular achievements in analysing natural images, but it faces challenges for medical applications partly due to inadequate images. OBJECTIVE: Aiming to classify malignant and benign pulmonary nodules using CT images, we explore different strategies to utilize the state-of-the-art deep convolutional neural networks (CNN). METHODS: Experiments are conducted using the Lung Image Database Consortium image collection (LIDC-IDRI), which is a public database containing 1018 cases. Three strategies are implemented including to 1) modify some state-of-the-art CNN architectures, 2) integrate different CNNs and 3) adopt transfer learning. Totally, 11 deep CNN models are compared using the same dataset. RESULTS: Study demonstrates that, for the model modification scheme, a concise CifarNet performs better than the other modified CNNs with more complex architectures, achieving an area under ROC curve of AUC = 0.90. Integrated CNN models do not significantly improve the classification performance, but the model complexity is reduced. Transfer learning outperforms the other two schemes and ResNet with fine-tuning leads to the best performance with an AUC = 0.94, as well as the sensitivity of 91% and an overall accuracy of 88%. CONCLUSIONS: Model modification, model integration, and transfer learning can play important roles to identify and generate optimal deep CNN models in classifying pulmonary nodules based on CT images efficiently. Transfer learning is preferred when applying deep learning to medical imaging applications.
KW - Convolutional neural networks
KW - deep learning
KW - lung cancer
KW - nodule classification
KW - transfer learning
UR - http://www.scopus.com/inward/record.url?scp=85072385059&partnerID=8YFLogxK
UR - http://www.scopus.com/inward/citedby.url?scp=85072385059&partnerID=8YFLogxK
U2 - 10.3233/XST-180490
DO - 10.3233/XST-180490
M3 - Article
C2 - 31227682
AN - SCOPUS:85072385059
SN - 0895-3996
VL - 27
SP - 615
EP - 629
JO - Journal of X-Ray Science and Technology
JF - Journal of X-Ray Science and Technology
IS - 4
ER -