합성곱 신경망을 활용한 위내시경 이미지 분류에서전이학습의 효용성 평가
Evaluation of Transfer Learning in Gastroscopy Image Classification using Convolutional Neual Network
0
조회
0
다운로드
4
인용
0
참고문헌
키워드
초록
Stomach cancer is the most diagnosed cancer in Korea. When gastric cancer is detected early, the 5-year survival rate is as high as 90%. Gastroscopy is a very useful method for early diagnosis. But the false negative rate of gastric cancer in the gastroscopy was 4.6~25.8% due to the subjective judgment of the physician. Recently, the image classification performance of the image recognition field has been advanced by the convolutional neural network. Convolutional neural networks perform well when diverse and sufficient amounts of data are supported. However, medical data is not easy to access and it is difficult to gather enough high-quality data that includes expert annotations. So This paper evaluates the efficacy of transfer learning in gastroscopy classification and diagnosis. We obtained 787 endoscopic images of gastric endoscopy at Gil Medical Center, Gachon University. The number of normal images was 200, and the number of abnormal images was 587. The image size was reconstructed and normalized. In the case of the ResNet50 structure, the classification accuracy before and after applying the transfer learning was improved from 0.9 to 0.947, and the AUC was also improved from 0.94 to 0.98. In the case of the InceptionV3 structure, the classification accuracy before and after applying the transfer learning was improved from 0.862 to 0.924, and the AUC was also improved from 0.89 to 0.97. In the case of the VGG16 structure, the classification accuracy before and after applying the transfer learning was improved from 0.87 to 0.938, and the AUC was also improved from 0.89 to 0.98. The difference in the performance of the CNN model before and after transfer learning was statistically significant when confirmed by T-test (p < 0.05). As a result, transfer learning is judged to be an effective method of medical data that is difficult to collect good quality data.
Abstract
Stomach cancer is the most diagnosed cancer in Korea. When gastric cancer is detected early, the 5-year survival rate is as high as 90%. Gastroscopy is a very useful method for early diagnosis. But the false negative rate of gastric cancer in the gastroscopy was 4.6~25.8% due to the subjective judgment of the physician. Recently, the image classification performance of the image recognition field has been advanced by the convolutional neural network. Convolutional neural networks perform well when diverse and sufficient amounts of data are supported. However, medical data is not easy to access and it is difficult to gather enough high-quality data that includes expert annotations. So This paper evaluates the efficacy of transfer learning in gastroscopy classification and diagnosis. We obtained 787 endoscopic images of gastric endoscopy at Gil Medical Center, Gachon University. The number of normal images was 200, and the number of abnormal images was 587. The image size was reconstructed and normalized. In the case of the ResNet50 structure, the classification accuracy before and after applying the transfer learning was improved from 0.9 to 0.947, and the AUC was also improved from 0.94 to 0.98. In the case of the InceptionV3 structure, the classification accuracy before and after applying the transfer learning was improved from 0.862 to 0.924, and the AUC was also improved from 0.89 to 0.97. In the case of the VGG16 structure, the classification accuracy before and after applying the transfer learning was improved from 0.87 to 0.938, and the AUC was also improved from 0.89 to 0.98. The difference in the performance of the CNN model before and after transfer learning was statistically significant when confirmed by T-test (p < 0.05). As a result, transfer learning is judged to be an effective method of medical data that is difficult to collect good quality data.
참고문헌 (25)
- H. Katai, "Five-year survival analysis of surgically resected gastric cancer cases in Japan: a retrospective analysis of more than 100,000 patients from the nationwide registry of the Japanese Gastric Cancer Association (2001-2007)", Gastric Cancer, 21(1), pp.144~154 (2018)
- 박현아, "위암 검진 권고안", 대한의사협회지, 58(5), pp.373~384 (2015), 대한의사협회[DOI]
- T. Hirasawa, "Application of artificial intelligence using a convolutional neural network for detecting gastric cancer in endoscopic images", Gastric Cancer, (0123456789), pp.1~8 (2018)
- S. Menon, "How commonly is upper gastrointestinal cancer missed at endoscopy? A meta-analysis", Endosc. Int. Open, 02(02), pp.E46~E50 (2014)
- Y. Shimodate, "Gastric superficial neoplasia : high miss rate but slow progression", pp.722~726 (2017)
- K. Y. Hosokawa O, "Difference in accuracy between gastroscopy and colonoscopy for detection of cancer", Hepatogastroenterology, 54, pp.442~444 (2007)
- M. Häfner, "Combining Gaussian Markov random fields with the discretewavelet transform for endoscopic image classification", DSP 2009 16th Int. Conf. Digit. Signal Process. Proc., pp.1~6 (2009)
- P. Wang, "Classification of endoscopic images based on texture and neural network", Annu. Reports Res. React. Institute, Kyoto Univ, 4, pp.3691~3695 (2001)
- G. H. Yann LeCun, "Deep learning", Nature, 521, pp.436~444 (2015)
- M. I. Razzak, "Deep Learning for Medical Image Processing : Overview, Challenges and Future", CoRR, 1704.06825, pp.1~30 (2017)
- A. Krizhevsky, "ImageNet Classification with Deep Convolutional Neural Networks", Adv. Neural Inf. Process. Syst., pp.1~9 (2012)
- K. He, "Delving Deep into Rectifiers : Surpassing Human-Level Performance on ImageNet Classification", ~ (2014)
- V. Gulshan, "Development and validation of a deep learning algorithm for detection of diabetic retinopathy in retinal fundus photographs", JAMA-J. Am. Med. Assoc., 316(22), pp.2402~2410 (2016)
- A. Esteva, "Dermatologist-level classification of skin cancer with deep neural networks", Nature, 542(7639), pp.115~118 (2017)
- H. C. Shin, "Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning", IEEE Trans. Med. Imaging, 35(5), pp.1285~1298 (2016)
- G. Wimmer, "CNN Transfer Learning for the Automated Diagnosis of Celiac Disease" (2016)
- F. Zhang, "Deep classification of vehicle makers and models: The effectiveness of pre-training and data enhancement", 2015 IEEE Int. Conf. Robot. Biomimetics, IEEE-ROBIO 2015, pp.231~236 (2015)
- K. Simonyan, "Very Deep Convolutional Networks for Large-Scale Image Recognition", 1~14 (2014)
- C. Szegedy, "Rethinking the Inception Architecture for Computer Vision", ~ (2015)
- K. He, "Deep Residual Learning for Image Recognition", 2016 IEEE Conf. Comput. Vis. Pattern Recognit., pp.770~778 (2016)
- L. F. -F. Olga Russakovsky, "ImageNet Large Scale Visual Recognition Challenge", Int. J. Comput. Vis, 115, pp.211~252 (2015)
- A. Y. Ng, "Preventing ‘Overfitting’ of Cross-Validation data", CEUR Workshop Proc., 1542, pp.33~36 (2015)
- R. R. Selvaraju, "Grad-CAM : Why did you say that?", 1~4 (2016)
- M. Lin, "Network In Network", 1~10 (2013)