Introducing diffusion tensor to high order variational model for image reconstruction

Second order total variation (SOTV) models have advantages for image restoration over their first order counterparts including their ability to remove the staircase artefact in the restored image. However, such models tend to blur the reconstructed image when discretised for numerical solution [1–5]. To overcome this drawback, we introduce a new tensor weighted second order (TWSO) model for image restoration. Specifically, we develop a novel regulariser for the SOTV model that uses the Frobenius norm of the product of the isotropic SOTV Hessian matrix and an anisotropic tensor. We then adapt the alternating direction method of multipliers (ADMM) to solve the proposed model by breaking down the original problem into several subproblems. All the subproblems have closed-forms and can be solved efficiently. The proposed method is compared with state-of-the-art approaches such as tensor-based anisotropic diffusion, total generalised variation, and Euler's elastica. We validate the proposed TWSO model using extensive experimental results on a large number of images from the Berkeley BSDS500. We also demonstrate that our method effectively reduces both the staircase and blurring effects and outperforms existing approaches for image inpainting and denoising applications.

[1]  M. Grasmair,et al.  Anisotropic Total Variation Filtering , 2010 .

[2]  Carola-Bibiane Schönlieb,et al.  Combined First and Second Order Total Variation Inpainting using Split Bregman , 2013, Image Process. Line.

[3]  Antonin Chambolle,et al.  A First-Order Primal-Dual Algorithm for Convex Problems with Applications to Imaging , 2011, Journal of Mathematical Imaging and Vision.

[4]  Li Bai,et al.  Implementation of high‐order variational models made easy for image processing , 2016 .

[5]  Joachim Weickert,et al.  Anisotropic diffusion in image processing , 1996 .

[6]  Michael Unser,et al.  Hessian Schatten-Norm Regularization for Linear Inverse Problems , 2012, IEEE Transactions on Image Processing.

[7]  Kai-Fu Yang,et al.  Color Constancy Using Double-Opponency , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[8]  Petros Maragos,et al.  Tensor-based image diffusions derived from generalizations of the Total Variation and Beltrami Functionals , 2010, 2010 IEEE International Conference on Image Processing.

[9]  Christoph Schnörr,et al.  A class of quasi-variational inequalities for adaptive image denoising and decomposition , 2012, Computational Optimization and Applications.

[10]  Karl Kunisch,et al.  Total Generalized Variation , 2010, SIAM J. Imaging Sci..

[11]  Tony F. Chan,et al.  Euler's Elastica and Curvature-Based Inpainting , 2003, SIAM J. Appl. Math..

[12]  Michael Felsberg,et al.  On Tensor-Based PDEs and Their Corresponding Variational Formulations with Application to Color Image Denoising , 2012, ECCV.

[13]  Marc Teboulle,et al.  A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems , 2009, SIAM J. Imaging Sci..

[14]  Ke Chen,et al.  Image denoising using the Gaussian curvature of the image surface , 2016 .

[15]  M. Bergounioux,et al.  A Second-Order Model for Image Denoising , 2010 .

[16]  Jitendra Malik,et al.  Scale-Space and Edge Detection Using Anisotropic Diffusion , 1990, IEEE Trans. Pattern Anal. Mach. Intell..

[17]  Arvid Lundervold,et al.  Noise removal using fourth-order partial differential equation with applications to medical magnetic resonance images in space and time , 2003, IEEE Trans. Image Process..

[18]  L. Rudin,et al.  Nonlinear total variation based noise removal algorithms , 1992 .

[19]  R. W. Liu,et al.  Generalized total variation-based MRI Rician denoising model with spatially adaptive regularization parameters. , 2014, Magnetic resonance imaging.

[20]  Wanquan Liu,et al.  Image Segmentation with Depth Information via Simplified Variational Level Set Formulation , 2017, Journal of Mathematical Imaging and Vision.

[21]  Tommi Kärkkäinen,et al.  A New Augmented Lagrangian Approach for L1-mean Curvature Image Denoising , 2015, SIAM J. Imaging Sci..

