confidence-aware levenberg-marquardt …...confidence-aware levenberg-marquardt optimization for...

26
Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on Image Processing (ICIP’16) Cosmin Bercea, Andreas Maier, Thomas Köhler September 28, 2016 Pattern Recognition Lab (CS 5)

Upload: others

Post on 31-May-2020

20 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Confidence-aware Levenberg-Marquardt Optimizationfor Joint Motion Estimation and Super-Resolution

2016 IEEE International Conference on Image Processing (ICIP’16)

Cosmin Bercea, Andreas Maier, Thomas KöhlerSeptember 28, 2016Pattern Recognition Lab (CS 5)

Page 2: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Introduction

Page 3: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Multiframe Super-Resolution: Basic Idea

• Given: multiple low-resolution images• Idea: Exploit subpixel motion to reconstruct high-resolution image

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 3

26 low-resolution frames 3 x High-resolution image

Page 4: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Robustness Issues

Super-resolution reconstruction is sensitive to:• Motion estimation uncertainty

Registration is error-prone

1Köhler et al., “Robust Multiframe Super-Resolution Employing Iteratively Re-Weighted Minimization.,” IEEE TCI, 2016.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 4

Superresolved image1

Page 5: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Robustness Issues

Super-resolution reconstruction is sensitive to:• Motion estimation uncertainty

Registration is error-prone• Outliers

• Deviation of the real and assumed motion model! e.g.: non-rigid deformation assuming rigid motion

• Invalid pixels• Space variant noise• . . .

2Yu He et al., “A Nonlinear Least Square Technique for Simultaneous Image Registration and Super-Resolution.,” IEEE TIP, 2007.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 5

Superresolved image2

Page 6: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Proposed Method

Page 7: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Modeling the Image Formation

• Given: sequence of low-resolution framesy = (y(1)>, . . . , y(K )>)>, y(k) 2 RM

• y is assembled from the HR image x 2 RN by:

y = W(✓)x + ✏ (1)

• W(✓) = DHM(✓) models subsampling, blur, and subpixel motion• ✏ is additive noise

• ✓ models a rigid transformation (3 degrees of freedom):! rotation angle ' and translation t = (tu, tv)

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 7

Page 8: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Energy Function

E(x, ✓) = (y�W(✓)x)>B(y�W(✓)x) + �R(x) (2)

• Weighted deviation between observation and model approximation

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 8

Page 9: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Energy Function

E(x, ✓) = y�W(✓)x)>B(y�W(✓)x) + �R(x)

• Weighted deviation between observation and model approximation• Edge preserving WBTV3regularization given by:

R(x) = ||ASx||1 =PX

l=�P

PX

m=�P

����Al,mSl,mx����

1

• S models vertical and horizontal shifts around a local neighborhood P• A are weights to control influence of the prior

3Köhler et al., “Robust Multiframe Super-Resolution Employing Iteratively Re-Weighted Minimization.,” IEEE TCI, 2016.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 9

Page 10: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Non-Linear Least-Squares Estimation

• Our energy function is non linear w.r.t ✓! non-linear least-squares estimation of x and ✓:

E(x, ✓) =

�����

�����

B

12(y�W(✓)x)p

�A12L

12x

!�����

�����

2

2(3)

where L is a majorization of the WBTV term

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 10

Page 11: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Numerical Optimization

Super-Resolution Step

�x,�✓

UpdateWeights Maps

Bt ,At

Find DampingParameter

µt

xt = xt�1 +�x,✓t = ✓t�1 +�✓

t t + 1

• Iterative confidence-aware optimization scheme• The Taylor series expansion of our energy function in (3) yields small

parameter updates according to:

✓�✓�x

◆=h(Pt)>Pt

i�1(Pt)>ft (4)

• P, f derived based on the Jacobian matrix

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 11

Page 12: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Numerical Optimization

Super-Resolution Step

�x,�✓

UpdateWeights Maps

Bt ,At

Find DampingParameter

µt

xt = xt�1 +�x,✓t = ✓t�1 +�✓

t t + 1

Compute small changes �x and �✓ for the high-resolution image x andthe motion parameters ✓

✓�✓�x

◆=h(Pt)>Pt

i�1(Pt)>ft (5)

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 12

Page 13: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Numerical Optimization

Super-Resolution Step

�x,�✓

UpdateWeights Maps

Bt ,At

Find DampingParameter

µt

xt = xt�1 +�x,✓t = ✓t�1 +�✓

t t + 1

Compute small changes �x and �✓ for the high-resolution image x andthe motion parameters ✓ using a Levenberg-Marquardt optimization:

