Your browser doesn't support javascript.
AutoCovNet: Unsupervised feature learning using autoencoder and feature merging for detection of COVID-19 from chest X-ray images.
Rashid, Nayeeb; Hossain, Md Adnan Faisal; Ali, Mohammad; Islam Sukanya, Mumtahina; Mahmud, Tanvir; Fattah, Shaikh Anowarul.
  • Rashid N; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
  • Hossain MAF; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
  • Ali M; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
  • Islam Sukanya M; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
  • Mahmud T; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
  • Fattah SA; Department of EEE, BUET, ECE Building, West Palashi, Dhaka 1205, Bangladesh.
Biocybern Biomed Eng ; 41(4): 1685-1701, 2021.
Article in English | MEDLINE | ID: covidwho-1471886
ABSTRACT
With the onset of the COVID-19 pandemic, the automated diagnosis has become one of the most trending topics of research for faster mass screening. Deep learning-based approaches have been established as the most promising methods in this regard. However, the limitation of the labeled data is the main bottleneck of the data-hungry deep learning methods. In this paper, a two-stage deep CNN based scheme is proposed to detect COVID-19 from chest X-ray images for achieving optimum performance with limited training images. In the first stage, an encoder-decoder based autoencoder network is proposed, trained on chest X-ray images in an unsupervised manner, and the network learns to reconstruct the X-ray images. An encoder-merging network is proposed for the second stage that consists of different layers of the encoder model followed by a merging network. Here the encoder model is initialized with the weights learned on the first stage and the outputs from different layers of the encoder model are used effectively by being connected to a proposed merging network. An intelligent feature merging scheme is introduced in the proposed merging network. Finally, the encoder-merging network is trained for feature extraction of the X-ray images in a supervised manner and resulting features are used in the classification layers of the proposed architecture. Considering the final classification task, an EfficientNet-B4 network is utilized in both stages. An end to end training is performed for datasets containing classes COVID-19, Normal, Bacterial Pneumonia, Viral Pneumonia. The proposed method offers very satisfactory performances compared to the state of the art methods and achieves an accuracy of 9013% on the 4-class, 9645% on a 3-class, and 9939% on 2-class classification.
Keywords

Full text: Available Collection: International databases Database: MEDLINE Type of study: Diagnostic study Language: English Journal: Biocybern Biomed Eng Year: 2021 Document Type: Article Affiliation country: J.bbe.2021.09.004

Similar

MEDLINE

...
LILACS

LIS


Full text: Available Collection: International databases Database: MEDLINE Type of study: Diagnostic study Language: English Journal: Biocybern Biomed Eng Year: 2021 Document Type: Article Affiliation country: J.bbe.2021.09.004