khosravi-uncertaintyawaretransfer-2021.pdf (9.29 MB)
An uncertainty-aware transfer learning-based framework for covid-19 diagnosis
journal contribution
posted on 2021-04-01, 00:00 authored by A Shamsi, H Asgharnezhad, S S Jokandan, Abbas KhosraviAbbas Khosravi, P M Kebria, Darius Nahavandi, Saeid Nahavandi, D SrinivasanThe early and reliable detection of COVID-19 infected patients is essential to prevent and limit its outbreak. The PCR tests for COVID-19 detection are not available in many countries, and also, there are genuine concerns about their reliability and performance. Motivated by these shortcomings, this article proposes a deep uncertainty-aware transfer learning framework for COVID-19 detection using medical images. Four popular convolutional neural networks (CNNs), including VGG16, ResNet50, DenseNet121, and InceptionResNetV2, are first applied to extract deep features from chest X-ray and computed tomography (CT) images. Extracted features are then processed by different machine learning and statistical modeling techniques to identify COVID-19 cases. We also calculate and report the epistemic uncertainty of classification results to identify regions where the trained models are not confident about their decisions (out of distribution problem). Comprehensive simulation results for X-ray and CT image data sets indicate that linear support vector machine and neural network models achieve the best results as measured by accuracy, sensitivity, specificity, and area under the receiver operating characteristic (ROC) curve (AUC). Also, it is found that predictive uncertainty estimates are much higher for CT images compared to X-ray images.
History
Journal
IEEE transactions on neural networks and learning systemsVolume
32Issue
4Pagination
1408 - 1417Publisher
IEEELocation
Piscataway, N.J.Publisher DOI
Link to full text
ISSN
2162-237XeISSN
2162-2388Language
EnglishPublication classification
C1 Refereed article in a scholarly journalUsage metrics
Categories
No categories selectedKeywords
ClassificationComputed tomographyComputer ScienceComputer Science, Artificial IntelligenceComputer Science, Hardware & ArchitectureComputer Science, Theory & MethodsCOVID-19Data modelsdeep learningEngineeringEngineering, Electrical & ElectronicFeature extractionScience & TechnologyTechnologyTrainingtransfer learningUncertaintyuncertainty quantificationX-ray imaging
Licence
Exports
RefWorks
BibTeX
Ref. manager
Endnote
DataCite
NLM
DC