Work

A Simple Derivation of AMP and its State Evolution via First-Order Cancellation

Year: 2020
Type: conference-paper
Abstract: We consider the linear regression problem, where the goal is to recover the vector x ∈ ℝnfrom measurements y = Ax+w ∈ ℝmunder known matrix A and unknown noise w. For large i.i.d. sub-Gaussian A, the approximate message passing (AMP) algorithm is precisely analyzable through a state-evolution (SE) formalism, which furthermore shows that AMP is Bayes optimal in certain regimes. The rigorous SE proof, however, is long and complicated. And, although the AMP algorithm can be derived as an approximation of loop belief propagation (LBP), this viewpoint provides little insight into why large i.i.d. A matrices are important for AMP, and why AMP has a state evolution. In this work, we provide a heuristic derivation of AMP and its state evolution, based on the idea of "firstorder cancellation," that provides insights missing from the LBP derivation while being much shorter than the rigorous SE proof.
Language: English
Open Access status: green
OpenAlex ID: W2961549563
FWCI: 2.074
Cites: 18
Cited by: 12
Related to: 20