Abnormal Chest X-ray Identification With Generative Adversarial One-Class Classifier

03/05/2019 ∙ by Yuxing Tang, et al. ∙ 0

Being one of the most common diagnostic imaging tests, chest radiography requires timely reporting of potential findings in the images. In this paper, we propose an end-to-end architecture for abnormal chest X-ray identification using generative adversarial one-class learning. Unlike previous approaches, our method takes only normal chest X-ray images as input. The architecture is composed of three deep neural networks, each of which learned by competing while collaborating among them to model the underlying content structure of the normal chest X-rays. Given a chest X-ray image in the testing phase, if it is normal, the learned architecture can well model and reconstruct the content; if it is abnormal, since the content is unseen in the training phase, the model would perform poorly in its reconstruction. It thus enables distinguishing abnormal chest X-rays from normal ones. Quantitative and qualitative experiments demonstrate the effectiveness and efficiency of our approach, where an AUC of 0.841 is achieved on the challenging NIH Chest X-ray dataset in a one-class learning setting, with the potential in reducing the workload for radiologists.

READ FULL TEXT VIEW PDF
POST COMMENT

Comments

There are no comments yet.

Authors

page 2

page 4

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

The chest radiograph (chest X-ray, or CXR) is the most commonly requested radiological examination owing to its effectiveness in the characterization and detection of cardiothoracic and pulmonary abnormalities. It is also widely used in lung cancer prevention and screening. Timely radiologist reporting of every image is desired, but not always possible due to heavy workload. Consequently, an automatic system of CXR abnormality classification would be advantageous, allowing reporting works focusing more on pathology analysis of abnormal CXRs.

Recently, deep learning based approaches have been proposed as the solution to automatic classification and detection of abnormalities in CXRs with promising results 

[1, 2, 3]

. A large set of labeled training data are required to build discriminative convolutional neural networks (CNNs) for such a purpose. However, it is not always possible to include or annotate all kinds of abnormalities for large scale training, for the reason that some forms of anomaly are very rare, while on the other hand, normal CXRs are much easier to obtain.

Inspired by one-class classification [4, 5], which tries to classify data of a specific category among all data by learning from a training set containing only the data of that unique category, in this paper, we present a method that can automatically identify abnormal CXRs by learning only from normal ones: capturing special characteristics of normal CXR collection and figuring out how the unseen abnormal collection differentiates from them. More specifically, we propose an end-to-end generative adversarial one-class learning approach, for normal versus abnormal CXR classification, by training solely from normal CXRs. The proposed architecture, similar to generative adversarial networks (GANs) [6]

, is composed of three main modules: a U-Net autoencoder, a CNN discriminator and an encoder, which compete to learn while collaborating with each other for the target task. The adversarially trained generative model is capable of reconstructing the normal CXRs while performing poorly on reconstructing the abnormal ones, since only the normal CXRs are involved in training and those with various anomalies are unseen by the model. Such reconstruction differentiation enables the proposed model to identify abnormal CXRs.

Previous work  [7, 8]

adopted GANs to synthesize CXRs in order to augment the training set for classifying abnormalities with less training samples. Although we are reconstructing CXRs, we are not augmenting the training set and we only use CXR samples from a single class to train a one-class classifier. In our work, the reconstructed CXRs are considered as enhancing the inlier (normal) samples and distorting the outlier (abnormal) samples.

Figure 1: Framework of the proposed deep adversarial one-class learning model for abnormal chest X-ray identification.

2 Proposed Method

2.1 Problem Formulation

In our one-class learning scenario, given a training set , where is the number of samples in , is a normal CXR image with label (, ). The test set contains CXRs, each of which is labeled as either normal () or abnormal (). In the training stage, the goal is to learn a function from , which models the data distribution of normal CXRs, meanwhile, minimizes the anomaly score so that . In the inference phase, is expected to output smaller anomaly scores given normal X-ray images and larger scores given abnormal X-rays, such that , thus differentiating between normal and abnormal CXRs.

2.2 The Proposed Architecture

The proposed adversarial one-class learning framework is inspired by the generative adversarial networks (GANs) [6]. GAN is formulated as a two-player game, where the generator

takes a random noise vector in a latent space as input and produces a sample in the data space while the discriminator

identifies if a certain sample comes from the true data distribution or . The training procedure is to solve a minimax problem which alternates between training and , such that is optimized to generate samples that are not distinguishable by .

