English

Recurrent Generative Adversarial Networks for Proximal Learning and Automated Compressive Image Recovery

Artificial Intelligence 2017-11-29 v1 Information Retrieval Machine Learning

Abstract

Recovering images from undersampled linear measurements typically leads to an ill-posed linear inverse problem, that asks for proper statistical priors. Building effective priors is however challenged by the low train and test overhead dictated by real-time tasks; and the need for retrieving visually "plausible" and physically "feasible" images with minimal hallucination. To cope with these challenges, we design a cascaded network architecture that unrolls the proximal gradient iterations by permeating benefits from generative residual networks (ResNet) to modeling the proximal operator. A mixture of pixel-wise and perceptual costs is then deployed to train proximals. The overall architecture resembles back-and-forth projection onto the intersection of feasible and plausible images. Extensive computational experiments are examined for a global task of reconstructing MR images of pediatric patients, and a more local task of superresolving CelebA faces, that are insightful to design efficient architectures. Our observations indicate that for MRI reconstruction, a recurrent ResNet with a single residual block effectively learns the proximal. This simple architecture appears to significantly outperform the alternative deep ResNet architecture by 2dB SNR, and the conventional compressed-sensing MRI by 4dB SNR with 100x faster inference. For image superresolution, our preliminary results indicate that modeling the denoising proximal demands deep ResNets.

Keywords

Cite

@article{arxiv.1711.10046,
  title  = {Recurrent Generative Adversarial Networks for Proximal Learning and Automated Compressive Image Recovery},
  author = {Morteza Mardani and Hatef Monajemi and Vardan Papyan and Shreyas Vasanawala and David Donoho and John Pauly},
  journal= {arXiv preprint arXiv:1711.10046},
  year   = {2017}
}

Comments

11 pages, 11 figures