Deep Learning-Based COVID-19 Diagnostics of Low-Quality CT Images

• Daniel Ferber UNICAMP
• Felipe Vieira UNICAMP
• João Dalben UNICAMP
• Mariana Ferraz UNICAMP
• Nicholas Sato UNICAMP
• Gabriel Oliveira UNICAMP
• Rafael Padilha UNICAMP
• Zanoni Dias UNICAMP

Resumo

Mass testing of the population is among the most effective measures to combat the COVID-19 pandemic. Among existing diagnostic methods, deep learning-based solutions have the potential to be affordable, quick and accurate. However, these techniques often rely on high-quality datasets, which are not always available in medical scenarios. In this work, we use convolutional neural networks to diagnose COVID-19 on computed tomography images from the COVIDx-CT dataset. The available scans often present noisy artifacts, originated from sensor- and capturing-related errors, that can negatively impact the performance of the model if left untreated. In this sense, we explore several preprocessing strategies to reduce their impact and obtain a more accurate method. Our best model, a ResNet50 fine-tuned with preprocessed images, obtained 97.84% accuracy when prompted with a single image and 99.50% when processing multiple images from the same patient. In addition to achieving high accuracy, interpretability experiments show that the network correctly learned features from the lung and chest area.

Palavras-chave: COVID-19 diagnostic, CT image analysis, Deep Learning

Referências

Anthimopoulos, M., Christodoulidis, S., Ebner, L., Christe, A., Mougiakakou, S.: Lung pattern classification for interstitial lung diseases using a deep convolutional neural network. IEEE Trans. Med. Imaging 35(5), 1207–1216 (2016)

Ardakani, A.A., Kanafi, A.R., Acharya, U.R., Khadem, N., Mohammadi, A.: Application of deep learning technique to manage COVID-19 in routine clinical practice using CT images: Results of 10 convolutional neural networks. Comput. Biol. Med. 121, 103795 (2020)

Borakati, A., Perera, A., Johnson, J., Sood, T.: Diagnostic accuracy of X-ray versus CT in COVID-19: a propensity-matched database study. Br. Med. J. Open Access (BMJ Open) 10(11), e042946 (2020)

Brinati, D., Campagner, A., Ferrari, D., Locatelli, M., Banfi, G., Cabitza, F.: Detection of COVID-19 infection from routine blood exams with machine learning: a feasibility study. J. Med. Syst. 44(8), 1–12 (2020)

Deng, J., Dong, W., Socher, R., Li, L.J., Li, K., Fei-Fei, L.: ImageNet: a large-scale hierarchical image database. In: IEEE International Conference on Computer Vision and Pattern Recognition (CVPR), pp. 248–255 (2009)

Gunraj, H., Wang, L., Wong, A.: COVIDNet-CT: A tailored deep convolutional neural network design for detection of COVID-19 cases from chest CT images. Front. Med. 7 (2020)

He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: IEEE International Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778 (2016)

Litjens, G., et al.: A survey on deep learning in medical image analysis. Med. Image Anal. 42, 60–88 (2017)

Mei, X., et al.: Artificial intelligence-enabled rapid diagnosis of patients with COVID-19. Nat. Med. 26(8), 1224–1228 (2020)

Oliveira, G., et al.: COVID-19 X-ray image diagnostic with deep neural networks. In: BSB 2020. LNCS, vol. 12558, pp. 57–68. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-65775-8_6

Raghu, M., Zhang, C., Kleinberg, J., Bengio, S.: Transfusion: understanding transfer learning for medical imaging. In: Advances in Neural Information Processing Systems (NIPS), pp. 3347–3357 (2019)

Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., Batra, D.: Grad-cam: Visual explanations from deep networks via gradient-based localization. In: IEEE International Conference on Computer Vision (ICCV), pp. 618–626 (2017)

Shi, F., et al.: Review of artificial intelligence techniques in imaging data acquisition, segmentation, and diagnosis for COVID-19. IEEE Rev. Biomed. Eng. 14, 4–15 (2020)

Smyrlaki, I., et al.: Massive and rapid COVID-19 testing is feasible by extraction-free SARS-CoV-2 RT-PCR. Nat. Commun. 11(1), 1–12 (2020)

Vandenberg, O., Martiny, D., Rochas, O., van Belkum, A., Kozlakidis, Z.: Considerations for diagnostic COVID-19 tests. Nat. Rev. Microbiol. 19(3), 171–183 (2021)

Xu, X., et al.: A deep learning system to screen novel coronavirus disease 2019 pneumonia. Engineering 6(10), 1122–1129 (2020)