An end-to-end framework for diagnosing COVID-19 pneumonia via Parallel Recursive MLP module and Bi-LTSM correlation

Yiwen Liu, Wenyu Xing, Mingbo Zhao, Mingquan Lin

Research output: Contribution to journalConference articlepeer-review

1 Scopus citations

Abstract

To fully extract the feature information of lung parenchyma in Chest X-ray (CXR) images and realize the auxiliary diagnosis of Corona Virus Disease 2019 (COVID-19) pneumonia, this paper proposed an end-to-end deep learning model, which is mainly composed of object detection, depth feature generation, and multi-channel fusion classification. Firstly, the convolutional neural network (CNN) and region proposal network (RPN)-based object detection module was adopted to detect chest cavity region of interest (ROI). Then, according to the obtained coordinate information of ROI and the convolution feature map of original image, the new convolution feature maps of ROI were obtained with number of 13. By screening 4 representative feature maps form 4 convolution layers with different receptive fields and combining with original ROI image, the 5-dimensional (5D) feature maps were generated as the multi-channel input of classification module. Moreover, in each channel of classification module, three pyramidal recursive multilayer perceptrons (RMLPs) were employed to achieve cross-scale and cross-channel feature analysis. Finally, the correlation analysis of multi-channel output was realized by bi-directional long short memory (Bi-LSTM) module, and the auxiliary diagnosis of pneumonia disease was realized through fully connected layer and SoftMax function. Experimental results show that the proposed model has better classification performance and diagnosis effect than previous methods, with great clinical application potential.

Original languageEnglish (US)
Pages (from-to)416-425
Number of pages10
JournalProceedings of Machine Learning Research
Volume227
StatePublished - 2023
Externally publishedYes
Event6th International Conference on Medical Imaging with Deep Learning, MIDL 2023 - Nashville, United States
Duration: Jul 10 2023Jul 12 2023

Bibliographical note

Publisher Copyright:
© 2023 CC-BY 4.0, Y. Liu, W. Xing, M. Zhao & M. Lin.

Keywords

  • deep learning
  • end-to-end framework
  • multi-task learning
  • Pneumonia
  • RMLP-Bi-LSTM

Fingerprint

Dive into the research topics of 'An end-to-end framework for diagnosing COVID-19 pneumonia via Parallel Recursive MLP module and Bi-LTSM correlation'. Together they form a unique fingerprint.

Cite this