Link to paper

The full paper is available here.

You can also find the paper on PapersWithCode here.

Abstract

  • Images taken in low light conditions have motion blur and photon shot noise.
  • State-of-the-art image restoration networks are limited to well-illuminated scenes and don’t perform well with photon shot noise.
  • A new blur estimation technique is proposed to address photon-limited conditions.
  • The proposed method uses a gradient-based backpropagation method to estimate the blur kernel.
  • The blur kernel is modeled using a low-dimensional representation with key points on the motion trajectory.
  • This low-dimensional representation provides improved kernel estimates and better deconvolution performance.

Paper Content

Introduction

  • Paper studies photon-limited blind deconvolution problem
  • Blur kernel and latent clean image are recovered from blurred image
  • Forward model includes blurred-noisy image, latent clean image, and blur kernel
  • Deep neural networks have been used to solve blind image deconvolution
  • Iterative kernel estimation method proposed to update blur kernel
  • Kernel modeled in low-dimensional space by specifying key points of motion trajectory
  • Iterative framework used to prove regularity of kernel estimation problem
  • Better blur kernel estimates in photon-limited regimes
  • Traditional Blind Deconvolution uses joint optimization framework with kernel and image priors
  • Levin et. al pointed out that this framework favours the no-blur degenerate solution
  • Some methods model the blur kernel in terms of the camera trajectory and then recover both the trajectory and the clean image
  • Poisson deconvolution problem has been studied for decades
  • Deep Learning Methods have been proposed for the blind image deblurring task
  • Neural Networks and Iterative Methods have shown promising results

Method

Kernel as structured motion estimation

  • Camera motion blur can be modeled as a latent clean image convolved with a blur kernel.
  • The blur kernel is generated from a two-dimensional trajectory.
  • A differentiable and low-dimensional representation of the blur kernel is proposed.
  • A neural network is used to learn the transformations from the key points to the blur kernels.

Proposed iterative scheme

  • Initial estimate of direction and magnitude of blur is generated
  • Initialization uses rectilinear motion kernel, parameterized by length and orientation
  • Iterative scheme is divided into three stages
  • Stage I searches for appropriate kernel representation in latent space
  • Stage II fine-tunes kernel obtained from Stage I
  • Initialization uses minor variation of kernel estimation in PolyBlur
  • Cost function is evaluated using non-blind solver and Kernel-Trajectory Network
  • Cost function includes reblurring loss and 1-norm sparsity prior
  • Networks are retrained using synthetic and GoPro datasets
  • Poisson noise is added to blurred images for retraining

Quantitative comparison

  • Evaluated proposed method on 3 datasets
  • Compared to state-of-the-art deblurring methods
  • Used PSNR, SSIM, and LPIPS-Alex/VGG to compare image quality
  • Evaluated on BSD100 with synthetic blur and Poisson noise
  • Evaluated on Levin dataset with Poisson noise
  • Evaluated on RealBlur-J dataset with retrained end-to-end networks

Qualitative comparison

  • End-to-end trained methods compared to approach in supplementary document
  • Figure 1 and 2 show examples from RealBlur dataset
  • Figure 3 shows examples from PLDD dataset
  • Figure 4 shows reconstruction results on synthetically blurred images from Levin dataset

Ablation study

  • Performance of the scheme improves when increasing the number of key points used for representing kernels.
  • Using a Kernel Trajectory Network improves the performance of the scheme significantly.

Conclusion

  • We use an iterative framework for the photon-limited blind deconvolution problem.
  • We use a non-blind solver to deconvolve Poisson corrupted and blurred images.
  • We outperform end-to-end trained image restoration networks.
  • We use a Poisson noise model for photon-limited settings.
  • The signal-to-noise ratio for the measurements is low in the photon-limited regime.
  • We explore photon-limited deconvolution for α ∈ [10,40].
  • We use a kernel initialization method to estimate a rectilinear kernel.
  • We use a KTN architecture to represent the motion blur.
  • We use symmetric boundary conditions for the inverse problem scheme.
  • We set the initial step size as δ = 10^5 and δ = 2.0 for Stage I and II respectively.
  • We limit the number of iterations to 150.
  • We use a Nvidia TitanX GPU for the experiments.
  • We view the blur kernel as the continuous camera trajectory reduced to K key points.
  • We learn a differentiable representation from the vectorized K key points to a blur kernel.
  • We calculate the reblurring loss using symmetric boundary conditions.