Few-Shot Learning Approach on Tuberculosis Classification Based on Chest X-Ray Images

• URL: http://arxiv.org/abs/2409.11644v1
• Date: Wed, 18 Sep 2024 02:15:01 GMT
• Title: Few-Shot Learning Approach on Tuberculosis Classification Based on Chest X-Ray Images
• Authors: A. A. G. Yogi Pramana, Faiz Ihza Permana, Muhammad Fazil Maulana, Dzikri Rahadian Fudholi,
• Abstract summary: Class imbalance in TB chest X-ray datasets presents a challenge for accurate classification. We propose a few-shot learning approach using the Prototypical Network algorithm to address this issue. Experimental results demonstrate classification accuracies of 98.93% for ResNet-18, 98.60% for ResNet-50, and 33.33% for VGG16.
• Score: 0.0
• License: http://creativecommons.org/licenses/by-nc-nd/4.0/
• Abstract: Tuberculosis (TB) is caused by the bacterium Mycobacterium tuberculosis, primarily affecting the lungs. Early detection is crucial for improving treatment effectiveness and reducing transmission risk. Artificial intelligence (AI), particularly through image classification of chest X-rays, can assist in TB detection. However, class imbalance in TB chest X-ray datasets presents a challenge for accurate classification. In this paper, we propose a few-shot learning (FSL) approach using the Prototypical Network algorithm to address this issue. We compare the performance of ResNet-18, ResNet-50, and VGG16 in feature extraction from the TBX11K Chest X-ray dataset. Experimental results demonstrate classification accuracies of 98.93% for ResNet-18, 98.60% for ResNet-50, and 33.33% for VGG16. These findings indicate that the proposed method outperforms others in mitigating data imbalance, which is particularly beneficial for disease classification applications.

Related papers

• Revisiting Computer-Aided Tuberculosis Diagnosis [56.80999479735375]
  Tuberculosis (TB) is a major global health threat, causing millions of deaths annually. Computer-aided tuberculosis diagnosis (CTD) using deep learning has shown promise, but progress is hindered by limited training data. We establish a large-scale dataset, namely the Tuberculosis X-ray (TBX11K) dataset, which contains 11,200 chest X-ray (CXR) images with corresponding bounding box annotations for TB areas. This dataset enables the training of sophisticated detectors for high-quality CTD.
  arXiv  Detail & Related papers  (2023-07-06T08:27:48Z)
• Attention-based Saliency Maps Improve Interpretability of Pneumothorax Classification [52.77024349608834]
  To investigate chest radiograph (CXR) classification performance of vision transformers (ViT) and interpretability of attention-based saliency. ViTs were fine-tuned for lung disease classification using four public data sets: CheXpert, Chest X-Ray 14, MIMIC CXR, and VinBigData. ViTs had comparable CXR classification AUCs compared with state-of-the-art CNNs.
  arXiv  Detail & Related papers  (2023-03-03T12:05:41Z)
• Computer-aided Tuberculosis Diagnosis with Attribute Reasoning Assistance [58.01014026139231]
  We propose a new large-scale tuberculosis (TB) chest X-ray dataset (TBX-Att) We establish an attribute-assisted weakly-supervised framework to classify and localize TB by leveraging the attribute information. The proposed model is evaluated on the TBX-Att dataset and will serve as a solid baseline for future research.
  arXiv  Detail & Related papers  (2022-07-01T07:50:35Z)
• A novel framework based on deep learning and ANOVA feature selection method for diagnosis of COVID-19 cases from chest X-ray Images [0.0]
  COVID-19 was first identified in Wuhan and quickly spread worldwide. Most accessible method for COVID-19 identification is RT-PCR. Compared to RT-PCR, chest CT scans and chest X-ray images provide superior results. DenseNet169 was employed to extract features from X-ray images.
  arXiv  Detail & Related papers  (2021-09-30T16:10:31Z)
• Evaluation of Convolutional Neural Networks for COVID-19 Classification on Chest X-Rays [7.509537244787675]
  We propose the evaluation of convolutional neural networks to identify pneumonia due to COVID-19 in Digital Chest X-rays. The proposed methodology consists of a preprocessing step of the XR, data augmentation, and classification by the convolutional architectures DenseNet121, InResceptionNetV2, InceptionV3, MovileNetV2, ResNet50, and VGG16. The obtained results demonstrate that the VGG16 architecture presented a superior performance in the classification of XR.
  arXiv  Detail & Related papers  (2021-09-06T12:57:55Z)
• Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualization [0.0]
  Tuberculosis is a chronic lung disease that occurs due to bacterial infection and is one of the top 10 leading causes of death. We have detected TB reliably from the chest X-ray images using image pre-processing, data augmentation, image segmentation, and deep-learning classification techniques.
  arXiv  Detail & Related papers  (2020-07-29T15:11:34Z)
• An Efficient Mixture of Deep and Machine Learning Models for COVID-19 and Tuberculosis Detection Using X-Ray Images in Resource Limited Settings [2.408714894793063]
  Clinicians in the frontline need to assess quickly whether a patient with symptoms indeed has COVID-19 or not. The difficulty of this task is exacerbated in low resource settings that may not have access to biotechnology tests. We propose the extraction of deep features (DF) from chest X-ray images and their subsequent classification using machine learning methods.
  arXiv  Detail & Related papers  (2020-07-16T09:49:49Z)
• Predicting COVID-19 Pneumonia Severity on Chest X-ray with Deep Learning [57.00601760750389]
  We present a severity score prediction model for COVID-19 pneumonia for frontal chest X-ray images. Such a tool can gauge severity of COVID-19 lung infections that can be used for escalation or de-escalation of care.
  arXiv  Detail & Related papers  (2020-05-24T23:13:16Z)
• Adaptive Feature Selection Guided Deep Forest for COVID-19 Classification with Chest CT [49.09507792800059]
  We propose an Adaptive Feature Selection guided Deep Forest (AFS-DF) for COVID-19 classification based on chest CT images. We evaluate our proposed AFS-DF on COVID-19 dataset with 1495 patients of COVID-19 and 1027 patients of community acquired pneumonia (CAP)
  arXiv  Detail & Related papers  (2020-05-07T06:00:02Z)
• JCS: An Explainable COVID-19 Diagnosis System by Joint Classification and Segmentation [95.57532063232198]
  coronavirus disease 2019 (COVID-19) has caused a pandemic disease in over 200 countries. To control the infection, identifying and separating the infected people is the most crucial step. This paper develops a novel Joint Classification and (JCS) system to perform real-time and explainable COVID-19 chest CT diagnosis.
  arXiv  Detail & Related papers  (2020-04-15T12:30:40Z)
• Detection of Coronavirus (COVID-19) Associated Pneumonia based on Generative Adversarial Networks and a Fine-Tuned Deep Transfer Learning Model using Chest X-ray Dataset [4.664495510551646]
  This paper presents a pneumonia chest x-ray detection based on generative adversarial networks (GAN) with a fine-tuned deep transfer learning for a limited dataset. The dataset used in this research consists of 5863 X-ray images with two categories: Normal and Pneumonia.
  arXiv  Detail & Related papers  (2020-04-02T08:14:37Z)

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.