Improved GAN: Using a transformer module generator approach for material decomposition

Comput Biol Med. 2022 Oct:149:105952. doi: 10.1016/j.compbiomed.2022.105952. Epub 2022 Aug 11.

Abstract

Dual-energy computed tomography (CT) can be used for material decomposition, allowing for the precise quantitative mapping of body substances; this has a wide range of clinical applications, including disease diagnosis, treatment response evaluation and prognosis prediction. However, dual-energy CT has not yet become the mainstream technique in most clinical settings due to its limited accessibility. To fully take advantage of material quantification, researchers have attempted to use deep learning to generate material decomposition maps from conventional single-energy CT images, mainly by synthesizing another single-energy CT image from a conventional single-energy CT image to form a dual-energy CT image first and then generate material decomposition maps. This is not a straightforward process, and it potentially introduces many inaccuracies after multiple steps. In this work, we proposed a generative adversarial network (GAN) framework as the base and improved its generator; this approach combines convolutional neural networks (CNNs) and a transformer module to directly generate material decomposition maps from conventional single-energy CT images. Our model pays attention to both local and global information. Then, we compared our method with 6 competitive deep learning methods on water (calcium) and calcium (water) substrate density image datasets. The average PSNR, SSIM, MAE, and RMSE of the generated and ground truth of the water (calcium) substrate density images were 32.7207, 0.9685, 0.0323, and 0.0555, respectively. Furthermore, the average PSNR, SSIM, MAE, and RMSE of the generated and ground truth of the calcium (water) substrate density images were 30.2823, 0.9449, 0.0652, and 0.0715, respectively. Our model achieved better performance and stronger stability than competing approaches.

Keywords: Deep learning; Dual-energy CT; Generative adversarial network; Material decomposition; Transformer module.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Calcium*
  • Image Processing, Computer-Assisted* / methods
  • Neural Networks, Computer
  • Tomography, X-Ray Computed / methods
  • Water

Substances

  • Water
  • Calcium