Abstract:Background: Chest X-ray (CXR) is one of the most commonly performed imaging tests worldwide. Due to its wide usage, there is a growing need for automated and generalizable methods to accurately diagnose these images. Traditional methods for chest X-ray analysis often struggle with generalization across diverse datasets due to variations in imaging protocols, patient demographics, and the presence of overlapping anatomical structures. Therefore, there is a significant demand for advanced diagnostic tools that can consistently identify abnormalities across different patient populations and imaging settings. We propose a method that can provide a generalizable diagnosis of chest X-ray. Method: Our method utilizes an attention-guided decomposer network (ADSC) to extract disease maps from chest X-ray images. The ADSC employs one encoder and multiple decoders, incorporating a novel self-consistency loss to ensure consistent functionality across its modules. The attention-guided encoder captures salient features of abnormalities, while three distinct decoders generate a normal synthesized image, a disease map, and a reconstructed input image, respectively. A discriminator differentiates the real and the synthesized normal chest X-rays, enhancing the quality of generated images. The disease map along with the original chest X-ray image are fed to a DenseNet-121 classifier modified for multi-class classification of the input X-ray. Results: Experimental results on multiple publicly available datasets demonstrate the effectiveness of our approach. For multi-class classification, we achieve up to a 3% improvement in AUROC score for certain abnormalities compared to the existing methods. For binary classification (normal versus abnormal), our method surpasses existing approaches across various datasets. In terms of generalizability, we train our model on one dataset and tested it on multiple datasets. The standard deviation of AUROC scores for different test datasets is calculated to measure the variability of performance across datasets. Our model exhibits superior generalization across datasets from diverse sources. Conclusions: Our model shows promising results for the generalizable diagnosis of chest X-rays. The impacts of using the attention mechanism and the self-consistency loss in our method are evident from the results. In the future, we plan to incorporate Explainable AI techniques to provide explanations for model decisions. Additionally, we aim to design data augmentation techniques to reduce class imbalance in our model.

Chest X-ray Image Classification: A Causal Perspective

Instrumental Variable Learning for Chest X-ray Classification

Deep Learning-based Multi-Class COVID-19 Classification with X-ray Images

Improving Classification Model Performance on Chest X-Rays through Lung Segmentation

Artificially Generated Visual Scanpath Improves Multilabel Thoracic Disease Classification in Chest X-Ray Images

DeepCXray: Automatically Diagnosing Diseases on Chest X-Rays Using Deep Neural Networks

COVID-CBR: A Deep Learning Architecture Featuring Case-Based Reasoning for Classification of COVID-19 from Chest X-Ray Images

Unlearning Spurious Correlations in Chest X-ray Classification

COVID-19 Classification from Chest X-Ray Images: A Framework of Deep Explainable Artificial Intelligence

Multi-Label Chest X-Ray Classification via Deep Learning

ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases

Diagnose Chest Pathology In X-Ray Images By Learning Multi-Attention Convolutional Neural Network

Multi-label Chest X-ray Image Classification Via Category-Wise Residual Attention Learning

Expert Uncertainty and Severity Aware Chest X-Ray Classification by Multi-Relationship Graph Learning

Improving Computer-Aided Thoracic Disease Diagnosis through Comparative Analysis Using Chest X-ray Images Taken at Different Times

Robust and Interpretable COVID-19 Diagnosis on Chest X-ray Images using Adversarial Training

Deep Mining External Imperfect Data for Chest X-Ray Disease Screening

COVID-Classifier: an automated machine learning model to assist in the diagnosis of COVID-19 infection in chest X-ray images

Deep Hiearchical Multi-Label Classification Applied to Chest X-Ray Abnormality Taxonomies

Generalizable diagnosis of chest radiographs through attention-guided decomposition of images utilizing self-consistency loss

Classification of COVID-19 from CXR Images in a 15-class Scenario: an Attempt to Avoid Bias in the System