학술논문

Lung-GANs: Unsupervised Representation Learning for Lung Disease Classification Using Chest CT and X-Ray Images
Document Type
Periodical
Source
IEEE Transactions on Engineering Management IEEE Trans. Eng. Manage. Engineering Management, IEEE Transactions on. 70(8):2774-2786 Aug, 2023
Subject
Engineering Profession
Pulmonary diseases
X-ray imaging
Computed tomography
Generators
Feature extraction
COVID-19
Training
CT scan
generative adversarial networks
lung disease
pediatric pneumonia
pneumonia
tuberculosis
unsupervised representation learning
X-ray
Language
ISSN
0018-9391
1558-0040
Abstract
Lung diseases are a tremendous challenge to the health and life of people globally, accounting for 5 out of 30 most common causes of death. Early diagnosis is crucial to help in faster recovery and improve long-term survival rates. Deep learning techniques offer a great promise for automated, fast, and reliable detection of lung diseases from medical images. Specifically, convolutional neural networks have accomplished encouraging results in disease detection. In spite of that, the performance of such supervised models depends heavily on the availability of large labeled data, the collection of which is an expensive and tedious task, specially for a novel disease. Therefore, in this article, we propose a deep unsupervised framework to classify lung diseases from chest CT and X-ray images. Our framework introduces multiple-layer generative adversarial networks called Lung-GANs that learn interpretable representations of lung disease images using only unlabeled data. We use the lung features learned by the model to train a support vector machine and a stacking classifier. We demonstrate through experiments that the proposed method outperforms the current state-of-the-art unsupervised models in lung disease classification. Our model obtained an accuracy of 94%–99.5% on all the six large-scale publicly available lung disease datasets used in this study. Hence, the proposed framework will simplify lung disease detection by reducing the time for diagnosis and increasing the convenience of diagnostics.