Classification of chest diseases from x-ray images on the chexpert dataset
This work proposes a method to classify tuberculosis (TB) disease in a chest radiograph using convolutional neural network (CNN) algorithms. The main contribution of this work is to detect and classify TB disease in addition to the other five different diseases. This is achieved by using a transfer...
Saved in:
Main Authors: | , , |
---|---|
Format: | Conference or Workshop Item |
Published: |
2021
|
Subjects: | |
Online Access: | http://eprints.utm.my/id/eprint/95141/ http://dx.doi.org/10.1007/978-981-16-0749-3_64 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
id |
my.utm.95141 |
---|---|
record_format |
eprints |
spelling |
my.utm.951412022-04-29T22:02:29Z http://eprints.utm.my/id/eprint/95141/ Classification of chest diseases from x-ray images on the chexpert dataset Saleem, H. N. Sheikh, U. U. Khalid, S. A. TK Electrical engineering. Electronics Nuclear engineering This work proposes a method to classify tuberculosis (TB) disease in a chest radiograph using convolutional neural network (CNN) algorithms. The main contribution of this work is to detect and classify TB disease in addition to the other five different diseases. This is achieved by using a transfer learning technique that utilizes a pre-trained CNN network to classify the TB disease. A comprehensive verification using TensorFlow is carried out to train and validate the proposed technique. This work aimed to use different pre-trained models on the CheXpert dataset and compare the area under the curve (AUC) between the CNN models. From the simulations, it was found that it is possible to classify the TB disease in addition to the other five diseases without having a degradation in the accuracy. The results confirm that transfer learning technique is superior to other methods, which exhibits less time for training and validating the datasets, and has good performance. This work achieved excellent performance in classifying three different diseases (atelec-tasis, edema, and tuberculosis) with AUC of 0.912, 0.945, and 0.954, respectively. Also, this work achieved second-best performance for classifying pleural effusion and consolidation diseases with AUC of 0.928 and 0.917, respectively. The method proposed in this work can be used for classification of diseases in chest radiograph as an early diagnosis tool in a clinical environment. 2021 Conference or Workshop Item PeerReviewed Saleem, H. N. and Sheikh, U. U. and Khalid, S. A. (2021) Classification of chest diseases from x-ray images on the chexpert dataset. In: 2nd International Conference on Electrical and Electronics Engineering, ICEEE 2021, 2-3 Jan 2021, Virtual, Online. http://dx.doi.org/10.1007/978-981-16-0749-3_64 |
institution |
Universiti Teknologi Malaysia |
building |
UTM Library |
collection |
Institutional Repository |
continent |
Asia |
country |
Malaysia |
content_provider |
Universiti Teknologi Malaysia |
content_source |
UTM Institutional Repository |
url_provider |
http://eprints.utm.my/ |
topic |
TK Electrical engineering. Electronics Nuclear engineering |
spellingShingle |
TK Electrical engineering. Electronics Nuclear engineering Saleem, H. N. Sheikh, U. U. Khalid, S. A. Classification of chest diseases from x-ray images on the chexpert dataset |
description |
This work proposes a method to classify tuberculosis (TB) disease in a chest radiograph using convolutional neural network (CNN) algorithms. The main contribution of this work is to detect and classify TB disease in addition to the other five different diseases. This is achieved by using a transfer learning technique that utilizes a pre-trained CNN network to classify the TB disease. A comprehensive verification using TensorFlow is carried out to train and validate the proposed technique. This work aimed to use different pre-trained models on the CheXpert dataset and compare the area under the curve (AUC) between the CNN models. From the simulations, it was found that it is possible to classify the TB disease in addition to the other five diseases without having a degradation in the accuracy. The results confirm that transfer learning technique is superior to other methods, which exhibits less time for training and validating the datasets, and has good performance. This work achieved excellent performance in classifying three different diseases (atelec-tasis, edema, and tuberculosis) with AUC of 0.912, 0.945, and 0.954, respectively. Also, this work achieved second-best performance for classifying pleural effusion and consolidation diseases with AUC of 0.928 and 0.917, respectively. The method proposed in this work can be used for classification of diseases in chest radiograph as an early diagnosis tool in a clinical environment. |
format |
Conference or Workshop Item |
author |
Saleem, H. N. Sheikh, U. U. Khalid, S. A. |
author_facet |
Saleem, H. N. Sheikh, U. U. Khalid, S. A. |
author_sort |
Saleem, H. N. |
title |
Classification of chest diseases from x-ray images on the chexpert dataset |
title_short |
Classification of chest diseases from x-ray images on the chexpert dataset |
title_full |
Classification of chest diseases from x-ray images on the chexpert dataset |
title_fullStr |
Classification of chest diseases from x-ray images on the chexpert dataset |
title_full_unstemmed |
Classification of chest diseases from x-ray images on the chexpert dataset |
title_sort |
classification of chest diseases from x-ray images on the chexpert dataset |
publishDate |
2021 |
url |
http://eprints.utm.my/id/eprint/95141/ http://dx.doi.org/10.1007/978-981-16-0749-3_64 |
_version_ |
1732945437168500736 |
score |
13.154949 |