Abstract
In this work we present a new type of efficient deep-unrolling networks forsolving imaging inverse problems. Classical deep-unrolling methods require fullforward operator and its adjoint across each layer, and hence can becomputationally more expensive than other end-to-end methods such asFBP-ConvNet, especially in 3D image reconstruction tasks. We propose astochastic (ordered-subsets) extension of the Learned Primal-Dual (LPD) whichis the state-of-the-art unrolling network. In our unrolling network, we onlyuse a subset of the forward and adjoint operator, to achieve computationalefficiency. We consider 3 ways of training the proposed network to cope withdifferent scenarios of the availability of the training data, including (1)supervised training on paired data, (2) unsupervised adversarial training whichenable us to train the network without paired ground-truth data, (3)equivariant self-supervised training approach, which utilizes equivariantstructure which is prevalent in many imaging applications, and only requiresmeasurement data. Our numerical results demonstrate the effectiveness of ourapproach in X-ray CT imaging task, showing that our networks achieve similarreconstruction accuracies as the full-batch LPD, while require only a fractionof the computation.