Adversarial Sparse-View CBCT Artifact Reduction

TL;DR

This paper introduces an adversarial deep learning approach with a novel discriminator architecture to effectively reduce streak artifacts in sparse-view CBCT images, improving image quality while preserving anatomical details.

Contribution

It presents a new adversarial training framework with a feature pyramid network discriminator and focus map for artifact reduction in sparse-view CBCT, outperforming existing methods.

Findings

Significant artifact correction in clinical CBCT images reconstructed with 1/3 projections.

Outperforms baseline methods quantitatively and qualitatively.

Produces more perceptually realistic images while maintaining anatomical accuracy.

Abstract

We present an effective post-processing method to reduce the artifacts from sparsely reconstructed cone-beam CT (CBCT) images. The proposed method is based on the state-of-the-art, image-to-image generative models with a perceptual loss as regulation. Unlike the traditional CT artifact-reduction approaches, our method is trained in an adversarial fashion that yields more perceptually realistic outputs while preserving the anatomical structures. To address the streak artifacts that are inherently local and appear across various scales, we further propose a novel discriminator architecture based on feature pyramid networks and a differentially modulated focus map to induce the adversarial training. Our experimental results show that the proposed method can greatly correct the cone-beam artifacts from clinical CBCT images reconstructed using 1/3 projections, and outperforms strong baseline methods both quantitatively and qualitatively.

Tables

Table 1. Table 1: Quantitative sparse-view CBCT artifact reduction results of different models.

		$x_s$		Baseline			Ours
				MSE	Perc.		Focus	FPN	FPN+Focus
SSIM		0.839		0.849	0.858		0.879	0.871	0.884
PSNR (dB)		34.07		34.24	35.39		36.26	36.38	36.14
RMSE (${\mathrm{𝟏𝟎}}^{-\mathrm{𝟐}}$)		1.98		1.96	1.70		1.54	1.52	1.56

Equations

$$\underset{D}{\min }{\mathcal{L}}_A(D;G,\Lambda )$$

$$\underset{G}{\min }{\mathcal{L}}_A(G;D,\Lambda )$$

$${\mathcal{L}}_P=\frac{1}{N_i}\Vert {\varphi }^{(i)}(x_d)-{\varphi }^{(i)}(G(x_s)){\Vert }_1.$$

$${\lambda }_{m,n}=\frac{1}{Z_j}\Vert {\varphi }_{m,n}^{(j)}(x_d)-{\varphi }_{m,n}^{(j)}(G(x_s))\Vert ,$$

$$Z_j=\frac{1}{N_j}\sum _{m,n}\Vert {\varphi }_{m,n}^{(j)}(x_d)-{\varphi }_{m,n}^{(j)}(G(x_s))\Vert .$$

Full text

1 Introduction

Cone-beam computed tomography (CBCT) is a variant type of computed tomography (CT). Compared with conventional CT, CBCT usually has shorter examination time, resulting in fewer motion artifacts and better X-ray tube efficiency. One way to further shorten the acquisition time and enhance the healthcare experience is to take fewer X-ray measurements during each CBCT scan. However, due to the “cone-beam” projection geometry, CBCT images typically contain more pronounced streak artifacts than CT images and this is even worse when fewer X-ray projections are used during the CBCT reconstruction [1].

A number of approaches have been proposed to address the artifacts [13, 8] that are commonly encountered in CBCT images. However, to our best knowledge, no scheme has been proposed to correct the cone-beam artifacts introduced by sparse-view CBCT reconstruction in a post-processing step. Instead of reducing artifacts from the CBCT images directly, many other systems [11, 14] propose to introduce better sparse-view reconstruction methods that yield less artifacts. Although encouraging improvements have been made, the image quality from the current solutions are still not satisfactory when only a small number of views are used. This work attempts to fill this gap by refining the sparsely reconstructed CBCT images through a novel cone-beam artifact reduction method.