Network Architecture: Three essential modules, namely, a U-Net [9] like autoencoder (generator), a convolutional neural network (discriminator) and an encoder network, together constitute the generative adversarial one-class learning architecture (See Fig. 1). The U-Net like autoencoder (denoted as ) first maps an input CXR image with Gaussian noise into a lower-dimensional latent space using a fully convolutional network ( encoder ), which is then inversely mapped back using a deconvolutional network (decoder ) to generate the reconstructed image . The U-Net like encoder-decoder with skip connections is adopted to preserve high-resolution features through concatenation in the up-sampling (deconvolution) process, and a CNN discriminator (denoted as ) is looped for adversarial training to produce better and more realistic reconstruction. The first two modules function as a conditional GAN, which are conditioned on the real input image . A second encoder

is padded after the autoencoder, which further encodes the generated fake image into another latent space

, in order to force the consistency between two latent vectors and and corresponding intermediate feature maps from the two encoders.

The intuition behind the proposed framework is that it is able to reconstruct the normal CXRs while performing poorly on reconstructing the abnormal ones, since only the normal chest X-rays are used for training, and the abnormal CXR image contents are unseen. Such differentiation of reconstruction behaviors enables distinguishing abnormal CXRs from normal ones.

Loss Functions:

The objective of the proposed architecture is to jointly optimize the three modules in an end-to-end manner. To this end, we design four different loss functions:

The image reconstruction loss of the autoencoder is formulated as the mean absolute error (-norm) of a real input CXR image and its reconstructed image , to measure the similarity between the image pairs:

(1)

The adversarial learning loss of the conditional GAN is modeled as the binary cross-entropy loss for classification of real CXR and generated fake CXR :

(2)

The encoding consistency loss models the consistency between the two latent space and , which is formulated as the mean square error (-norm):

(3)

The feature map consistency loss measures the overall similarity between intermediate feature maps of the two encoders:

(4)

where is the feature map of the layer of the encoder.

The final objective function is:

(5)

where all s

are trade-off hyperparameters that control the relative importance of each of the four terms.

2.3 Inference

In the testing phase, a CXR is passed through the framework and an anomaly score is calculated by:

(6)

where indicates the likelihood that a generated CXR from looks realistic, and is normalized to on all testing samples for binary classification evaluation. Ideally, the anomaly scores of any abnormal samples should be larger than the scores from normal samples .

3 Experimental Results

3.1 Dataset and Implementation Details

We evaluated the proposed framework for normal versus abnormal CXR classification on the NIH Clinical Center Chest X-ray dataset111https://nihcc.app.box.com/v/ChestXray-NIHCC [1] , which contains 112,120 frontal-view CXR images of 30,805 unique patients. Cardiothoracic and pulmonary abnormalities include cardiomegaly, lung infiltrate, mass, nodule, pneumonia, pneumothorax, pulmonary atelectasis, consolidation, edema, emphysema, fibrosis, hernia, pleural effusion and thickening. We performed two experiments on this dataset: In the first experiment, we used 4,479 normal (without any abnormal pulmonary or cardiac findings) and no (zero) abnormal CXRs for training, 849 normal and 857 abnormal CXRs for validation, 677 normal and 667 abnormal CXRs for testing. In this experiment, the abnormal CXRs contain at least one of the above 14 findings. In the second experiment, we classified the CXRs with lung opacities (visual signal for pneumonia) from normal CXRs222https://www.kaggle.com/c/rsna-pneumonia-detection-challenge. 6,000 normal CXRs were used for one-class training, 1,025 normal CXRs and 1,025 CXRs with lung opacities were used for validation, 1,000 normal and 1,000 CXRs with lung opacities for testing. The training/validation/testing subsets were split by patient ID so there was no patient overlap among these three subsets.

The framework was implemented using the PyTorch library. The U-Net autoencoder consists of a 5-layer CNN encoder and a 5-layer deconvolutional decoder with skip connections (both have batch normalization and leaky ReLU after each layer except for the first layer). The second encoder and discriminator has the similar structure as the encoder in the autoencoder.

kernels were used in both down and up sampling, and the latent vector size was 100. The images were resized to

pixels, with a batch size of 64. The network was initialized with standard normal distribution and optimized using Adam gradient descent optimizer (momentums

= 0.5, = 0.999) with an initial learning rate of . , and

were empirically set to 20, 4 and 8 respectively. The training converged in 15 epochs, taking only 5-7 minutes on an NVIDIA TITAN Xp GPU. The inference time for each CXR was 0.48 ms on average.

