M3Lung-Sys: A Deep Learning System for Multi-Class Lung Pneumonia Screening from CT Imaging


Fudan University, China     Shanghai Public Health Clinical Center, Fudan University, China
Inception Institute of Artificial Intelligence (IIAI), Abu Dhabi, UAE


Abstract

To counter the outbreak of COVID-19, the accurate diagnosis of suspected cases plays a crucial role in timely quarantine, medical treatment, and preventing the spread of the pandemic. Considering the limited training cases and resources (e.g, time and budget), we propose a Multi-task Multi-slice Deep Learning System (M3Lung-Sys) for multi-class lung pneumonia screening from CT imaging, which only consists of two 2D CNN networks, i.e., sliceand patient-level classification networks. The former aims to seek the feature representations from abundant CT slices instead of limited CT volumes, and for the overall pneumonia screening, the latter one could recover the temporal information by feature refinement and aggregation between different slices. In addition to distinguish COVID-19 from Healthy, H1N1, and CAP cases, our M3Lung-Sys also be able to locate the areas of relevant lesions, without any pixel-level annotation. To further demonstrate the effectiveness of our model, we conduct extensive experiments on a chest CT imaging dataset with a total of 734 patients (251 healthy people, 245 COVID-19 patients, 105 H1N1 patients, and 133 CAP patients). The quantitative results with plenty of metrics indicate the superiority of our proposed model on both slice- and patient-level classification tasks. More importantly, the generated lesion location maps make our system interpretable and more valuable to clinicians.



Dataset




The Ethics Committee of Shanghai Public Health Clinical Center, Fudan University approved the protocol of this study and waived the requirement for patient-informed consent (YJ-2020-S035-01). A search through the medical records in our hospital information system was conducted, with the final dataset consisting of 245 patients with COVID-19 pneumonia, 105 patients with H1N1 pneumonia, 133 patients with CAP and 251 healthy subjects with non-pneumonia. Of the 734 enrolled people with 415 (56.5%) men, the mean age was 41.8±15.9 years (range, 2 ~ 96 years).



M3Lung-Sys

The proposed Multi-task Multi-slice Deep Learning System (M3Lung-Sys) consists of two components, the Image Preprocessing and Classification Network.


M3Lung-Sys Framework

Specifically, the Image Preprocessing receives raw CT exams, and prepare them for model training or inference. For the Classification Network, we divide it into two subnets (stages), i.e., slice-level and patient-level classification networks, with the purpose of jointly COVID-19 screening and location.Slice-level classification network is trained with multiple CT slice images and predicts the corresponding slice-level categories, i.e., Healthy, COVID-19, H1N1 or CAP. Besides, patient-level classification network only has four layers, and takes a volume of CT slice features, instead of images as input, which are extracted by the former network, to output the patient-level labels. Both classification networks are trained separately due to different tasks, but can be used concurrently in an end-to-end manner for the efficiency. More importantly, for cases classified as positive (i.e., COVID-19, H1N1 or CAP), our system can locate suspected area of abnormality without any pixel-level annotations.



Results











The visualizations of our system outputs for one COVID-19 patient. We show a CT sequence by sampling every five slices. The lesion location maps, with the predicted slicelevel diagnosis on the upper right, are attached at the bottom of the corresponding raw CT slices. At the end of the sequence, the probability of each category, predicted by the patient-level classification network is provided, i.e., 0: Healthy; 1: COVID-19; 2: H1N1; 3: CAP.



Paper

M3Lung-Sys: A Deep Learning System for Multi-Class Lung Pneumonia Screening from CT Imaging

Xuelin Qian, Huazhu Fu, Weiya Shi, Tao Chen, Yanwei Fu*, Fei Shan, and Xiangyang Xue

[Paper] [Bibtex]



Acknowledgements

The website is modified from this template.