In relation to this study, there are many works that leverage deep neural networks (DNNs) for low-dose CT (LDCT) denoising. [2] used a residual encoder-decoder architecture to reduce the noise from LDCT images, and achieved superior performance over traditional approaches. More recently, [10] introduced generative adversarial networks (GANs) [3] into their architecture to obtain more realistic outputs, and this work was further improved by [12] where a combination of perceptual loss [5] and adversarial loss was used.

Similarly, this work also proposes to use DNNs for sparse-view CBCT artifact reduction. We train an image-to-image generative model with perceptual loss to obtain outputs that are perceptually close to the dense-view CBCT images. To address the artifacts at various levels, we further contribute to the literature with a novel discriminator architecture based on feature pyramid networks (FPN) [6] and a differentially modulated focus map so that the adversarial training is biased to the artifacts at multiple scales. The proposed approach is evaluated on clinical CBCT images. Experimental results demonstrate that our method outperforms strong baseline methods both qualitatively and quantitatively.

2 Methods

Let $x_{s}$ be a sparse-view CBCT image, which is reconstructed from a sparse set or low number of projections (or views), and $x_{d}$ be its dense-view counterpart, which is reconstructed from a dense set or a high number of projections (or views). The proposed method is formulated under an image-to-image generative model as illustrated in Fig. 1 where we train a generator that transforms $x_{s}$ to an ideally artifact-free image that looks like $x_{d}$. The discriminator is used for the adversarial training, and the perceptual network is included for additional perceptual and structural regularization. We use LSGAN [7] against a regular GAN to achieve more stable adversarial learning. The adversarial objective functions for the proposed model can be written as

[TABLE]

where $\Lambda$ is a focus map detailed in Sec. 2.2. Here, we apply a PatchGAN-like [4] design to the discriminator so that the realness is patch based and the output is a score map. The generator $G$ and discriminator $D$ are trained in an adversarial fashion. $D$ distinguishes between $x_{d}$ and the generated CBCT image $G(x_{s})$ (Eq. 1), while $G$ generates CBCT image samples as “real” as possible so that $D$ cannot tell if they are dense-view CBCT images or generated by $G$ (Eq. 2).

Training with the adversarial loss, alone, usually introduces additional artifacts, and previous works often use MSE loss to induce the learning [10]. However, as shown by [12], MSE loss does not handle streak artifacts very well. Therefore, we adopt the choice of [12] by using a perceptual loss to induce the learning and give more realistic outputs. Let $\phi^{(i)}(\cdot)$ denote the feature maps extracted by the $i$-th layer of the perceptual network $\phi$ and $N_{i}$ denote the number of elements in $\phi^{(i)}(\cdot)$, the perceptual loss can be computed by

[TABLE]

In this work, the perceptual network $\phi$ is a pretrained VGG16 net [9] and we empirically find that $i=8$ works well.

2.1 Network Structure

The generator is based on an encoder-decoder architecture [4]. As shown in Fig. 2, the generator has four encoding blocks (in gray) and four decoding blocks (in blue). Each encoding block contains a convolutional layer followed by a batch normalization layer and a leaky ReLU layer. Similarly, each decoding block contains a deconvolutional layer followed by a batch normalization layer and a ReLU layer. Both the convolutional and deconvolutional layers have a $4\times 4$ kernel with a stride of $2$ so that they can downsample and upsample the outputs, respectively. Outputs from the encoding blocks are shuttled to the corresponding decoding blocks using skip connections. This design allows the low-level context information from the encoding blocks to be used directly together with the decoded high-level information during generation.

A typical discriminator (Fig. 2 Discriminator A) usually contains a set of encoding blocks followed by a classifier to determine the input’s realness. In this case, the discrimination is performed at a fixed granularity that is fine when the task is a generative task such as style transfer or image translation, or there is a systematic error to be corrected such as JPEG decompression or super-resolution. For sparse-view CBCT images, the artifacts appear randomly with different scales. To capture such a variation of artifacts, we propose a discriminator that handles the adversarial training at different granularities.

