DOI QR코드

DOI QR Code

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)
  • 김한웅 (연세대학교 의과대학 의학공학교실) ;
  • 김병남 (연세대학교 의과대학 의학공학교실) ;
  • 이지은 (연세대학교 의과대학 의학공학교실) ;
  • 장원석 (연세대학교 의과대학 의학공학교실) ;
  • 유선국 (연세대학교 의과대학 의학공학교실)
  • Received : 2017.07.25
  • Accepted : 2017.10.11
  • Published : 2017.10.31

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

References

  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. https://doi.org/10.1148/radiol.2302030049
  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. https://doi.org/10.1118/1.3528204
  8. Romans Lois E, Computed tomography for technologists: a comprehensive text, Baltimore, Maryland: Wolters Kluwer Health, 2010, pp. 36.