✓�✓�x

◆=h(Pt)>Pt + µ · diag

�(Pt)>Pt

� i�1(Pt)>ft (6)

damping parameter µ: µ = 0! Gauss-Newtonµ� 0! gradient descent

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 13

Page 14: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Numerical Optimization

Super-Resolution Step

�x,�✓

UpdateWeights Maps

Bt ,At

Find DampingParameter

µt

xt = xt�1 +�x,✓t = ✓t�1 +�✓

t t + 1

(a) Original (b) Data Weights B (c) Prior Weights A

Weights are computed proportional to the inverse of the residual erros4

4Köhler et al., “Robust Multiframe Super-Resolution Employing Iteratively Re-Weighted Minimization.,” IEEE TCI, 2016.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 14

Page 15: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Numerical Optimization

Super-Resolution Step

�x,�✓

UpdateWeights Maps

Bt ,At

Find DampingParameter

µt

xt = xt�1 +�x,✓t = ✓t�1 +�✓

t t + 1

• A good damping parameter µ is crucial for a good performance of theLevenberg-Marquardt iterations

• Perform an one-dimensional search to minimize the confidence weightedresidual error:

µt = arg minµ

������(Bt)

12 [y�W (✓(µ)) x(µ)]

������2

2, (7)

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 15

Page 16: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Gauss-Newton vs. Levenberg-Marquardt

Gauss-Newton under practical conditions (e.g affected by outliers):• converges to a worse local minima• converges slowly

2 4 6 8 10 12 14 16 18 20

25

26

27

28

29

30

31

Iteration

PS

NR

[dB

]

2 4 6 8 10 12 14 16 18 20

25

26

27

28

29

30

31

Iteration

PS

NR

[dB

]

Left: Iterations without outliers. Right: Iterations in the presence of outliers due to invalid pixels.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 16

Page 17: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Experiments and Results

Page 18: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Experimental Setup

Experiments:• Real datasets5

• Simulated data with known ground truth6

Compared methods:• Joint Motion Estimation And Super-Resolution (JMSR)7

• Iteratively Re-Weighted Minimization Super-Resolution (IRWSR)8

• Proposed confidence-aware L.M. optimization

5https://users.soe.ucsc.edu/ milanfar/software/sr-datasets.html6http://live.ece.utexas.edu/research/quality/subjective.htm7Yu He et al., “A Nonlinear Least Square Technique for Simultaneous Image Registration and Super-Resolution.,” IEEE TIP, 2007.8Köhler et al., “Robust Multiframe Super-Resolution Employing Iteratively Re-Weighted Minimization.,” IEEE TCI, 2016.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 18

Page 19: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Real Data: Emily Sequence

(a) 2 original frames (b) JMSR (c) IRWSR (d) Proposed

• Head movements generate outliers• Presence of motion uncertainty

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 19

Page 20: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Real Data: Alpaca Sequence

(a) 2 original frames (b) JMSR (c) IRWSR (d) Proposed

• Alpaca movement generates outliers• Presence of motion uncertainty

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 20

Page 21: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Real Data: Results

(e) Original (f) JMSR (g) IRWSR (h) Proposed

• Car drives away from the camera! cannot be modeled with a rigid transformation

• Our method is robust against mis-registered frames and refines motionestimate for the remaining frames

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 21

Page 22: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Simulated Data: Results

Setup:• LR altered with salt and pepper noise• Results averaged over n = 10 random test sequences

PSNR evaluation for sequenceswithout outliers (top row) andsequences with outliers due toinvalid pixels (bottom row):! Increased PSNR by ⇡ 3 dB.

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 22

Page 23: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Simulated Data: Results

(a) Original (b) JMSR (c) IRWSR (d) Proposed

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 23

Page 24: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Summary and Conclusion

Page 25: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Summary and Conclusion

• Combine robustness to outliers in the formation process with therefinement of the motion estimation

• Levenberg-Marquard iteration scheme to boost convergence• Outperform both two-stage and joint state of the art approaches.

Outlook: Extend the model to more general types of motion:! e.g.: affine transformations

MATLAB code of this method is available on our webpage as part of oursuper-resolution toolbox:

www5.cs.fau.de/research/software/multi-frame-super-resolution-toolbox

C. Bercea et. al: “Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution”, 2016 IEEE ICIP 25

Page 26: Confidence-aware Levenberg-Marquardt …...Confidence-aware Levenberg-Marquardt Optimization for Joint Motion Estimation and Super-Resolution 2016 IEEE International Conference on

Thank you very much for your attention!