Browse > Article
http://dx.doi.org/10.9718/JBER.2017.38.5.237

Performance Improvement of Convolutional Neural Network for Pulmonary Nodule Detection  

Kim, HanWoong (Department of Medical Engineering, Yonsei University College of Medicine)
Kim, Byeongnam (Department of Medical Engineering, Yonsei University College of Medicine)
Lee, JeeEun (Department of Medical Engineering, Yonsei University College of Medicine)
Jang, Won Seuk (Department of Medical Engineering, Yonsei University College of Medicine)
Yoo, Sun K. (Department of Medical Engineering, Yonsei University College of Medicine)
Publication Information
Journal of Biomedical Engineering Research / v.38, no.5, 2017 , pp. 237-241 More about this Journal
Abstract
Early detection of the pulmonary nodule is important for diagnosis and treatment of lung cancer. Recently, CT has been used as a screening tool for lung nodule detection. And, it has been reported that computer aided detection(CAD) systems can improve the accuracy of the radiologist in detection nodules on CT scan. The previous study has been proposed a method using Convolutional Neural Network(CNN) in Lung CAD system. But the proposed model has a limitation in accuracy due to its sparse layer structure. Therefore, we propose a Deep Convolutional Neural Network to overcome this limitation. The model proposed in this work is consist of 14 layers including 8 convolutional layers and 4 fully connected layers. The CNN model is trained and tested with 61,404 regions-of-interest (ROIs) patches of lung image including 39,760 nodules and 21,644 non-nodules extracted from the Lung Image Database Consortium(LIDC) dataset. We could obtain the classification accuracy of 91.79% with the CNN model presented in this work. To prevent overfitting, we trained the model with Augmented Dataset and regularization term in the cost function. With L1, L2 regularization at Training process, we obtained 92.39%, 92.52% of accuracy respectively. And we obtained 93.52% with data augmentation. In conclusion, we could obtain the accuracy of 93.75% with L2 Regularization and Data Augmentation.
Keywords
Pulmonary nodule detection; Convolutional Neural Network; Machine learning;
Citations & Related Records
연도 인용수 순위
  • Reference
1 Awai K., Murao K., Ozawa A., Komi M., Hayakawa H., Hori, S., & Nishimura, Y. "Pulmonary nodules at chest CT: effect of computer-aided diagnosis on radiologists' detection performance." Radiology, vol. 230, no. 2, pp. 347-352, 2004.   DOI
2 Liang M. and Hu X. "Recurrent convolutional neural network for object recognition." in Proc. the IEEE Conference on Computer Vision and Pattern Recognitio, Boston, USA, 2015, pp. 3367-3375.
3 Karpathy, Andrej. "Lessons learned from manually classifying CIFAR-10." Published online at http://karpathy.github.io/2011/04/27/manually-classifying-cifar10, 2011.
4 Graham, Benjamin. "Fractional max-pooling." arXiv preprint arXiv:1412.6071, 2014.
5 Li, W., Cao, P., Zhao, D., and Wang, J. "Pulmonary nodule classification with deep convolutional neural networks on computed tomography images." Computational and mathematical methods in medicine, vol. 2016, 7, Article ID: 6215085, 2016.
6 Li Q., Cai W., Wang X., Zhou Y., Feng D. D., and Chen M. "Medical image classification with convolutional neural network." In Proc. Control Automation Robotics & Vision (ICARCV), 2014 13th International Conference on. IEEE, Singapore. 2014, pp. 844-848.
7 Armato S. G., McLennan G., Bidaut L., McNitt-Gray M. F., Meyer C. R., Reeves A. P. and Kazerooni. "The lung image database consortium (LIDC) and image database resource initiative (IDRI): a completed reference database of lung nodules on CT scans." Medical physics, vol. 38, no. 2, pp. 915-931, 2011.   DOI
8 Romans Lois E, Computed tomography for technologists: a comprehensive text, Baltimore, Maryland: Wolters Kluwer Health, 2010, pp. 36.