3.2 Quantitative and Qualitative Results

We first quantitatively evaluate the classification performance using the area under the receiver operating characteristic (AUC) score. We conduct ablation studies to the proposed framework. The baseline method is the U-Net autoencoder truncated from the proposed model. Then we add a second encoder and decoder to respectively, to compare their performance with the whole network (). The training and evaluation process is repeated five times to reduce randomness. The normal and abnormal (or lung opacities) CXR classification results in terms of AUCs are shown in Table 1. The proposed model achieves an average AUC of 0.841 on the testing set for normal and abnormal CXRs with 14 major findings, and an average AUC of 0.802 for normal versus abnormal CXRs with lung opacities, which largely outperforms the baseline U-Net model without adversarial learning. Each of the proposed modules contributes to the final model improvement. The overall result is encouraging given the fact that the dataset contains difficult cases (e.g., mild abnormalities) and only normal CXRs are used in the training stage.

Dataset/ Normal vs. Normal vs.
Method Abnormal Lung opcacities
0.6270.036 0.5920.021
0.6870.032 0.6590.025
0.7370.028 0.7200.034
0.8410.030 0.8020.033
Table 1: Comparison of classification performance in terms of AUC on the test sets from two experiments. (: U-Net autoencoder, : second encoder, : discriminator.)
Figure 2: Examples of ground truth CXRs and their corresponding reconstructed images in training and testing stages.

We show qualitative examples of real CXRs and reconstructed images by our framework in Figure 2. As can be seen from the figure, our model is able to generate normal CXRs of high quality in the training stage (see left column of Figure 2). In the testing stage, the proposed model reconstructs normal CXR images (middle column) with much better quality than abnormal CXRs (right column), where not only the abnormal image contents are blurry and messy, but also the geometrical structures of the chest regions are distorted. This confirms how our one-class classifier can indeed differentiate the normal CXRs and abnormal ones.

4 Conclusion

In this paper, we present an end-to-end trained generative adversarial one-class classifier for abnormal chest X-ray detection, by learning only from normal CXRs. The proposed method is able to reconstruct the normal CXRs while performing poorly on reconstructing the abnormal ones, since the abnormal CXR image contents are unseen during training. Our method is fast and effective, with less manual annotation effort needed. Quantitative and qualitative experimental results demonstrate encouraging performance, showing a potential of reducing workload for radiologists. The proposed method could possibly be extended and applied to other image modalities in future work.

5 Acknowledgments

This research was supported by the Intramural Research Program of the National Institutes of Health Clinical Center and by the Ping An Technology Co., Ltd. through a Cooperative Research and Development Agreement. The authors thank NVIDIA for GPU donation.

References

  • [1] X. Wang, Y. Peng, L. Lu, Z. Lu, M. Bagheri, and R. M. Summers, “Chestx-ray8: Hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases,” in CVPR, 2017.
  • [2] E.J. Yates, L.C. Yates, and H. Harvey,

    Machine learning “red dot”: open-source, cloud, deep convolutional neural networks in chest radiograph binary normality classification,”

    Clinical Radiology, vol. 73, no. 9, pp. 827 – 831, 2018.
  • [3] Y. Tang, X. Wang, A. P. Harrison, L. Lu, J. Xiao, and R. M. Summers, “Attention-guided curriculum learning for weakly supervised classification and localization of thoracic diseases on chest radiographs,” in Machine Learning in Medical Imaging, 2018.
  • [4] M. M. Moya and D. R. Hush, “Network constraints and multi-objective optimization for one-class classification,” Neural Networks, vol. 9, no. 3, pp. 463 – 474, 1996.
  • [5] M. Sabokrou, M. Khalooei, M. Fathy, and E. Adeli,

    “Adversarially learned one-class classifier for novelty detection,”

    in CVPR, 2018.
  • [6] I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde-Farley, S. Ozair, A. Courville, and Y. Bengio, “Generative adversarial nets,” in NIPS, 2014.
  • [7] A. Madani, M. Moradi, A. Karargyris, and T. Syeda-Mahmood,

    Semi-supervised learning with generative adversarial networks for chest x-ray classification with ability of data domain adaptation,”

    in ISBI, 2018.
  • [8] H. Salehinejad et al., “Generalization of deep neural networks for chest pathology classification in x-rays using generative adversarial networks,” in ICASSP, 2018.
  • [9] O. Ronneberger, P.Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” in MICCAI, 2015.