COVID-MTL: Multitask Learning with Shift3D and Random-weighted Loss for
Automated Diagnosis and Severity Assessment of COVID-19
- URL: http://arxiv.org/abs/2012.05509v3
- Date: Thu, 31 Dec 2020 14:27:16 GMT
- Title: COVID-MTL: Multitask Learning with Shift3D and Random-weighted Loss for
Automated Diagnosis and Severity Assessment of COVID-19
- Authors: Guoqing Bao, Huai Chen, Tongliang Liu, Guanzhong Gong, Yong Yin,
Lisheng Wang and Xiuying Wang
- Abstract summary: There is an urgent need for automated methods to assist accurate and effective assessment of COVID-19.
We present an end-to-end multitask learning framework (COVID-MTL) that is capable of automated and simultaneous detection (against both radiology and NAT) and severity assessment of COVID-19.
- Score: 39.57518533765393
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: There is an urgent need for automated methods to assist accurate and
effective assessment of COVID-19. Radiology and nucleic acid test (NAT) are
complementary COVID-19 diagnosis methods. In this paper, we present an
end-to-end multitask learning (MTL) framework (COVID-MTL) that is capable of
automated and simultaneous detection (against both radiology and NAT) and
severity assessment of COVID-19. COVID-MTL learns different COVID-19 tasks in
parallel through our novel random-weighted loss function, which assigns
learning weights under Dirichlet distribution to prevent task dominance; our
new 3D real-time augmentation algorithm (Shift3D) introduces space variances
for 3D CNN components by shifting low-level feature representations of
volumetric inputs in three dimensions; thereby, the MTL framework is able to
accelerate convergence and improve joint learning performance compared to
single-task models. By only using chest CT scans, COVID-MTL was trained on 930
CT scans and tested on separate 399 cases. COVID-MTL achieved AUCs of 0.939 and
0.846, and accuracies of 90.23% and 79.20% for detection of COVID-19 against
radiology and NAT, respectively, which outperformed the state-of-the-art
models. Meanwhile, COVID-MTL yielded AUC of 0.800 $\pm$ 0.020 and 0.813 $\pm$
0.021 (with transfer learning) for classifying control/suspected, mild/regular,
and severe/critically-ill cases. To decipher the recognition mechanism, we also
identified high-throughput lung features that were significantly related (P <
0.001) to the positivity and severity of COVID-19.
Related papers
- Auto-Lesion Segmentation with a Novel Intensity Dark Channel Prior for
COVID-19 Detection [0.24578723416255752]
This study develops a CT-based radiomics framework for differentiation of COVID-19 from other lung diseases.
The model categorizes images into three classes: COVID-19, non-COVID-19, or normal.
The best performing classification model, Residual Neural Network with 50 layers (Resnet-50), attained an average accuracy, precision, recall, and F1-score of 98.8%, 99%, 98%, and 98% respectively.
arXiv Detail & Related papers (2023-09-22T06:09:48Z) - A Novel Implementation of Machine Learning for the Efficient,
Explainable Diagnosis of COVID-19 from Chest CT [0.0]
The aim of this study was to take a novel approach in the machine learning-based detection of COVID-19 from chest CT scans.
The proposed model attained an overall accuracy of 0.927 and a sensitivity of 0.958.
arXiv Detail & Related papers (2022-06-15T18:35:22Z) - COVID-19 identification from volumetric chest CT scans using a
progressively resized 3D-CNN incorporating segmentation, augmentation, and
class-rebalancing [4.446085353384894]
COVID-19 is a global pandemic disease overgrowing worldwide.
Computer-aided screening tools with greater sensitivity is imperative for disease diagnosis and prognosis.
This article proposes a 3D Convolutional Neural Network (CNN)-based classification approach.
arXiv Detail & Related papers (2021-02-11T18:16:18Z) - A novel multiple instance learning framework for COVID-19 severity
assessment via data augmentation and self-supervised learning [64.90342559393275]
How to fast and accurately assess the severity level of COVID-19 is an essential problem, when millions of people are suffering from the pandemic around the world.
We observe that there are two issues -- weak annotation and insufficient data that may obstruct automatic COVID-19 severity assessment with CT images.
Our method could obtain an average accuracy of 95.8%, with 93.6% sensitivity and 96.4% specificity, which outperformed previous works.
arXiv Detail & Related papers (2021-02-07T16:30:18Z) - Classification of COVID-19 X-ray Images Using a Combination of Deep and
Handcrafted Features [0.0]
We use a combination of deep convolutional and handcrafted features extracted from X-ray chest scans to discriminate between healthy, common pneumonia, and COVID-19 patients.
We achieve an accuracy of 0.988 in the classification task with our combined approach compared to 0.963 and 0.983 accuracy for the handcrafted features with SVM and CNN respectively.
arXiv Detail & Related papers (2021-01-19T21:09:46Z) - Automated Model Design and Benchmarking of 3D Deep Learning Models for
COVID-19 Detection with Chest CT Scans [72.04652116817238]
We propose a differentiable neural architecture search (DNAS) framework to automatically search for the 3D DL models for 3D chest CT scans classification.
We also exploit the Class Activation Mapping (CAM) technique on our models to provide the interpretability of the results.
arXiv Detail & Related papers (2021-01-14T03:45:01Z) - M3Lung-Sys: A Deep Learning System for Multi-Class Lung Pneumonia
Screening from CT Imaging [85.00066186644466]
We propose a Multi-task Multi-slice Deep Learning System (M3Lung-Sys) for multi-class lung pneumonia screening from CT imaging.
In addition to distinguish COVID-19 from Healthy, H1N1, and CAP cases, our M 3 Lung-Sys also be able to locate the areas of relevant lesions.
arXiv Detail & Related papers (2020-10-07T06:22:24Z) - Classification of COVID-19 in CT Scans using Multi-Source Transfer
Learning [91.3755431537592]
We propose the use of Multi-Source Transfer Learning to improve upon traditional Transfer Learning for the classification of COVID-19 from CT scans.
With our multi-source fine-tuning approach, our models outperformed baseline models fine-tuned with ImageNet.
Our best performing model was able to achieve an accuracy of 0.893 and a Recall score of 0.897, outperforming its baseline Recall score by 9.3%.
arXiv Detail & Related papers (2020-09-22T11:53:06Z) - CovidDeep: SARS-CoV-2/COVID-19 Test Based on Wearable Medical Sensors
and Efficient Neural Networks [51.589769497681175]
The novel coronavirus (SARS-CoV-2) has led to a pandemic.
The current testing regime based on Reverse Transcription-Polymerase Chain Reaction for SARS-CoV-2 has been unable to keep up with testing demands.
We propose a framework called CovidDeep that combines efficient DNNs with commercially available WMSs for pervasive testing of the virus.
arXiv Detail & Related papers (2020-07-20T21:47:28Z) - Machine Learning Automatically Detects COVID-19 using Chest CTs in a
Large Multicenter Cohort [43.99203831722203]
Our retrospective study obtained 2096 chest CTs from 16 institutions.
A metric-based approach for classification of COVID-19 used interpretable features.
A deep learning-based classifier differentiated COVID-19 via 3D features extracted from CT attenuation and probability distribution of airspace opacities.
arXiv Detail & Related papers (2020-06-09T00:40:35Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.