Viral Pneumonia Screening on Chest X-ray Images Using Confidence-Aware Anomaly Detection

Abstract

Cluster of viral pneumonia occurrences during a short period of time may be a harbinger of an outbreak or pandemic, like SARS, MERS, and recent COVID-19. Rapid and accurate detection of viral pneumonia using chest X-ray can be significantly useful in large-scale screening and epidemic prevention, particularly when other chest imaging modalities are less available. Viral pneumonia often have diverse causes and exhibit notably different visual appearances on X-ray images. The evolution of viruses and the emergence of novel mutated viruses further result in substantial dataset shift, which greatly limits the performance of classification approaches. In this paper, we formulate the task of differentiating viral pneumonia from non-viral pneumonia and healthy controls into an one-class classification-based anomaly detection problem, and thus propose the confidence-aware anomaly detection (CAAD) model, which consists of a shared feature extractor, an anomaly detection module, and a confidence prediction module. If the anomaly score produced by the anomaly detection module is large enough or the confidence score estimated by the confidence prediction module is small enough, we accept the input as an anomaly case (i.e., viral pneumonia). The major advantage of our approach over binary classification is that we avoid modeling individual viral pneumonia classes explicitly and treat all known viral pneumonia cases as anomalies to reinforce the one-class model. The proposed model outperforms binary classification models on the clinical X-VIRAL dataset that contains 5,977 viral pneumonia (no COVID-19) cases, 18,619 non-viral pneumonia cases, and 18,774 healthy controls.

Accepted to IEEE Trans. Medical Imaging. 12 pages

References in corpus (10)

• A Simple Framework for Contrastive Learning of Visual Representations
• Covid-19: Automatic detection from X-Ray images utilizing Transfer Learning with Convolutional Neural Networks
• A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networks
• CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning
• COVID-19 Image Data Collection: Prospective Predictions Are the Future
• Rapid AI Development Cycle for the Coronavirus (COVID-19) Pandemic: Initial Results for Automated Detection & Patient Monitoring using Deep Learning CT Image Analysis
• Diagnosis of Coronavirus Disease 2019 (COVID-19) with Structured Latent Multi-View Representation Learning
• Learning Representations of Ultrahigh-dimensional Data for Random Distance-based Outlier Detection
• Deep Mining External Imperfect Data for Chest X-ray Disease Screening
• Learning Diagnosis of COVID-19 from a Single Radiological Image

Cited by in corpus (32)

• Can AI help in screening Viral and COVID-19 pneumonia?
• Contrastive Cross-site Learning with Redesigned Net for COVID-19 CT Classification
• POCOVID-Net: Automatic Detection of COVID-19 From a New Lung Ultrasound Imaging Dataset (POCUS)
• Multi-stage transfer learning for lung segmentation using portable X-ray devices for patients with COVID-19
• Evolving Deep Convolutional Neural Network by Hybrid Sine-Cosine and Extreme Learning Machine for Real-time COVID19 Diagnosis from X-Ray Images
• Decision and Feature Level Fusion of Deep Features Extracted from Public COVID-19 Data-sets
• Medical Imaging and Computational Image Analysis in COVID-19 Diagnosis: A Review
• MiniSeg: An Extremely Minimum Network for Efficient COVID-19 Segmentation
• Detection of Covid-19 From Chest X-ray Images Using Artificial Intelligence: An Early Review
• COVID-19 Screening Using Residual Attention Network an Artificial Intelligence Approach
• Vision Transformer for COVID-19 CXR Diagnosis using Chest X-ray Feature Corpus
• COVID-MobileXpert: On-Device COVID-19 Patient Triage and Follow-up using Chest X-rays
• A Survey on Applications of Artificial Intelligence in Fighting Against COVID-19
• Uncertainty-Aware Semi-Supervised Method Using Large Unlabeled and Limited Labeled COVID-19 Data
• Pay Attention to the cough: Early Diagnosis of COVID-19 using Interpretable Symptoms Embeddings with Cough Sound Signal Processing
• Hypergraph Learning for Identification of COVID-19 with CT Imaging
• Early Screening of SARS-CoV-2 by Intelligent Analysis of X-Ray Images
• Automated Model Design and Benchmarking of 3D Deep Learning Models for COVID-19 Detection with Chest CT Scans
• ESAD: End-to-end Deep Semi-supervised Anomaly Detection
• Learning to Match Distributions for Domain Adaptation
• Label-Free Segmentation of COVID-19 Lesions in Lung CT
• COVID-19 Pneumonia Severity Prediction using Hybrid Convolution-Attention Neural Architectures
• Diagnosis/Prognosis of COVID-19 Images: Challenges, Opportunities, and Applications
• Learning Invariant Representations across Domains and Tasks
• Semi-supervised Learning for COVID-19 Image Classification via ResNet
• Understanding Human Innate Immune System Dependencies using Graph Neural Networks
• Comparative Evaluation of 3D and 2D Deep Learning Techniques for Semantic Segmentation in CT Scans
• Reducing Risk and Uncertainty of Deep Neural Networks on Diagnosing COVID-19 Infection
• A Tale of Two Countries: A Longitudinal Cross-Country Study of Mobile Users' Reactions to the COVID-19 Pandemic Through the Lens of App Popularity
• Recognition of COVID-19 Disease Utilizing X-Ray Imaging of the Chest Using CNN
• DARNet: Dual-Attention Residual Network for Automatic Diagnosis of COVID-19 via CT Images
• Advancement of Deep Learning in Pneumonia and Covid-19 Classification and Localization: A Qualitative and Quantitative Analysis