Skip to main content

Few-Shot Image Classification for Automatic COVID-19 Diagnosis

  • Conference paper
  • First Online:
Pattern Recognition and Image Analysis (IbPRIA 2023)

Abstract

Developing robust and performant methods for diagnosing COVID-19, particularly for triaging processes, is crucial. This study introduces a completely automated system to detect COVID-19 by means of the analysis of Chest X-Ray scans (CXR). The proposed methodology is based on few-shot techniques, enabling to work on small image datasets. Moreover, a set of additions have been done to enhance the diagnostic capabilities. First, a network to extract the lung region to rely only on the most relevant image area. Second, a new cost function to penalize each misclassification according to the clinical consequences. Third, a system to combine different predictions from the same image to increase the robustness of the diagnoses. The proposed approach was validated on the public dataset COVIDGR-1.0, yielding a classification accuracy of 79.10% ± 3.41% and, thus, outperforming other state-of-the-art methods. In conclusion, the proposed methodology has proven to be suitable for the diagnosis of COVID-19.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 89.00
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 119.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

References

  1. Afshar, P., Heidarian, S., Naderkhani, F., Oikonomou, A., Plataniotis, K.N., Mohammadi, A.: COVID-CAPS: a capsule network-based framework for identification of COVID-19 cases from X-ray images. Pattern Recogn. Lett. 138, 638–43 (2020)

    Article  Google Scholar 

  2. Chowdhury, M.E., et al.: Can AI help in screening viral and COVID-19 pneumonia? IEEE Access 8, 132665–76 (2020)

    Article  Google Scholar 

  3. Cohen, J.P., Morrison, P., Dao, L., Roth, K., Duong, T.Q., Ghassemi, M.: COVID-19 image data collection: prospective predictions are the future. arXiv 2006.11988 (2020). https://github.com/ieee8023/covid-chestxray-dataset

  4. Cui, H., Wei, D., Ma, K., Gu, S., Zheng, Y.: A unified framework for generalized low-shot medical image segmentation with scarce data. IEEE Trans. Med. Imaging 40(10), 2656–2671 (2020)

    Article  Google Scholar 

  5. Fernández-Miranda, P.M., et al.: Developing a training web application for improving the COVID-19 diagnostic accuracy on chest X-ray. J. Digit. Imaging 34(2), 242–256 (2021)

    Article  Google Scholar 

  6. Huang, C., et al.: Clinical features of patients infected with 2019 novel coronavirus in Wuhan. China Lancet 395(10223), 497–506 (2020)

    Article  Google Scholar 

  7. Jadon, S.: COVID-19 detection from scarce chest X-ray image data using few-shot deep learning approach. In: Proceedings of the SPIE, vol. 11601, p. 116010X (2021)

    Google Scholar 

  8. Jaeger, S., Candemir, S., Antani, S., Wáng, Y.X.J., Lu, P.X., Thoma, G.: Two public chest X-ray datasets for computer-aided screening of pulmonary diseases. Quant. Imaging Med. Surg. 4(6), 475 (2014)

    Google Scholar 

  9. Jin, K.N., Do, K.H., Da Nam, B., Hwang, S.H., Choi, M., Yong, H.S.: Korean clinical imaging guidelines for justification of diagnostic imaging study for COVID-19. J. Korean Soc. Radiol. 83(2), 265–283 (2022)

    Article  Google Scholar 

  10. Kang, H., et al.: Diagnosis of coronavirus disease 2019 (COVID-19) with structured latent multi-view representation learning. IEEE Trans. Med. Imaging 39(8), 2606–2614 (2020)

    Article  Google Scholar 

  11. Khan, A.I., Shah, J.L., Bhat, M.M.: CoroNet: a deep neural network for detection and diagnosis of COVID-19 from chest X-ray images. Comput. Meth. Prog. Bio. 196, 105581 (2020)

    Article  Google Scholar 

  12. Loey, M., Smarandache, F., M Khalifa, N.E.: Within the lack of chest COVID-19 X-ray dataset: a novel detection model based on GAN and deep transfer learning. Symmetry 12(4), 651 (2020)

    Google Scholar 

  13. Nishio, M., Noguchi, S., Matsuo, H., Murakami, T.: Automatic classification between COVID-19 pneumonia, non-COVID-19 pneumonia, and the healthy on chest X-ray image: combination of data augmentation methods. Sci. Rep. 10(1), 1–6 (2020)

    Article  Google Scholar 

  14. Oh, Y., Park, S., Ye, J.C.: Deep learning COVID-19 features on CXR using limited training data sets. IEEE Trans. Med. Imaging 39(8), 2688–700 (2020)

    Article  Google Scholar 

  15. Ouchicha, C., Ammor, O., Meknassi, M.: CVDNet: a novel deep learning architecture for detection of coronavirus (COVID-19) from chest X-ray images. Chaos, Solitons Fractals 140, 110245 (2020)

    Article  MathSciNet  Google Scholar 

  16. Rey-Area, M., Guirado, E., Tabik, S., Ruiz-Hidalgo, J.: FuCiTNet: improving the generalization of deep learning networks by the fusion of learned class-inherent transformations. Inf. Fusion 63, 188–95 (2020)

    Article  Google Scholar 

  17. Roberts, M., et al.: Common pitfalls and recommendations for using machine learning to detect and prognosticate for COVID-19 using chest radiographs and CT scans. Nat. Mach. Intell. 3(3), 199–217 (2021)

    Article  Google Scholar 

  18. Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: Proceedings of the MICCAI, pp. 234–41 (2015)

    Google Scholar 

  19. Rubner, Y., Tomasi, C., Guibas, L.J.: The earth mover’s distance as a metric for image retrieval. Int. J. Comput. Vision 40(2), 99–121 (2000)

    Article  MATH  Google Scholar 

  20. Shiraishi, J., et al.: Development of a digital image database for chest radiographs with and without a lung nodule: receiver operating characteristic analysis of radiologists’ detection of pulmonary nodules. Am. J. Roentgenol. 174(1), 71–74 (2000)

    Article  Google Scholar 

  21. Shorfuzzaman, M., Hossain, M.S.: MetaCOVID: a siamese neural network framework with contrastive loss for n-shot diagnosis of COVID-19 patients. Pattern Recogn. 113, 107700 (2021)

    Article  Google Scholar 

  22. Tabik, S., et al.: COVIDGR dataset and COVID-SDNet methodology for predicting COVID-19 based on chest X-ray images. IEEE J. Biomed. Health 24(12), 3595–605 (2020)

    Article  Google Scholar 

  23. Teixeira, L.O., et al.: Impact of lung segmentation on the diagnosis and explanation of COVID-19 in chest X-ray images. Sensors 21(21), 7116 (2021)

    Article  Google Scholar 

  24. Vinyals, O., Blundell, C., Lillicrap, T., Wierstra, D., et al.: Matching networks for one shot learning. In: Advances in Neural Information Processing Systems, vol. 29, pp. 3630–8 (2016)

    Google Scholar 

  25. Wang, L., Lin, Z.Q., Wong, A.: COVID-Net: a tailored deep convolutional neural network design for detection of COVID-19 cases from chest X-ray images. Sci. Rep. 10(1), 1–12 (2020)

    Google Scholar 

  26. Wang, N., Liu, H., Xu, C.: Deep learning for the detection of COVID-19 using transfer learning and model integration. In: IEEE International Conference on Electronics Information and Emergency Communication, pp. 281–284. IEEE (2020)

    Google Scholar 

  27. Wong, H.Y.F., et al.: Frequency and distribution of chest radiographic findings in COVID-19 positive patients. Radiology 296, 201160 (2020)

    Article  Google Scholar 

Download references

Acknowledgements

This work has received financial support from the Spanish Ministry of Science and Innovation under grant PID2020-112623GB-I00, Consellería de Cultura, Educación e Ordenación Universitaria under grants ED431C 2021/48, ED431G-2019/04, ED481A-2018 and ED431C 2018/29 and the European Regional Development Fund (ERDF), which acknowledges the CiTIUS-Research Center on Intelligent Technologies of the University of Santiago de Compostela as a Research Center of the Galician University System.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Nicolás Vila-Blanco .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Cores, D., Vila-Blanco, N., Mucientes, M., Carreira, M.J. (2023). Few-Shot Image Classification for Automatic COVID-19 Diagnosis. In: Pertusa, A., Gallego, A.J., Sánchez, J.A., Domingues, I. (eds) Pattern Recognition and Image Analysis. IbPRIA 2023. Lecture Notes in Computer Science, vol 14062. Springer, Cham. https://doi.org/10.1007/978-3-031-36616-1_43

Download citation

  • DOI: https://doi.org/10.1007/978-3-031-36616-1_43

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-031-36615-4

  • Online ISBN: 978-3-031-36616-1

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics