Iteration-Dependent Networks and Losses for Unrolled deep Learned FBSEM PET Image Reconstruction
Corda-D'Incan G, Schnabel JA, Reader AJ (2020)
Publication Type: Conference contribution
Publication year: 2020
Publisher: Institute of Electrical and Electronics Engineers Inc.
Conference Proceedings Title: 2020 IEEE Nuclear Science Symposium and Medical Imaging Conference, NSS/MIC 2020
Event location: Boston, MA, USA
ISBN: 9781728176932
DOI: 10.1109/NSS/MIC42677.2020.9507780
Abstract
We present here an enhanced version of FBSEM-Net, a deep learned regularised model-based image reconstruction algorithm. FBSEM-Net unrolls the maximum a posteriori expectation-maximisation algorithm and replaces the regularisation step by a residual convolutional neural network. Both the gradient of the prior and the regularisation strength are learnt by the network from training data. Nonetheless, some issues arise from its original implementation that we improve upon in this work to obtain a more practical implementation. Specifically, in this implementation, two theoretical improvements are included: i) iteration-dependent networks are used which allows adaptation to varying noise levels as the number of iterations evolves, ii) iteration-dependent targets are used, so that the deep learnt regulariser remains a pure denoising step without any artificial acceleration of the algorithm. Furthermore, we present a new sequential training method for fully unrolled deep networks where the iterative reconstruction is split and the network is trained on each of its modules separately to match the total number of iterations used to reconstruct the targets. The results obtained on 2D simulated test data show that FBSEM-Net using iteration-dependent networks outperforms the original version. Additionally, we found that using iteration-dependent targets not only helps to reduce the variance for different training runs of the network, thus offering greater stability, but also gives the possibility of using a lower number of iterations for test time than what was used for training. Ultimately, we demonstrate that sequential training successfully addresses potential memory issues raised during the training of unrolled networks, without notably impacting the network's performance compared to conventional training.
Involved external institutions
United Kingdom (GB)How to cite
APA:
Corda-D'Incan, G., Schnabel, J.A., & Reader, A.J. (2020). Iteration-Dependent Networks and Losses for Unrolled deep Learned FBSEM PET Image Reconstruction. In 2020 IEEE Nuclear Science Symposium and Medical Imaging Conference, NSS/MIC 2020. Boston, MA, USA: Institute of Electrical and Electronics Engineers Inc..
MLA:
Corda-D'Incan, Guillaume, Julia A. Schnabel, and Andrew J. Reader. "Iteration-Dependent Networks and Losses for Unrolled deep Learned FBSEM PET Image Reconstruction." Proceedings of the 2020 IEEE Nuclear Science Symposium and Medical Imaging Conference, NSS/MIC 2020, Boston, MA, USA Institute of Electrical and Electronics Engineers Inc., 2020.
BibTeX: Download