Deep Pneumonia: Attention-Based Contrastive Learning for
Class-Imbalanced Pneumonia Lesion Recognition in Chest X-rays
- URL: http://arxiv.org/abs/2207.11393v1
- Date: Sat, 23 Jul 2022 02:28:37 GMT
- Title: Deep Pneumonia: Attention-Based Contrastive Learning for
Class-Imbalanced Pneumonia Lesion Recognition in Chest X-rays
- Authors: Xinxu Wei, Haohan Bai, Xianshi Zhang and Yongjie Li
- Abstract summary: We propose a deep learning framework named Attention-Based Contrastive Learning for Class-Imbalanced X-Ray Pneumonia Lesion Recognition.
Our proposed framework can be used as a reliable computer-aided pneumonia diagnosis system to assist doctors to better diagnose pneumonia cases accurately.
- Score: 11.229472535033558
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: Computer-aided X-ray pneumonia lesion recognition is important for accurate
diagnosis of pneumonia. With the emergence of deep learning, the identification
accuracy of pneumonia has been greatly improved, but there are still some
challenges due to the fuzzy appearance of chest X-rays. In this paper, we
propose a deep learning framework named Attention-Based Contrastive Learning
for Class-Imbalanced X-Ray Pneumonia Lesion Recognition (denoted as Deep
Pneumonia). We adopt self-supervised contrastive learning strategy to pre-train
the model without using extra pneumonia data for fully mining the limited
available dataset. In order to leverage the location information of the lesion
area that the doctor has painstakingly marked, we propose mask-guided hard
attention strategy and feature learning with contrastive regulation strategy
which are applied on the attention map and the extracted features respectively
to guide the model to focus more attention on the lesion area where contains
more discriminative features for improving the recognition performance. In
addition, we adopt Class-Balanced Loss instead of traditional Cross-Entropy as
the loss function of classification to tackle the problem of serious class
imbalance between different classes of pneumonia in the dataset. The
experimental results show that our proposed framework can be used as a reliable
computer-aided pneumonia diagnosis system to assist doctors to better diagnose
pneumonia cases accurately.
Related papers
- FA-Net: A Fuzzy Attention-aided Deep Neural Network for Pneumonia Detection in Chest X-Rays [28.319405767795047]
Pneumonia is a respiratory infection caused by bacteria, fungi, or viruses.
Early diagnosis is crucial to ensure effective treatment and increase survival rates.
We have developed a computer-aided diagnosis system for automatic pneumonia detection using chest X-ray images.
arXiv Detail & Related papers (2024-06-21T13:08:40Z) - Expert Uncertainty and Severity Aware Chest X-Ray Classification by
Multi-Relationship Graph Learning [48.29204631769816]
We re-extract disease labels from CXR reports to make them more realistic by considering disease severity and uncertainty in classification.
Our experimental results show that models considering disease severity and uncertainty outperform previous state-of-the-art methods.
arXiv Detail & Related papers (2023-09-06T19:19:41Z) - Prediction of Pneumonia and COVID-19 Using Deep Neural Networks [0.0]
We propose machine-learning techniques for predicting Pneumonia from chest X-ray images.
DenseNet121 outperforms other models, achieving an accuracy rate of 99.58%.
arXiv Detail & Related papers (2023-08-20T21:26:37Z) - Deep Reinforcement Learning Framework for Thoracic Diseases
Classification via Prior Knowledge Guidance [49.87607548975686]
The scarcity of labeled data for related diseases poses a huge challenge to an accurate diagnosis.
We propose a novel deep reinforcement learning framework, which introduces prior knowledge to direct the learning of diagnostic agents.
Our approach's performance was demonstrated using the well-known NIHX-ray 14 and CheXpert datasets.
arXiv Detail & Related papers (2023-06-02T01:46:31Z) - Pneumonia Detection in Chest X-Ray Images : Handling Class Imbalance [0.0]
People all over the globe are affected by pneumonia but deaths due to it are highest in Sub-Saharan Asia and South Asia.
In recent years, the overall incidence and mortality rate of pneumonia regardless of the utilization of effective vaccines and compelling antibiotics has escalated.
The widespread prevalence of pneumonia has caused the research community to come up with a framework that helps detect, diagnose and analyze diseases accurately and promptly.
arXiv Detail & Related papers (2023-01-20T09:17:39Z) - Improving Pneumonia Localization via Cross-Attention on Medical Images
and Reports [9.034599866957945]
We propose a novel weakly-supervised attention-driven deep learning model that leverages encoded information in medical reports during training to facilitate better localization.
Our model also performs classification of attributes that are associated to pneumonia and extracted from medical reports for supervision.
In this paper, we explore and analyze the model using chest X-ray datasets and demonstrate qualitatively and quantitatively that the introduction of textual information improves pneumonia localization.
arXiv Detail & Related papers (2021-10-06T22:47:48Z) - CoRSAI: A System for Robust Interpretation of CT Scans of COVID-19
Patients Using Deep Learning [133.87426554801252]
We adopted an approach based on using an ensemble of deep convolutionalneural networks for segmentation of lung CT scans.
Using our models we are able to segment the lesions, evaluatepatients dynamics, estimate relative volume of lungs affected by lesions and evaluate the lung damage stage.
arXiv Detail & Related papers (2021-05-25T12:06:55Z) - Contralaterally Enhanced Networks for Thoracic Disease Detection [120.60868136876599]
There exist many similar structures in the left and right parts of the chest, such as ribs, lung fields and bronchial tubes.
This kind of similarities can be used to identify diseases in chest X-rays, according to the experience of broad-certificated radiologists.
We propose a deep end-to-end module to exploit the contralateral context information for enhancing feature representations of disease proposals.
arXiv Detail & Related papers (2020-10-09T10:15:26Z) - Deep Learning for Automatic Pneumonia Detection [72.55423549641714]
Pneumonia is the leading cause of death among young children and one of the top mortality causes worldwide.
Computer-aided diagnosis systems showed the potential for improving diagnostic accuracy.
We develop the computational approach for pneumonia regions detection based on single-shot detectors, squeeze-and-excitation deep convolution neural networks, augmentations and multi-task learning.
arXiv Detail & Related papers (2020-05-28T10:54:34Z) - Viral Pneumonia Screening on Chest X-ray Images Using Confidence-Aware
Anomaly Detection [86.81773672627406]
Clusters of viral pneumonia 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.
Viral pneumonia often have diverse causes and exhibit notably different visual appearances on X-ray images.
arXiv Detail & Related papers (2020-03-27T11:32:18Z)
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.