Variational Networks: An Optimal Control Approach to Early Stopping Variational Methods for Image Restoration.

Alexander Effland, Erich Kobler, Karl Kunisch, Thomas Pock
Author Information
  1. Alexander Effland: 1Institute of Computer Graphics and Vision, Graz University of Technology, Graz, Austria. ORCID
  2. Erich Kobler: 1Institute of Computer Graphics and Vision, Graz University of Technology, Graz, Austria.
  3. Karl Kunisch: 2Institute of Mathematics and Scientific Computing, University of Graz, Graz, Austria.
  4. Thomas Pock: 1Institute of Computer Graphics and Vision, Graz University of Technology, Graz, Austria.

Abstract

We investigate a well-known phenomenon of variational approaches in image processing, where typically the best image quality is achieved when the gradient flow process is stopped before converging to a stationary point. This paradox originates from a tradeoff between optimization and modeling errors of the underlying variational model and holds true even if deep learning methods are used to learn highly expressive regularizers from data. In this paper, we take advantage of this paradox and introduce an optimal stopping time into the gradient flow process, which in turn is learned from data by means of an optimal control approach. After a time discretization, we obtain variational networks, which can be interpreted as a particular type of recurrent neural networks. The learned variational networks achieve competitive results for image denoising and image deblurring on a standard benchmark data set. One of the key theoretical results is the development of first- and second-order conditions to verify optimal stopping time. A nonlinear spectral analysis of the gradient of the learned regularizer gives enlightening insights into the different regularization properties.

Keywords

References

  1. IEEE Trans Image Process. 2014 Mar;23(3):1060-72 [PMID: 24474375]
  2. Nature. 2015 May 28;521(7553):436-44 [PMID: 26017442]
  3. IEEE Trans Pattern Anal Mach Intell. 2017 Jun;39(6):1256-1272 [PMID: 27529868]
  4. Magn Reson Med. 2018 Jun;79(6):3055-3071 [PMID: 29115689]

Word Cloud

Created with Highcharts 10.0.0variationalimagenetworksVariationalgradientflowdataoptimalstoppingtimelearnedprocessparadoxlearningcontrolresultsOptimalEarlyinvestigatewell-knownphenomenonapproachesprocessingtypicallybestqualityachievedstoppedconvergingstationarypointoriginatestradeoffoptimizationmodelingerrorsunderlyingmodelholdstrueevendeepmethodsusedlearnhighlyexpressiveregularizerspapertakeadvantageintroduceturnmeansapproachdiscretizationobtaincaninterpretedparticulartyperecurrentneuralachievecompetitivedenoisingdeblurringstandardbenchmarksetOnekeytheoreticaldevelopmentfirst-second-orderconditionsverifynonlinearspectralanalysisregularizergivesenlighteninginsightsdifferentregularizationpropertiesNetworks:ControlApproachStoppingMethodsImageRestorationDeepGradienttheoryproblems

Similar Articles

Cited By (1)