LeDNet: Localization-enabled Deep Neural Network for Multi-Label Radiography Image Classification

Lalit Pant,Shubham Arora
2024-07-04
Abstract:Multi-label radiography image classification has long been a topic of interest in neural networks research. In this paper, we intend to classify such images using convolution neural networks with novel localization techniques. We will use the chest x-ray images to detect thoracic diseases for this purpose. For accurate diagnosis, it is crucial to train the network with good quality images. But many chest X-ray images have irrelevant external objects like distractions created by faulty scans, electronic devices scanned next to lung region, scans inadvertently capturing bodily air etc. To address these, we propose a combination of localization and deep learning algorithms called LeDNet to predict thoracic diseases with higher accuracy. We identify and extract the lung region masks from chest x-ray images through localization. These masks are superimposed on the original X-ray images to create the mask overlay images. DenseNet-121 classification models are then used for feature selection to retrieve features of the entire chest X-ray images and the localized mask overlay images. These features are then used to predict disease classification. Our experiments involve comparing classification results obtained with original CheXpert images and mask overlay images. The comparison is demonstrated through accuracy and loss curve analyses.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper aims to address the accuracy issues in multi-label radiographic image classification, particularly in the application of detecting thoracic diseases in chest X-ray images. Specifically, the paper proposes a convolutional neural network (CNN) method combined with localization techniques—LeDNet (Localization-enabled Deep Neural Network)—to improve the classification accuracy of thoracic diseases. The main issues include: 1. **Irrelevant Object Interference**: Many chest X-ray images contain irrelevant objects such as scanning errors, electronic devices, and gases from other parts of the body, which may lead to misdiagnosis or inaccurate prediction results. 2. **Data Quality Issues**: The quality of images in the original dataset varies, requiring preprocessing and localization to remove noise and thus improve the performance of the classification model. To address these issues, the authors propose the following methods: - Utilize localization techniques to extract lung region masks from chest X-ray images. - Overlay these masks onto the original images to create mask overlay images. - Use the DenseNet-121 model to perform feature selection and classification on both the entire chest X-ray images and the local mask images. - Evaluate the performance improvement of the model by comparing the classification results obtained using the original CheXpert images and the mask overlay images. The ultimate goal is to improve the accuracy and robustness of deep learning-based thoracic disease classification algorithms by enhancing the image preprocessing workflow.