Generation of attenuation correction factors from time-of-flight PET emission data using high-resolution residual U-net

Biomed Phys Eng Express. 2021 Sep 6;7(6). doi: 10.1088/2057-1976/ac21aa.

Abstract

Attenuation correction of annihilation photons is essential in PET image reconstruction for providing accurate quantitative activity maps. In the absence of an aligned CT device to obtain attenuation information, we propose the high-resolution residual U-net (HRU-Net) to extract attenuation correction factors (ACF) directly from time-of-flight (TOF) PET emission data. HRU-Net is built upon the U-Net encoding-decoding architecture and it utilizes four blocks of modified residual connections in each stage. In each residual block, concatenation is performed to incorporate input and output feature vectors. In addition, flexible and efficient elements of convolutional neural network (CNN) such as dilated convolutions, pre-activation order of a batch normalization (BN) layer, a rectified linear unit (ReLU) layer and a convolution layer, and residual connections are utilized to extract high resolution features. To illustrate the effectiveness of the proposed method, HRU-Net estimated ACF, attenuation maps and activity maps are compared with maximum likelihood ACF (MLACF) algorithm, U-Net, and HC-Net. An ablation study is conducted using non-TOF and TOF sinograms as inputs of networks. The experimental results show that HRU-Net with TOF projections as inputs leads to normalized root mean square error (NRMSE) of 4.84% ± 1.58%, outperforming MLACF, U-Net and HC-Net with NRMSE of 47.82% ± 13.62%, 6.92% ± 1.94%, and 7.99% ± 2.49%, respectively.

Keywords: attenuation correction; convolutional neural network (CNN); deep learning; positron emission tomography (PET).

Publication types

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

MeSH terms

  • Algorithms
  • Neural Networks, Computer
  • Positron-Emission Tomography*