The core idea is to create a feature pyramid and perform discrimination at multiple scales. As illustrated in Fig. 2 Discriminator B, the network uses two outputs and makes decisions based on different levels of semantic feature maps. We adapt the design from FPN [6] so that the feature pyramid has strong semantics at all scales. Specifically, we first use three encoding blocks to extract features at different levels. Next, we use an upsample block (in purple) to incorporate the stronger semantic features from the top layer into the outputs of the middle layer. The upsample block consists of a unsampling layer and a $3\times 3$ convolutional layer (to smooth the outputs). Because the feature maps from the encoding blocks have different channel sizes, we place a lateral block (in green, essentially a $1\times 1$ convolutional layer) after each encoding block to match this channel difference. In the end, there are two classifiers to make joint decisions on the semantics at different scales. Each classifier contains two blocks. The first block (in yellow) has the same layers as an encoding block, except that the convolutional layer has a $3\times 3$ kernel with a stride of $1$. The second block (in green) is simply a $1\times 1$ convolutional layer with stride $1$. Let $D_{1}(x)$ and $D_{2}(x)$ denote the outputs from the two classifiers, then the new adversarial loss can be given by $\min_{D}\mathcal{L}_{A}(D;G,\Lambda_{1},\Lambda_{2})=\sum_{i=1}^{2}\mathcal{L}_{A}(D_{i};G,\Lambda_{i})$ and $\min_{G}\mathcal{L}_{A}(G;D,\Lambda_{1},\Lambda_{2})=\sum_{i=1}^{2}\mathcal{L}_{A}(G;D_{i},\Lambda_{i})$. We also experimented with deeper discriminators with more classifiers for richer feature semantics, but found that they contribute only minor improvements over the current setting.

2.2 Focus Map

When an image from the generator looks mostly “real” (Fig 3 (b)), the score map (Fig 3 (c)) output by the discriminator will be overwhelmed by borderline scores (those values close to 0.5). This saturates the adversarial training as borderline scores make little contribution to the weight update of the discriminator. To address this problem, we propose to introduce a modulation factor to the adversarial loss so that the borderline scores are down-weighted during training. Observing that when a generated region is visually close to the corresponding region of a dense-view image (Fig 3 (a)), it is more likely to be “real” and causes the discriminator to give a borderline score. Therefore, we use a feature difference map (Fig 3 (d)) to perform this modulation.

Let $\phi_{m,n}^{(j)}(\cdot)$ denote the $(m,n)$-th feature vector of $\phi^{(j)}(\cdot)$, then the $(m,n)$-th element of the feature difference map $\Lambda$ between $x_{d}$ and $G(x_{s})$ is defined as

[TABLE]

where $Z_{j}$ is a normalization term given by

[TABLE]

We use the same perceptual network $\phi$ as the one used for computing the perceptual loss, and $j$ is chosen to match the resolution of $D_{1}(x)$ and $D_{2}(x)$. For the VGG16 net, we use $j=16$ for $\Lambda_{1}$ and $j=9$ for $\Lambda_{2}$.

3 Experiments

Datasets The CBCT images were obtained by a multi-source CBCT scanner dedicated for lower extremities. In total, knee images from 27 subjects are under investigation. Each subject is associated with a sparse-view image and a dense-view image that are reconstructed using 67 and 200 projection views, respectively. Each image is processed, slice by slice, along the sagittal direction where the streak artifacts are most pronounced. During the training, the inputs to the models are $256\times 256$ patches that randomly cropped from the slices.

Models Three variants of the proposed methods as well as two other baseline methods are compared:

(i) Baseline-MSE: a similar approach to [10] by combining MSE loss with GAN. 3D UNet111Identical to the 2D UNet used in this work with all the 2D convolutional and deconvolutional layers replaced by their 3D counterparts. and LSGAN is used for fair comparison;

(ii) Baseline-Perceptual: a similar approach to [12] by combining perceptual loss with GAN. It is also based on our UNet and LSGAN infrastructure for fair comparison;

(iii) Ours-FPN: our method using FPN as the discriminator and setting $\Lambda_{1}=\Lambda_{2}=\mathbf{1}$;

(iv) Ours-Focus: our method using focus map and conventional discriminator (Fig. 2 Discriminator A);

(v) Ours-Focus+FPN: our method using focus map as well as the FPN discriminator.