[22]  Antonin Chambolle,et al.  An Upwind Finite-Difference Method for Total Variation-Based Image Smoothing , 2011, SIAM J. Imaging Sci..

[23]  Xue-Cheng Tai,et al.  Augmented Lagrangian Method, Dual Methods, and Split Bregman Iteration for ROF, Vectorial TV, and High Order Models , 2010, SIAM J. Imaging Sci..

[24]  Li Bai,et al.  New variational image decomposition model for simultaneously denoising and segmenting optical coherence tomography images , 2015, Physics in medicine and biology.

[25]  Tony F. Chan,et al.  Mathematical Models for Local Nontexture Inpaintings , 2002, SIAM J. Appl. Math..

[26]  Defeng Wang,et al.  Box-constrained second-order total generalized variation minimization with a combined L1,2 data-fidelity term for image reconstruction , 2015, J. Electronic Imaging.

[27]  M. Hintermüller,et al.  Functional-analytic and numerical issues in splitting methods for total variation-based image reconstruction , 2014 .

[28]  Carola-Bibiane Schönlieb,et al.  A Combined First and Second Order Variational Approach for Image Reconstruction , 2012, Journal of Mathematical Imaging and Vision.

[29]  Chang-Ock Lee,et al.  A Nonlinear Structure Tensor with the Diffusivity Matrix Composed of the Image Gradient , 2009, Journal of Mathematical Imaging and Vision.

[30]  Li Bai,et al.  Denoising optical coherence tomography using second order total generalized variation decomposition , 2016, Biomed. Signal Process. Control..

[31]  Xue-Cheng Tai,et al.  Some Facts About Operator-Splitting and Alternating Direction Methods , 2016 .

[32]  Xue-Cheng Tai,et al.  Augmented Lagrangian method for a mean curvature based image denoising model , 2013 .

[33]  Li Bai,et al.  An edge-weighted second order variational model for image decomposition , 2016, Digit. Signal Process..

[34]  Hanno Scharr,et al.  Diffusion filtering without parameter tuning: Models and inference tools , 2010, 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition.

[35]  Li Bai,et al.  Automated segmentation of retinal layers from optical coherence tomography images using geodesic distance , 2016, Pattern Recognit..

[36]  Carola-Bibiane Schönlieb,et al.  Partial Differential Equation Methods for Image Inpainting , 2015, Cambridge monographs on applied and computational mathematics.

[37]  Xue-Cheng Tai,et al.  Augmented Lagrangian Method for Generalized TV-Stokes Model , 2012, J. Sci. Comput..

[38]  Stephen P. Boyd,et al.  Distributed Optimization and Statistical Learning via the Alternating Direction Method of Multipliers , 2011, Found. Trends Mach. Learn..

[39]  Richard G. Baraniuk,et al.  A Field Guide to Forward-Backward Splitting with a FASTA Implementation , 2014, ArXiv.

[40]  Wanquan Liu,et al.  Fast algorithm for color texture image inpainting using the non-local CTV model , 2015, J. Glob. Optim..

[41]  M. Nikolova A Variational Approach to Remove Outliers and Impulse Noise , 2004 .

[42]  Michael Felsberg,et al.  A Tensor Variational Formulation of Gradient Energy Total Variation , 2015, EMMCVPR.

[43]  Xiaoming Yuan,et al.  Adaptive Primal-Dual Hybrid Gradient Methods for Saddle-Point Problems , 2013, 1305.0546.

[44]  Yongjie Li,et al.  A Color Constancy Model with Double-Opponency Mechanisms , 2013, 2013 IEEE International Conference on Computer Vision.

[45]  Xue-Cheng Tai,et al.  A Fast Algorithm for Euler's Elastica Model Using Augmented Lagrangian Method , 2011, SIAM J. Imaging Sci..

[46]  Petros Maragos,et al.  Structure Tensor Total Variation , 2015, SIAM J. Imaging Sci..

[47]  Zheng Xu,et al.  Adaptive ADMM with Spectral Penalty Parameter Selection , 2016, AISTATS.

[48]  Li Bai,et al.  New second order Mumford–Shah model based on Γ-convergence approximation for image processing , 2016 .