Upscaling Vector Approximate Message Passing

In this paper we consider the problem of recovering a signal x of size N from noisy and compressed measurements y = Ax + w of size M, where the measurement matrix A is right-orthogonally invariant (ROI). Vector Approximate Message Passing (VAMP) demonstrates great reconstruction results for even highly ill-conditioned matrices A in relatively few iterations. However, performing each iteration is challenging due to either computational or memory point of view. On the other hand, a recently proposed Conjugate Gradient (CG) Expectation Propagation (CG-EP) framework is able to sacrifice some performance for efficiency, but requires access to exact singular spectrum of A. In this work we develop a CG-VAMP algorithm that does not require such information, is feasible to implement and converges to the neighborhood of the original VAMP.

[1]  Keigo Takeuchi,et al.  Rigorous Dynamics of Expectation-Propagation-Based Signal Recovery from Unitarily Invariant Measurements , 2020, IEEE Transactions on Information Theory.

[2]  Thierry Blu,et al.  Monte-Carlo Sure: A Black-Box Optimization of Regularization Parameters for General Denoising Algorithms , 2008, IEEE Transactions on Image Processing.

[3]  Sundeep Rangan,et al.  Vector approximate message passing , 2017, 2017 IEEE International Symposium on Information Theory (ISIT).

[4]  Chao-Kai Wen,et al.  Rigorous dynamics of expectation-propagation signal detection via the conjugate gradient method , 2017, 2017 IEEE 18th International Workshop on Signal Processing Advances in Wireless Communications (SPAWC).

[5]  Richard G. Baraniuk,et al.  From Denoising to Compressed Sensing , 2014, IEEE Transactions on Information Theory.

[6]  C. Stein Estimation of the Mean of a Multivariate Normal Distribution , 1981 .

[7]  Alessandro Foi,et al.  Image Denoising by Sparse 3-D Transform-Domain Collaborative Filtering , 2007, IEEE Transactions on Image Processing.

[8]  Sundeep Rangan,et al.  Denoising based Vector Approximate Message Passing , 2016, ArXiv.

[9]  Mike E. Davies,et al.  Near Optimal Compressed Sensing Without Priors: Parametric SURE Approximate Message Passing , 2014, IEEE Transactions on Signal Processing.

[10]  Sundeep Rangan,et al.  Plug in estimation in high dimensional linear inverse problems a rigorous analysis , 2018, NeurIPS.