Source-linked AI summary
TransCT: Dual-path Transformer for Low Dose Computed Tomography
Zhicheng Zhang, Lequan Yu, Xiaokun Liang, Wei Zhao, Lei Xing
TL;DR
Reducing X-ray exposure lowers radiation dose but degrades CT image quality. TransCT splits noisy LDCT into HF and LF components, uses LF textures to refine HF features, and reconstructs images piecewise. On the Mayo dataset, the authors report improved image-quality metrics over LDCT and superior performance and robustness relative to other methods.
Problem
Lowering X-ray exposure reduces radiation dose but degrades CT image quality, motivating methods that preserve image quality at lower dose.
Source: §1 · p. 1Method
TransCT decomposes LDCT into HF and LF parts, extracts LF content and texture features and HF embeddings, then uses a modified transformer and piecewise reconstruction.
Source: §1 · p. 2Results
Against LDCT, TransCT decreases RMSE by 40.5%, improves SSIM by 12.3%, and VIF by 93.7%.
Source: Quantitative Analysis · p. 6Takeaways & Limitations
On the Mayo dataset, the authors report that TransCT outperforms other methods and has superior robustness.
Source: Quantitative Analysis · p. 7
Abstract
from arXiv · showhide
Low dose computed tomography (LDCT) has attracted more and more attention in routine clinical diagnosis assessment, therapy planning, etc., which can reduce the dose of X-ray radiation to patients. However, the noise caused by low X-ray exposure degrades the CT image quality and then affects clinical diagnosis accuracy. In this paper, we train a transformer-based neural network to enhance the final CT image quality. To be specific, we first decompose the noisy LDCT image into two parts: high-frequency (HF) and low-frequency (LF) compositions. Then, we extract content features (X_{L_c}) and latent texture features (X_{L_t}) from the LF part, as well as HF embeddings (X_{H_f}) from the HF part. Further, we feed X_{L_t} and X_{H_f} into a modified transformer with three encoders and decoders to obtain well-refined HF texture features. After that, we combine these well-refined HF texture features with the pre-extracted X_{L_c} to encourage the restoration of high-quality LDCT images with the assistance of piecewise reconstruction. Extensive experiments on Mayo LDCT dataset show that our method produces superior results and outperforms other methods.
1 Introduction
Reducing X-ray exposure is a practical way to lower radiation dose, but lower exposure degrades CT images. TransCT uses LF latent textures to refine HF features and reconstruct LDCT images.
1 Introduction
- Lowering single X-ray exposure is described as the most convenient and affordable dose-reduction strategy, though it degrades CT image quality.The paper frames preserving image quality while reducing radiation dose as the research goal.Source: §1 · p. 1
- Prior LDCT approaches include model-based methods limited by model-description accuracy and data-driven reconstruction methods, including CNN and iterative-learning approaches.The paper also notes AUTOMAP’s first fully connected layer has high memory requirements for large-scale CT reconstruction.Source: §1 · pp. 1–2
- CNN-based LDCT methods rely on cascaded convolutions with limited receptive fields, hindering modeling of long-range dependencies and structural information.Transformers can model interactions between distant image regions through attention.Source: §1 · p. 2
- TransCT decomposes noisy LDCT into HF and LF parts, using LF content and latent textures alongside HF embeddings in a transformer-based restoration pipeline.The refined HF features are combined with LF content for piecewise reconstruction.Source: §1 · pp. 2–3
2 Method
TransCT separates noisy LDCT into frequency components, extracts LF content and texture features, and uses a modified transformer to refine HF features. It combines the refined features with LF content through staged reconstruction and trains against NDCT using MSE.
2 Method
- A Gaussian filter with standard deviation 1.5 decomposes the noisy LDCT image into HF and LF components.The method uses LF latent textures to assist noise removal in the HF part.Source: §2 · p. 3
2.1 TransCT
- Two shallow CNN paths extract content features and latent texture features from the LF image for subsequent reconstruction and transformer processing.The architecture also extracts HF embeddings from the HF component.Source: §2 · p. 3, Sequence · p. 4
Sequence
- The LF texture features and HF embeddings are reshaped into separate sequences as transformer inputs.The sequences provide token representations for the LF and HF components.Source: Sequence · p. 4
Transformer
- The modified transformer has three encoders and three decoders; decoder attention uses the final LF texture sequence as its key and value.The encoder processes LF tokens, while the HF sequence is passed through decoder attention.Source: Transformer · p. 4
Piecewise Reconstruction
- The transformer output is combined with LF content features and reconstructed in stages using residual blocks and sub-pixel upsampling.The first stage combines output Y with XLc2; the next combines the upsampled features with XLc1.Source: Piecewise Reconstruction · pp. 4–5
2.2 Loss Function
- The model is trained with MSE between its output and the normal-dose CT image.The passage identifies the normal-dose image as IND, the low-dose input as ILD, and θ as the network parameters.Source: §2.2 · p. 5
2.3 Implementation
- Training used Adam for 300 epochs with batch size 8 and a learning rate reduced from 0.0001 to 0.00001 after 180 epochs.The implementation was based on TensorFlow and used Xavier initialization.Source: §2.3 · p. 5
3 Experiments
Experiments used Mayo data and compared TransCT with established LDCT baselines. The authors report quantitative and visual gains, then examine piecewise reconstruction, model size, and module choices.
Datasets
- The dataset contains normal-dose and simulated quarter-dose abdominal CT from 10 patients, split into 7 training, 1 validation, and 2 testing cases.The comparison baselines were NLM, RED-CNN, and MAP-NN; the latter two were retrained on the same dataset.Source: Datasets · p. 5, Comparison with other methods · p. 5
Comparison with other methods
- On randomly selected Mayo test images, TransCT is closer to NDCT than LDCT; NLM obscures vessels, while RED-CNN and MAP-NN add light tissues and shadows.Fig. 2 shows the methods, and Fig. 3 shows zoomed regions.Source: Comparison with other methods · p. 6, Figure 2 · p. 6, Figure 3 · p. 7
- Against LDCT, TransCT decreases RMSE by 40.5%, improves SSIM by 12.3%, and improves VIF by 93.7%.The comparisons used 5-fold cross-validation on the Mayo dataset.Source: Quantitative Analysis · p. 6, Table 1 · p. 7
Quantitative Analysis
- With tube currents of 80mA for NDCT and 20mA for LDCT, the authors report that TransCT outperforms other methods with superior robustness.Source: Quantitative Analysis · p. 7
On the Influence of Piecewise Reconstruction
- The modified model's RMSE falls faster during about the first 20 epochs, but its convergence is inferior to TransCT with piecewise reconstruction.The comparison removes the piecewise reconstruction and several convolution layers.Source: §3.1 · p. 7, On the Influence of Piecewise… · p. 8, Figure 4 · p. 8
- Increasing the feed-forward hidden dimension from c to 2c or 4c reduces convergence-curve fluctuation and final convergent RMSE; TransCT uses 8c.Source: On the Influence of Model Size · p. 8
On the Influence of Model Size
- Replacing the transformer module with convolutional/residual blocks or removing the HF path reduced validation performance relative to TransCT with both modules.The ablations are reported in Table 2.Source: On the Influence of Model Size · p. 8
4 Conclusion
TransCT applies a transformer to LDCT by exploiting long-range pixel dependencies and using low-frequency texture features to restore high-frequency features. The authors propose future self-supervised learning to reduce training-data needs.
4 Conclusion
- TransCT is presented as the first transformer-based LDCT network, designed to explore long-range dependencies between image pixels.Source: §4 · p. 8
- The method uses low-frequency latent texture features to restore noisy high-frequency features, then combines low-frequency content and high-frequency features in piecewise reconstruction.The authors identify self-supervised learning as future work to lower training-data needs.Source: §4 · p. 9
Related papers
This paper cites
- Attention Is All You Need2017
- BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding2018
- ADAM: A METHOD FOR STOCHASTIC OPTIMIZATION2014
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale2020
- TensorFlow: A system for large-scale machine learning2016
- Self-Attention Generative Adversarial Networks2018