Skip to main content

TX-CNN: DETECTING TUBERCULOSIS IN CHEST X-RAY IMAGES USING CONVOLUTIONAL NEURAL NETWORK

In this paper, the authors propose a method to classify tuberculosis from chest X-ray images using Convolutional Neural Networks (CNN). They achieve a classification accuracy of 85.68%. They attribute the effectiveness of their approach to shuffle sampling with cross-validation while training the network.

Methodology

Convolutional Neural Network

This has been the ultimate tool for researchers and engineers for computer vision tasks. It has been widely used for many general purpose image and video related tasks. There are many great resources to learn about them. I will link a few of them at the end of this post.

In this paper, the authors study the famous AlexNet and GoogLeNet architectures in classifying tuberculosis images. A CNN model usually consists of convolutional layers, pooling layers and fully connected layers. Each layer is connected to the previous layers via kernels or filters. A CNN model learns parameters of the kernel to represent global and local features of the image.

Transfer Learning

This is a technique used to reuse features learned by a network on one domain/dataset in another domain/dataset. In domains such as the medical one, the number of labeled data available is generally less. So it is common in deep learning to train a network on a well known large dataset (like ImageNet) and then fine-tune the same model by retraining it on the medical data. This allows learning of basic features from the randomly initialized weights of the network during training on the large dataset and then later by fixing those features, learning medical related features during the fine tuning phase.

Since the chest X-ray images are not abundant to train large complex models, the authors chose this strategy to pretrain their networks on the ImageNet dataset and then fine-tune this model on the TB dataset. They also use shuffle-sampling and cross-validation while training on the chest X-ray images.

Shuffle Sampling

If you read this paper, you will see that the authors attribute the boost in accuracy for their models to this pre-training technique. Shuffle sampling is a data augmentation technique used to balance the categories of datasets. For very unbalanced datasets (some categories have much more data than others), it is very hard to learn a general classifier for all the categories. According to experiments in this paper, this technique increased training time by 2 hours for AlexNet and accuracy from 53.02% to 85.68%.
The steps followed for this technique are as follows
  1. Index every image in a category and take N as the number of images in the largest category.
  2. For each category, generate N unique numbers.
  3. For each category, calculate mod value of the N unique numbers with Nc (number of images in respective category)
  4. This mod value will give N numbers each of which is an index of an image in the respective category.
  5. Substitute this index with the corresponding image and the dataset expands to a size of N x Number of categories.
Please refer the image borrowed from the paper.

Learning Links



Comments

Post a Comment

Popular Posts

ES3Net: Accurate and Efficient Edge-based Self-Supervised Stereo Matching Network

Efficient and accurate depth estimation plays an indispensable role in many real-world applications, such as autonomous vehicles, 3D reconstruction, and drone navigation. Despite the precision of stereo matching, its computational intensity can pose significant challenges for edge deployment. Moreover, the struggle of acquiring ground-truth depths for training stereo-matching networks further amplifies these challenges. Enter ES3Net, the Edge-based Self-Supervised Stereo matching Network, a solution designed to mitigate these obstacles. The Challenges of Depth Estimation When it comes to applications like autonomous driving or drone navigation, the importance of accurate depth estimation is hard to overstate. It provides a foundational understanding of the 3D world, allowing for intelligent decision-making and navigation. Traditionally, stereo matching has provided greater accuracy than monocular depth estimation due to the availability of a reference image. However, it also bri...
Post a Comment
Read more

Ocean: Object-aware Anchor-free Tracking

The paper titled " Ocean: Object Aware Anchor Free Tracking " presents a novel approach to visual object tracking that is poised to outperform existing anchor-based approaches. The authors propose a unique anchor-free framework named Ocean, designed to address certain challenges in the current field of visual tracking. Introduction Visual object tracking is a crucial part of computer vision technology. The widely utilized anchor-based trackers have their limitations, which this paper attempts to address. The authors present the innovative Ocean framework, designed to transform the visual tracking field by improving adaptability and performance. The Problem with Anchor-Based Trackers Despite their wide usage, anchor-based trackers suffer from some notable drawbacks. They struggle with tracking objects experiencing drastic scale changes or those having high aspect ratios. The anchors, with their fixed scale and fixed ratios, can limit the flexibility of the trackers, making the...
Post a Comment
Read more

CLIP: Learning Transferable Visual Models From Natural Language Supervision

CLIP (Contrastive Language-Image Pre-training) is a new approach to learning visual representations proposed by researchers at OpenAI in 2021. Unlike traditional computer vision models which are trained on large labeled image datasets, CLIP learns directly from natural language supervision. The Core Idea The key insight behind CLIP is that we can connect images and text captions without generating the captions. By training the model to predict which caption goes with an image, it learns a rich visual representation of the world. As illustrated above, CLIP consists of two encoders - an image encoder and a text encoder. The image encoder takes in an image and outputs a visual representation vector. The text encoder takes in a caption and outputs a text representation vector. During training, these representations are optimized to be closer for matching image-text pairs, and farther apart for non-matching pairs. This is known as a contrastive loss objective. Benefits of CLIP There are sev...
Post a Comment
Read more