코로나바이러스 감염증19 데이터베이스에 기반을 둔 인공신경망 모델의 특성 평가
Evaluation of Deep-Learning Feature Based COVID-19 Classifier in Various Neural Network
0
조회
0
다운로드
2
인용
0
참고문헌
키워드
초록
Coronavirus disease(COVID-19) is highly infectious disease that directly affects the lungs. To observe the clinical findings from these lungs, the Chest Radiography(CXR) can be used in a fast manner. However, the diagnostic performance via CXR needs to be improved, since the identifying these findings are highly time-consuming and prone to human error. Therefore, Artificial Intelligence(AI) based tool may be useful to aid the diagnosis of COVID-19 via CXR. In this study, we explored various Deep learning(DL) approach to classify COVID-19, other viral pneumonia and normal. For the original dataset and lung-segmented dataset, the pre-trained AlexNet, SqueezeNet, ResNet18, DenseNet201 were transfer- trained and validated for 3 class - COVID-19, viral pneumonia, normal. In the results, AlexNet showed the highest mean accuracy of 99.15±2.69% and fastest training time of 1.61±0.56 min among 4 pre-trained neural networks. In this study, we demonstrated the performance of 4 pre-trained neural networks in COVID-19 diagnosis with CXR images. Further, we plotted the class activation map(CAM) of each network and demonstrated that the lung-segmentation pre-processing improve the performance of COVID-19 classifier with CXR images by excluding background features.
Abstract
Coronavirus disease(COVID-19) is highly infectious disease that directly affects the lungs. To observe the clinical findings from these lungs, the Chest Radiography(CXR) can be used in a fast manner. However, the diagnostic performance via CXR needs to be improved, since the identifying these findings are highly time-consuming and prone to human error. Therefore, Artificial Intelligence(AI) based tool may be useful to aid the diagnosis of COVID-19 via CXR. In this study, we explored various Deep learning(DL) approach to classify COVID-19, other viral pneumonia and normal. For the original dataset and lung-segmented dataset, the pre-trained AlexNet, SqueezeNet, ResNet18, DenseNet201 were transfer- trained and validated for 3 class - COVID-19, viral pneumonia, normal. In the results, AlexNet showed the highest mean accuracy of 99.15±2.69% and fastest training time of 1.61±0.56 min among 4 pre-trained neural networks. In this study, we demonstrated the performance of 4 pre-trained neural networks in COVID-19 diagnosis with CXR images. Further, we plotted the class activation map(CAM) of each network and demonstrated that the lung-segmentation pre-processing improve the performance of COVID-19 classifier with CXR images by excluding background features.
참고문헌 (18)
- 신형호, "코로나바이러스감염증-19 상황에서 일개 국가지정 의료기관의 이동형 병원 CT 활용 사례", 방사선기술과학, 43(2), pp.71~77 (2020), 대한방사선과학회
- Park M, "A systematic review of COVID-19 epidemiology based on current evidence", Journal of Clinical Medicine, 9(4), pp.967~ (2020)
- Chaudhuri S, "Modeling the role of respiratory droplets in Covid-19 type pandemics", Physics of Fluids, 32(6), pp.063309~ (2020)
- Wikramaratna P, "Estimating false-negative detection rate of SARS-CoV-2 by RT-PCR", medRxiv
- Wong HYF, "Frequency and distribution of chest radiographic findings in COVID-19 positive patients", Radiology, pp.201160~ (2020)
- Oh Y, "Deep learning covid-19 features on cxr using limited training data sets", IEEE Transactions on Medical Imaging (2020)
- Zhang J, "Covid-19screening on chest X-ray images using deep learning based anomaly detection", arXiv preprint arXiv:2003.12338, ~ (2003)
- Brunese L, "Explainable deep learning for pulmonary disease and coronavirus COVID-19 detection from X-rays", Computer Methods and Programs in Biomedicine, 196, pp.105608~ (2020)
- Wang L, "COVID-Net: A tailored deep convolutional neural network design for detection of COVID-19 cases from chest X-ray images", arXiv preprint arXiv:2003.09871, ~
- Suzuki K, "Overview of deep learning in medical imaging", Radiological Physics and Technology, 10(3), pp.257~273 (2017)
- Lundervold AS, "An overview of deep learning in medical imaging focusing on MRI", Zeitschrift für Medizinische Physik, 29(2), pp.102~127 (2019)
- Xu Y, "Deep learning predicts lung cancer treatment response from serial medical imaging", Clinical Cancer Research, 25(11), pp.3266~3275 (2019)
- 홍준용, "AI 의료영상 분석의 개요 및 연구 현황에 대한 고찰", 방사선기술과학, 43(3), pp.195~208 (2020), 대한방사선과학회
- "COVID-19 Radiography Database, Version 1" (2020)[URL]
- Krizhevsky A, "Imagenet classification with deep convolutional neural networks", Advances in Neural Information Processing Systems (2012)
- Iandola FN, "SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and< 0.5 MB model size", arXiv preprint arXiv:1602.07360, ~
- He K, "Deep residual learning for image recognition", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (2016)
- Huang G, "Densely connected convolutional networks", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (2017)