COVID-19 disease identification network based on weakly supervised feature selection

Math Biosci Eng. 2023 Mar 16;20(5):9327-9348. doi: 10.3934/mbe.2023409.

Abstract

The coronavirus disease 2019 (COVID-19) outbreak has resulted in countless infections and deaths worldwide, posing increasing challenges for the health care system. The use of artificial intelligence to assist in diagnosis not only had a high accuracy rate but also saved time and effort in the sudden outbreak phase with the lack of doctors and medical equipment. This study aimed to propose a weakly supervised COVID-19 classification network (W-COVNet). This network was divided into three main modules: weakly supervised feature selection module (W-FS), deep learning bilinear feature fusion module (DBFF) and Grad-CAM++ based network visualization module (Grad-Ⅴ). The first module, W-FS, mainly removed redundant background features from computed tomography (CT) images, performed feature selection and retained core feature regions. The second module, DBFF, mainly used two symmetric networks to extract different features and thus obtain rich complementary features. The third module, Grad-Ⅴ, allowed the visualization of lesions in unlabeled images. A fivefold cross-validation experiment showed an average classification accuracy of 85.3%, and a comparison with seven advanced classification models showed that our proposed network had a better performance.

Keywords: COVID-19; Classification; deep learning; feature selection; weakly supervised.

Publication types

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

MeSH terms

  • COVID-19* / classification
  • COVID-19* / diagnostic imaging
  • Datasets as Topic
  • Humans
  • Neural Networks, Computer*
  • Supervised Machine Learning*