We train all the models using Adam optimization with the learning rate $lr=10^{-4}$ and $\beta_{1}=0.5$. We use $\lambda_{a}=1.0$, $\lambda_{m}=100$, and $\lambda_{p}=10$ to control the weights between the adversarial loss, the MSE loss, and the perceptual loss. The values are chosen empirically and are the same for all models (if applicable). All the models are trained for 50 epochs with 5-fold cross-validation. We perform all the experiments on an Nvidia GeForce GTX 1070 GPU. During testing, the average processing time on $384\times\times 384\times 417$ CBCT volumes for the 2D UNet (generator of model (ii)-(v)) is 16.05 seconds, and for the 3D UNet (generator of model (i)) is 22.70 seconds.

Experimental Results Fig. 4 shows the qualitative results of the models. Although the baseline methods overall have some improvements over the sparse-view image, they still cannot handle the streak artifacts very well. “Baseline-Perceptual” produces less pronounced artifacts than “Baseline-MSE”, which demonstrates that using perceptual loss and processing the images slice by slice in 2D give better results than MSE loss with 3D generator. Our models (Fig. 4 (e-f)) in general produce less artifacts than the baseline models. We can barely see the streak artifacts. They generally produce similar outputs and the result from “Ours-Focus+FPN” is slightly better than “Ours-FPN” and “Ours-Focus”. This means that using FPN as the discriminator or applying a modulation factor to the adversarial loss can indeed induce the training to artifacts reduction.

We further investigate the image characteristics of each model in a region of interest (ROI). A good model should have similar image characteristics to the dense-view images in ROIs. When looking at the pixel-wise difference between the dense-view ROI and the model ROI, no structure information should be observed, resulting a random noise map. Fig. 5(a) shows the ROI differences of the models. We can see a clear bone structure from the ROI difference map between $x_{s}$ and $x_{d}$ (Fig. 5(a) third row), which demonstrates a significant difference in image characteristics between these two images. For “Baseline-MSE”, the bone structure is less recognizable, showing more similar image characteristics. For “Baseline-Perceptual” and our models, we can hardly see the structural information and mostly observe random noises. This indicates that these models have very similar image characteristics to a dense-view image. We also measure the mean and standard deviation of the pixel values within the ROI. We can see that our models have very close statistics with $x_{d}$, especially the pixel value statistics of “Ours-Focus” and “Ours-Focus+FPN” are almost identical to $x_{d}$, demonstrating better image characteristics.

We then evaluate the models quantitatively by comparing their outputs with the corresponding dense-view CBCT image. Three evaluation metrics are used: structural similarity (SSIM), peak signal-to-noise ratio (PSNR), and root mean square error (RMSE). Higher values for SSIM and PSNR and lower values for RMSE indicate better performance. We can see from Table 1 that the baseline methods give better scores than $x_{s}$. Similar to the case in the qualitative evaluation, “Baseline-Perceptual” performs better than “Baseline-MSE”. Our methods consistently outperform the baseline methods by a significant margin. “Ours-FPN” gives best performance in PSNR and RMSE. However, PSNR and RMSE only measure the pixel level difference between two images. To measure the performance in perceived similarity, SSIM is usually a better choice, and we find “Ours-FPN+Focus” has a slightly better performance on this metric. This confirms our observation in qualitative evaluation.

4 Conclusion

We have presented a novel approach to reducing artifacts from sparsely-reconstructed CBCT images. To our best knowledge, this is the first work that addresses artifacts introduced by sparse-view CBCT reconstruction in a post-processing step. We target this problem using an image-to-image generative model with a perceptual loss as regulation. The model generates perceptually realistic outputs while making the artifacts less pronounced. To further suppress the streak artifacts, we have also proposed a novel FPN based discriminator and a focus map to induce the adversarial training. Experimental results show that the proposed mechanism addresses the streak artifacts much better, and the proposed models outperform strong baseline methods both qualitatively and quantitatively.

Acknowledgement. The work presented here was supported in part by New York State through the Goergen Institute for Data Science at the University of Rochester and the corporate sponsor Carestream Health Inc.