brendt.wohlberg.net
HomePublications
› Publications
› Software

Cite Details

Jiaming Liu, Yu Sun, Weijie Gan, Xiaojian Xu, Brendt Wohlberg and Ulugbek S. Kamilov, "Stochastic Deep Unfolding for Imaging Inverse Problems", in Proceedings of IEEE International Conference on Acoustics, Speech, and Signal Processing (ICASSP), (Toronto, ON, Canada), doi:10.1109/ICASSP39728.2021.9414332, Jun 2021

Abstract

Deep unfolding networks are rapidly gaining attention for solving imaging inverse problems. However, the computational and memory complexity of existing deep unfolding networks scales with the size of the full measurement set, limiting their applicability to certain large-scale imaging inverse problems. We propose SCRED-Net as a novel methodology that introduces a stochastic approximation to the unfolded regularization by denoising (RED) algorithm. Our method uses only a subset of measurements within each cascade block, making it scalable to a large number of measurements for efficient end-to-end training. We present numerical results showing the effectiveness of SCRED-Net on intensity diffraction tomography (IDT) and sparse-view computed tomography (CT). Our results show that SCRED-Net matches the performance of a batch deep unfolding network at a fraction of training and operational complexity.

BibTeX Entry

@inproceedings{liu-2021-stochastic,
author = {Jiaming Liu and Yu Sun and Weijie Gan and Xiaojian Xu and Brendt Wohlberg and Ulugbek S. Kamilov},
title = {Stochastic Deep Unfolding for Imaging Inverse Problems},
year = {2021},
month = Jun,
booktitle = {Proceedings of IEEE International Conference on Acoustics, Speech, and Signal Processing (ICASSP)},
address = {Toronto, ON, Canada},
doi = {10.1109/ICASSP39728.2021.9414332}
}