R3L: Connecting Deep Reinforcement Learning To Recurrent Neural Networks For Image Denoising Via Residual Recovery

Rongkai Zhang, Jiang Zhu, Zhiyuan Zha, Justin Dauwels, Bihan Wen

Research output: Chapter in Book/Conference proceedings/Edited volumeConference contributionScientificpeer-review

8 Downloads (Pure)

Abstract

State-of-the-art image denoisers exploit various types of deep neural networks via deterministic training. Alternatively, very recent works utilize deep reinforcement learning for restoring images with diverse or unknown corruptions. Though deep reinforcement learning can generate effective policy networks for operator selection or architecture search in image restoration, how it is connected to the classic deterministic training in solving inverse problems remains unclear. In this work, we propose a novel image denoising scheme via Residual Recovery using Reinforcement Learning, dubbed R3L. We show that R3L is equivalent to a deep recurrent neural network that is trained using a stochastic reward, in contrast to many popular denoisers using supervised learning with deterministic losses. To benchmark the effectiveness of reinforcement learning in R3L, we train a recurrent neural network with the same architecture for residual recovery using the deterministic loss, thus to analyze how the two different training strategies affect the denoising performance. With such a unified benchmarking system, we demonstrate that the proposed R3L has better generalizability and robustness in image denoising when the estimated noise level varies, comparing to its counterparts using deterministic training, as well as various state-of the-art image denoising algorithms.
Original languageEnglish
Title of host publication2021 IEEE International Conference on Image Processing (ICIP)
Subtitle of host publicationProceedings
Place of PublicationPiscataway
PublisherIEEE
Pages1624-1628
Number of pages5
ISBN (Electronic)978-1-6654-4115-5
ISBN (Print)978-1-6654-3102-6
DOIs
Publication statusPublished - 2021
Event2021 IEEE International Conference on Image Processing (ICIP) - Virtual at Anchorage, United States
Duration: 19 Sep 202122 Sep 2021

Conference

Conference2021 IEEE International Conference on Image Processing (ICIP)
Country/TerritoryUnited States
CityVirtual at Anchorage
Period19/09/2122/09/21

Bibliographical note

Green Open Access added to TU Delft Institutional Repository ‘You share, we take care!’ – Taverne project https://www.openaccess.nl/en/you-share-we-take-care

Otherwise as indicated in the copyright section: the publisher is the copyright holder of this work and the author uses the Dutch legislation to make this work public

Keywords

  • Training
  • Recurrent neural networks
  • Supervised learning
  • Reinforcement learning
  • Benchmark testing
  • Search problems
  • Robustness
  • Recurrent Neural Network
  • Deep Reinforcement Learning
  • Image Denoising
  • Residual Recovery

Fingerprint

Dive into the research topics of 'R3L: Connecting Deep Reinforcement Learning To Recurrent Neural Networks For Image Denoising Via Residual Recovery'. Together they form a unique fingerprint.

Cite this