UMLF-COVID: an unsupervised meta-learning model specifically designed to identify X-ray images of COVID-19 patients

BMC Med Imaging. 2021 Nov 22;21(1):174. doi: 10.1186/s12880-021-00704-2.

Abstract

Background: With the rapid spread of COVID-19 worldwide, quick screening for possible COVID-19 patients has become the focus of international researchers. Recently, many deep learning-based Computed Tomography (CT) image/X-ray image fast screening models for potential COVID-19 patients have been proposed. However, the existing models still have two main problems. First, most of the existing supervised models are based on pre-trained model parameters. The pre-training model needs to be constructed on a dataset with features similar to those in COVID-19 X-ray images, which limits the construction and use of the model. Second, the number of categories based on the X-ray dataset of COVID-19 and other pneumonia patients is usually imbalanced. In addition, the quality is difficult to distinguish, leading to non-ideal results with the existing model in the multi-class classification COVID-19 recognition task. Moreover, no researchers have proposed a COVID-19 X-ray image learning model based on unsupervised meta-learning.

Methods: This paper first constructed an unsupervised meta-learning model for fast screening of COVID-19 patients (UMLF-COVID). This model does not require a pre-trained model, which solves the limitation problem of model construction, and the proposed unsupervised meta-learning framework solves the problem of sample imbalance and sample quality.

Results: The UMLF-COVID model is tested on two real datasets, each of which builds a three-category and four-category model. And the experimental results show that the accuracy of the UMLF-COVID model is 3-10% higher than that of the existing models.

Conclusion: In summary, we believe that the UMLF-COVID model is a good complement to COVID-19 X-ray fast screening models.

Keywords: CNN; COVID-19; UMLF-COVID; X-ray.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • COVID-19 / diagnostic imaging*
  • Datasets as Topic / statistics & numerical data
  • Deep Learning*
  • Humans
  • Image Processing, Computer-Assisted
  • SARS-CoV-2
  • Tomography, X-Ray Computed / methods*