Optimization-based AMP for Phase Retrieval: The Impact of Initialization and $\ell_2$-regularization
Junjie Ma, Ji Xu, Arian Maleki
Introduction
2 Informal statement of our results
where is the th component of and a Gaussian noise. The recent surge of interest has led to a better understanding of the theoretical aspects of this problem. Thanks to such research we now have access to several algorithms, inspired by different ideas, that are theoretically guaranteed to recover exactly in the noiseless setting. Despite all this progress, there is still a gap between the theoretical understanding of the recovery algorithms and what practitioners would like to know. For instance, for many algorithms, including Wirtinger flow and amplitude flow , the exact recovery is guaranteed with either or measurements, where is often a fixed but large constant that does not depend on . In both cases, it is often claimed that the large value of or the existence of is an artifact of the proving technique and the algorithm is expected to work with for a reasonably small value of . Such claims have left many users wondering
Which algorithm should we use? Since the theoretical analyses are not sharp, they do not shed any light on the relative performance of different algorithms. Answering this question through simulations is very challenging too, since many factors including the distribution of the noise, the true signal , and the number of measurements may have impact on the answer.
When can we trust the performance of these algorithms in the presence of noise? Suppose for a moment that we know the minimum number of measurements that is required for the exact recovery through simulations. Should we collect the same number of measurements in the noisy settings too?
What is the impact of initialization schemes, such as spectral initialization? Can we trust these initialization schemes in the presence of noise? How should we compare different initialization schemes?
Researchers have developed certain intuition based on a combination of theoretical and empirical results, to give heuristic answers to these questions. However, as demonstrated in a series of papers in the context of compressed sensing, such folklores are sometimes inaccurate . To address Question Q.1, several researchers have adopted the asymptotic framework , , and provided sharp analyses for the performance of several algorithms . This line of work studies recovery algorithms that are based on convex optimization. In this paper, we adopt the same asymptotic framework and study the following popular non-convex problem, known as amplitude-based optimization :
Since (1.2) is a non-convex problem, the algorithm to solve it matters. In this paper, we study a message passing algorithm that aims to solve (1.2). As a result of our studies we
present sharp characterization of the mean square error (even the constants are sharp) in both noiseless and noisy settings.
present a quantitative characterization of the gain initialization and regularization can offer to our algorithms.
Furthermore, the sharpness of our results enables us to present a quantitative and accurate comparison with convex optimization based recovery algorithms and give partial answers to Question Q.1 mentioned above. Below we introduce our message passing algorithm and informally state some of our main results. The careful and accurate statements of our results are postponed to Section 2.
The first point that we would like to discuss here is the effect of the regularizer on AMP.A. For the moment suppose that the noise is zero. Does including the regularizer in (1.2) benefit AMP.A? Clearly, any regularization may introduce unnecessary bias to the solution. Hence, if the final goal is to obtain exactly we should set . However, the optimization problem in (1.2) is non-convex and iterative algorithms intended to solve it can get stuck at bad local minima. In this regard, regularization can still help AMP.A to escape bad local minima through continuation. Continuation is popular in convex optimization for improving the convergence rate of iterative algorithms , and has been applied to the phase retrieval problem in . In continuation we start with a value of for which AMP.A is capable of finding the global minimizer of (1.2). Then, once converges we will either decrease or increase a little bit (depending on the final value of for which we want to solve the problem) and use the previous fixed point of AMP.A as the initialization for the new AMP.A. We continue this process until we reach the value of we are interested in. For instance, if we would like to solve the noiseless phase retrieval problem then should eventually go to zero so that we do not introduce unnecessary bias. The rationale behind continuation is the following. Let and be two different values of the regularization parameter, and they are close to each other. Suppose that the global minimizer of (1.2) with regularization parameter is and is given to the user. Suppose further that the user would like to find the global minimizer of (1.2) with . Then, it is conceivable that the global minimizer of the new problem is close to .Given the sometimes complex geometry of non-convex problems, this might not always be the case. Hence, the user can initialize AMP.A with and hope that the algorithm may converge to the global minimizer of (1.2) for .
A more general version of the continuation idea we discussed above is to let change at every iteration (denoted as ), and set according to :
Below we informally discuss some of the results we will prove in this paper.
We would like to make the following remarks about these two results:
As is clear from our second informal result, when , cannot converge to . This value of is different from the information theoretic lower bound . This discrepancy is in fact due to the type of continuation we used in this paper. Note that this issue does not happen in the complex-valued . The search for a better continuation strategy for the real-valued is left as future research.
Simulation results presented in our forthcoming paper show that for real-valued signals, AMP.A with can only recover when . As mentioned in our second informal result, continuation has improved the threshold of correct recovery to .
Now let us discuss the performance of AMP.A under noisy settings. We assume that the measurement noise is Gaussian and small. Clearly, in this setting exact recovery is impossible, hence we study the asymptotic mean square error defined as the following almost sure limit ()
3 Related work
Early theoretical results on phase retrieval, such as PhaseLift and PhaseCut , are based on semidefinite relaxations. For random Gaussian measurements, a variant of PhaseLift can recover the signal exactly (up to global phase) in the noiseless setting using measurements . However, PhaseLift (or PhaseCut) involves solving a semidefinite programming (SDP) and is computationally prohibitive for large-scale applications. A different convex optimization approach for phase retrieval, which has the same sample complexity, was independently proposed in and . This method is formulated in the natural signal space and does not involve lifting, and is therefore computationally more attractive than SDP-based counterparts. However, both methods require an anchor vector that has non-zero correlation with the true signal, and the quality of the recovery highly depends on the quality of the anchor.
Apart from convex relaxation approaches, non-convex optimization approaches attract considerable recent interests. These algorithms typically consist of a carefully designed initialization step (usually accomplished via a spectral method ) followed by iterations that refine the estimate. An early work in this direction is the alternating minimization algorithm proposed in , which has sub-optimal sample complexity. Another line of work includes the Wirtinger flow algorithm , truncated Wirtinger flow algorithm , and other variants. Other approaches include Kaczmarz method , trust region method , coordinate decent , prox-linear algorithm and Polyak subgradient method .
All the above theoretical results guarantee successful recovery with measurements (or more) where is a fixed often large constant. However, such theories are not capable of providing fair comparison among different algorithms. To resolve this issue researchers have started studying the performance of different algorithms under the asymptotic setting and . An interesting iterative projection method was proposed in, whose dynamics can be characterized exactly under this asymptotic setting. However, does not analyze the number of measurements required for this algorithm to work. The work in provides sharp characterization of the spectral initialization step (which is a key ingredient to many of the above algorithms). The analysis in reveals a phase transition phenomenon: spectral method produces an estimate not orthogonal to the signal if and only if is larger than a threshold (called “weak threshold” in ). Later, derived the information-theoretically optimal weak threshold (which is for the real-valued model and for the complex-valued model) and proved that the optimal weak threshold can be achieved by an optimally-tuned spectral method. Using the non-rigorous replica method from statistical physics, analyzes the exact threshold of (for the real-value setting) above which the PhaseMax method in and achieves perfect recovery. The analysis in shows that the performance of PhaseMax highly depends on initialization (see Fig. 1 of ), and the required is lower bounded by for real-valued models. The analysis in was later rigorously proved in via the Gaussian min-max framework , and a new algorithm called PhaseLamp was proposed. The PhaseLamp method has superior recovery performance over PhaseMax, but again it does not work when for real-valued models. A recent paper extends the asymptotic analysis of to the complex-valued setting, and it was shown that PhaseMax cannot work for . On the other hand, AMP.A proposed in this paper achieves perfect recovery when and , for the real and complex-valued models respectively. Further, focus on the noiseless scenario, while in this paper we also analyze the noise sensitivity of AMP.A. Finally, a recent paper derived an upper bound of such that PhaseLift achieves perfect recovery. The exact value of this upper bound can be derived by solving a three-variable convex optimization problem and empirically shows that for real-valued models.
3.2 Existing work based on AMP
Our work in this paper is based on the approximate message passing (AMP) framework , in particular the generalized approximate message passing (GAMP) algorithm developed and analyzed in . A key property of AMP (including GAMP) is that its asymptotic behavior can be characterized exactly via the state evolution platform .
For phase retrieval, a Bayesian GAMP algorithm has been proposed in . However, did not provide rigorous performance analysis, partly due to the heuristic treatments used in the algorithm (such as damping and restart). Another work related to ours is the recent paper (appeared on Arxiv while we are preparing this paper), which analyzed the phase transitions of the Bayesian GAMP algorithms for a class of nonlinear acquisition models. For the phase retrieval problem, a phase transition diagram was shown in [43, Fig. 1] under a Bernoulli-Gaussian signal prior. The numerical results in indeed achieve state-of-the-art reconstruction results for real-valued models. However, did not provide the analysis of their results and in particular did not mention how they handle a difficulty related to initialization. Further, the algorithm in is based on the Bayesian framework which assumes that the signal and the measurements are generated according to some known distributions. Contrary to and , this paper considers a version of GAMP derived from solving the popular optimization problem (1.2). We provide rigorous performance analysis of our algorithm for both real and complex-valued models. Note that the advantages and disadvantages of Bayesian and optimization-based techniques have been a long debate in the field of Statistics. Hence, we do not repeat those debates here. Given our experience in the fields of compressed sensing and phase retrieval, it seems that the performance of Bayesian algorithms are more sensitive to their assumptions than the optimization-based schemes. Furthermore, performance analyses of Bayesian algorithms are often very challenging under “non-ideal” situations which the algorithms are not designed for.
Here, we emphasize another advantage of our approach. Given the fact that the most popular schemes in practice are iterative algorithms derived for solving non-convex optimization problems, the detailed analyses of presented in our paper may also shed light on the performance of these algorithms and suggest new ideas to improve their performances.
3.3 Fundamental limits
It the literature of phase retrieval, it is well known that to make the signal-to-observation mapping injective one needs at least measurements (or in the case of real-valued models). On the other hand, the measurement thresholds obtained in this paper are and respectively. In fact, our algorithm can in principal recover the signal when and (or if continuation is not applied) for complex and real-valued models, provided that the algorithm is initialized close enough to the signal (though no known initialization strategy can accomplish this goal). Hence, our threshold are even smaller than the injectivity bounds. We emphasize that this is possible since the injectivity bounds derived in are defined for all (which can depend on in the worst case scenario). This is different from our assumption that is independent of , which is more relevant in applications where one has some freedom to randomize the sampling mechanism. In fact, several papers have observed that their algorithm can operate at the injectivity thresholds for real-valued models . These two different notions of thresholds were discussed in . In the context of phase retrieval, the reader is referred to the recent paper , which showed that by solving a compression-based optimization problem, the required number of observations for recovery is essentially the information dimension of the signal (see for the precise definition). For instance, if the signal is -sparse and complex-valued, then measurements suffice.
4 Organization of the paper
The structure of the rest of the paper is as follows: Section 2 mentions the asymptotic framework of the paper, and summarizes our main results on the asymptotic analysis of . Section 3 discusses the real-valued algorithm and its analysis. Section 4 presents the proofs of our main results.
Asymptotic analysis of AMP.Aformulae-sequenceAMPA\rm AMP.A
In this section, we present the asymptotic platform under which is studied, and we derive a set of equations, known as state evolution (SE), that capture the performance of under the asymptotic analysis.
Our analysis of is carried out based on a standard asymptotic framework developed in . In this framework, we let , while . Within this section, we will write , , and as , , and to make explicit their dependency on the signal dimension . In this section we focus on the complex-valued AMP. We postpone the discussion of the real-valued AMP until Section 3. Following , we introduce the following definition of converging sequences.
The sequence of instances is said to be a converging sequence if the following hold:
, as .
has i.i.d. Gaussian entries where .
Under the asymptotic framework introduced above, the behavior of can be characterized exactly. Roughly speaking, the estimate produced by in each iteration is approximately distributed as the (scaled) true signal additive Gaussian noise; in other words, can be modeled as , where behaves like an iid standard complex normal noise. We will clarify this claim in Theorem 1 below. The scaling constant and the noise standard deviation evolve according to a known deterministic rule, called the state evolution (SE), defined below.
In the above equations, the expectations are over all random variables involved: , where is independent of , and where is independent of both and . Further, the partial Wirtinger derivative is defined as:
The functions and are well defined except when both and are zero.
Most of the analysis in this paper is concerned with the noiseless case. For brevity, we will often write (where ) as . Further, when our focus is on and rather than , we will simply write as .
In Appendix B.2, we simplify the functions and into the following expressions (with being the phase of ):
The above expressions for and are more convenient for our analysis.
The state evolution framework for generalized AMP (GAMP) algorithms was first introduced and analyzed in and later formally proved in . As we will show later in Theorem 1, SE characterizes the macroscopic behavior of . To apply the results in to AMP.A, however, we need two generalizations. First, we need to extend the results in to complex-valued models. This is straightforward by applying a complex-valued version of the conditioning lemma introduced in . Second, existing results in require the function to be smooth. Our simulation results in case of complex-valued show that SE predicts the performance of despite the fact that is not smooth. Since our paper is long, we postpone the proof of this claim to another paper. Instead we use the smoothing idea discussed in to connect the SE equations presented in (2.1) with the iterations of in (1.6). Let be a small fixed number. Consider the following smoothed version of :
Note that as , and hence we expect the iterations of smoothed- converge to the iterations of .
Let be a converging sequence of instances. For each instance, let be an initial estimate independent of . Assume that the following hold almost surely
Let be the estimate produced by the smoothed initialized by (which is independent of ) and . Let denote a sequence of smoothing parameters for which as Then, for any iteration , the following holds almost surely
where , and is independent of . Further, and are determined by (2.1) with initialization and .
The proof of Theorem 1 is given in Section 4.2.
2 Convergence of the SE for noiseless model
We now analyze the dynamical behavior of the SE. Before we proceed, we point out that in phase retrieval, one can only hope to recover the signal up to global phase ambiguity , for generic signals without any structure. In light of (2.5), AMP.A is successful if and as .
Let us start with the following interesting feature of the state evolution, which can be seen from (2.2).
.
Hence, if denotes the phase of , then .
In light of this lemma, we can focus on real and nonnegative values of . In particular, we assume that and we are interested in whether and under what conditions can the SE converge to the fixed point . The following two values of will play critical roles in the analysis of SE:
Notice that is essential for the success of AMP.A. This can be seen from the fact that is always a fixed point of for any . From our definition of in Theorem 1, is equivalent to . This means that the initial estimate cannot be orthogonal to the true signal vector , otherwise there is no hope to recover the signal no matter how large is.
3 Noise sensitivity
So far we have only discussed the performance of in the ideal setting where the noise is not present in the measurements. In general, one can use (2.1) to calculate the asymptotic MSE (AMSE) of as a function of the variance of the noise and . However, as our next theorem demonstrates it is possible to obtain an explicit and informative expression for AMSE of in the high signal-to-noise ratio (SNR) regime.
The proof of this theorem can be found in Appendix E.
Extension to real-valued signals
Until now our focus is on complex-valued signals. In this section, our goal is to extend our results to real-valued signals. Since most of the results are similar to the complex-valued case, we will skip the details and only emphasize on the main differences.
In the real-valued case, uses the following iterations:
2 Asymptotic Analysis
Our analysis is based on the same asymptotic framework detailed in Section 2.2. The only difference is that the measurement matrix is now real Gaussian with and . In the real-valued setting, the state evolution (SE) recursion of in (3.1) becomes the following.
The expectations are over the following random variables: , where is independent of , and where independent of both and .
In Appendix B.3, we derived the following closed-form expressions of and :
As in the complex-valued case, we would like to study the dynamics of these two equations. The following lemma simplifies the analysis.
and in (3.3) and (3.3b) have the following properties:
.
.
Again the following two values of play a critical role in the performance of AMP:
The following two theorems correspond to Theorems 2 and 3 that explain the dynamics of SE for complex-valued signals. The proofs can be found in Section D.1 and Section D.2 respectively.
Note that in Theorem 5 the sequences converge for any . This result is stronger than the complex-valued counterpart, which requires and (see Theorem 2).
Finally, we discuss the performance of in the high SNR regime. See Section E for its proof.
Proofs of our main results
The functions that we have in (2.1) are related to the first and second kinds of elliptic integrals. Below we review some of the properties of these functions that will be used throughout our paper. Elliptic integrals (elliptic integral of the second kind) were originally proposed for the study of the arc length of ellipsoids. Since their appearance, elliptic integrals have appeared in many problems in physics and chemistry, such as characterization of planetary orbits. Three types of elliptic integrals are of particular importance, since a large class of elliptic integrals can be reduced to these three. We introduce two of them that are of particular interest in our work.
The first and second kinds of complete elliptic integrals, denoted by and (for ) respectively, are defined as
In the above definitions, we continued to use , to follow the convention in the literature of elliptic integrals. Previously, was defined to be the number of measurements, but such abuse of notation should not cause confusion as the exact meaning of is usually clear from the context.
Below, we list some properties of elliptic integrals that will be used in this paper. The proofs of these properties can be found in standard references for elliptic integrals and thus omitted (e.g., ).
The following hold for and defined in (4.1):
. Further, for , and behave as
On , is strictly increasing, is strictly decreasing, and is strictly increasing.
The derivatives of , and are given by (for )
Furthermore, we will use a few more elliptic integrals in our work. Next lemma and its proof connects these elliptic integrals to Type I and Type II elliptic integrals.
The following equalities hold for any :
We will only prove (4.4b). (4.4a) can be proved in the same way. The idea is to express the integrals using elliptic integrals defined in (4.1), and then apply known properties of elliptic integrals (Lemma 3) to simplify the results. The same tricks in proving (4.4b) are used to derive other related integrals in this paper. Below, we will provide the full details for the proof of (4.4b), and will not repeat such calculations elsewhere. The LHS of (4.4b) can be rewritten as:
The equality in (4.5) can be proved by combining the following identities together with straightfroward manipulations:
where and denote the complete elliptic integrals of the first and second kinds (see (4.1)). First, consider the identity (i) in (4.6):
where (a) is from the definition of and in (4.1), and (b) is from Lemma 3 (iii).
where (a) is due to Lemma 3 (iv) and (b) is from Lemma 3 (iii).
where step (a) follows from the third step of (4.7), and step (b) follows from Lemma 3 (iii).
Identity (iv) can be proved in a similar way:
where (a) is from the third step of (4.8), step (b) is from Lemma 3 (iv) and (c) is from Lemma 3 (iii).
Lastly, identity (v) can be proved as follows:
where step (a) follows from the derivations of the previous two identities and (b) is again due to Lemma 3 (iii). ∎
2 Proof of Theorem 1
Since the proof of the real-valued and complex valued signals look similar, for the sake of notational simplicity we present the proof for the real-valued signals. First note that according to [19, Lemma 13]The proof for a more general result was first presented in . However, we found easier to follow. The reader may also find [26, Claim 1] and related discussions useful, although no formal proof was provided. for the smoothed algorithm we know that almost surely
where and is independent of , and and satisfy the following iterations:
where , , where is independent of and . It is also straightforward to use an induction step similar to the one presented in the proof of Theorem 1 of and show that as , where satisfy
3 Proof of Theorem 2
The goal of this section is to prove Theorem 2. However, since the proof is very long we start with the proof sketch to help the reader navigate through the complete proof.
Our main goal is to study the dynamics of the iterations:
Notice that according to the assumptions of the theorem, we assume that we initialized the dynamical system with . Our first hope is that this dynamical system will not oscillate and will converge to the solutions of the following system of nonlinear equations:
Hence, the first step is to characterize and understand the fixed points of the solutions of (4.10). Toward this goal we should study the properties of and . In particular, we would like to know how the fixed points of behave for a given and how the fixed points of behave for a given value of and . The graphs of these functions are shown in Figure 2.
We list some of the important properties of these two functions. We refer the reader to Section 4.3.2 to see more accurate statement of these claims.
is a concave and strictly increasing function of , for any : This implies that can have two fixed points: one at zero and one at . Also, as is clear from the figure, the second fixed point is the stable one.
For the moment assume that the unstable fixed points do not affect the dynamics of . Let denote the non-zero fixed point of and the stable fixed point of .In the literature of dynamical systems, these functions are sometimes called nullclines. Nullclines are useful for qualitatively analyzing local dynamical behavior of two-dimensional maps (which is the case for the SE in this paper). We will prove in Lemma 11 that is a decreasing function and hence is well-defined on . Moreover, we will show that by choosing , is continuous on . and are shown in Fig. 3. Note that the places these curves intersect correspond to the fixed points of (4.10). Depending on the value of the two curves show the following different behaviors:
You may find the proof of this lemma in Section 4.3.4. Intuitively speaking, in this case we expect the state evolution to converge to the fixed point , meaning that AMP.A achieves exact recovery.
So far, we have studied the solutions of (4.10). But the ultimate goal of analysis of is the analysis of (4.9). In particular, it is important to show that the estimates converge to and do not oscillate. Unfortunately, the dynamics of do not monotonically move toward the fixed point , which makes the analysis of SE complicated.
where .
From the above lemma, we see that to understand the dynamics of the SE, we only focus on the region \mathcal{R}\overset{\scriptscriptstyle\Delta}{=}\left\{(\alpha,\sigma^{2})\big{|}0<\alpha\leq 1,0<\sigma^{2}\leq\sigma^{2}_{\max}\right\}. Since the dynamic of is complicated, we divide this region into smaller regions. See Figure 4 for an illustration.
We divide \mathcal{R}\overset{\scriptscriptstyle\Delta}{=}\left\{(\alpha,\sigma^{2})\big{|}0<\alpha\leq 1,0<\sigma^{2}\leq\sigma^{2}_{\max}\right\} into the following three sub-regions:
Our next lemma shows that if is in or for , then converges to . The following lemma demonstrates this claim.
remains in for all ;
This claim will be proved in Section 4.3.5. Notice that the condition is important for part (i) to hold: if is close to the origin (and thus in ), then can move to . However, this cannot happen when . In the proof given in Section 4.3.5, we showed that for any the possible locations of are bounded from below by a curve, and once is above this curve and also in region or , then we will prove that it cannot go to . Finally, we will prove the following Lemma that completes the proof.
The proof of this result is in Section 4.3.6. Combining the above two lemmas, it is straightforward to see that , and hence the proof is complete.
In this section we derive all the main properties of and that are used throughout the paper.
has the following properties (for ):
is a concave and strictly increasing function of , for any given .
, for and .
If , then there are two nonnegative solutions to : and . Further, is strongly globally attracting, meaning that
On the other hand, if then is the unique nonnegative fixed point and it is strongly globally attracting.
Part (i): From (2.2), it is easy to verify that is an increasing function of . We now prove its concavity. To this end, we calculate its first and second partial derivatives:
Hence, is a concave function of for .
Part (ii): Positivity of is obvious. Also, note that
Proof of (iii): The claim is a consequence of the concavity of (with respect to ) and the following condition:
The detailed proof is as follows. First, it is straightforward to verify that is always a solution to . Define
Since is a concave function of (as is concave), is decreasing. Let’s first consider . In this case we know that
where the second equality can be calculated from (4.13a). Since is a decreasing function of and is equal to zero at zero, and it does not have any other solution. Now, consider case . It is straightforward to confirm that
Furthermore, from (4.13a) we have \frac{\partial\psi_{1}(\alpha,\sigma^{2})}{\partial\alpha}\Big{|}_{\alpha\to\infty}=0, and so
Hence, has exactly one more solution for . Note that since from part (ii) , the solution of also satisfies .
Finally, the strong global attractiveness follows from the fact that is a strictly increasing function of .
has the following properties:
If , then is a locally unstable fixed point to , meaning that
For any , has a unique fixed point in for any . Further, the fixed point is (weakly) globally attracting in :
where .
For any , is an increasing function of if
where is the unique solution to
First note that the partial derivative of w.r.t. is given by
Part (i): Before we proceed, we first comment on the discontinuity of the partial derivative at . Note that the formula in (4.18) was derived for non-zero values of . Naively, one may plug in in the equation and assume that \frac{\partial\psi_{2}(\alpha,\sigma^{2};\delta)}{\partial\sigma^{2}}\Big{|}_{\alpha=1,\sigma^{2}=0}=\frac{4}{\delta}. This is not the case since the integral is divergent. It turns out that the derivative is a continuous function of . The technical details can be found in Appendix C.
Since is continuous at , we have
Note that if we set , then from (4.6) we have
It is then straightforward to use Lemma 3 to prove that
Hence, \frac{\partial\psi_{2}(\alpha,\sigma^{2};\delta)}{\partial\sigma^{2}}\Big{|}_{\alpha=1,\sigma^{2}=0}>1 for .
Part (ii): We first prove that the following equation has at least one solution for any and :
We next prove our claim by proving the following:
We next show that in (4.21) is a concave function of , and hence the minimum can only happen at either or . The first two derivatives w.r.t. are given by:
The concavity of implies that its minimum happens at either or . Hence, to prove (4.21), it suffices to prove that
which holds for . Hence, (4.21) holds. By combining (4.19) and (4.20) we conclude that has at least one fixed point between and . The next step is to prove the uniqueness of this fixed point. For the rest of the proof, we discuss two cases separately: a) and b) .
From (4.18), if , then , . This means that defined in (4.22) is monotonically decreasing in . Hence, the solution to is unique. Furthermore, the following property is a direct consequence of the monotonicity of :
where denotes the solution to .
. In this case, we will prove that there exists a threshold on , denoted as below, such that the following hold:
This means that is strictly decreasing on and increasing on . Note that since we have proved that has at least one solution, we conclude that there exist exactly two solutions to , one in and the second in , if . This is the case since (see (4.20)), and that (since the latter is the global minimum of in ).
Also, it is easy to prove (4.23). In fact, the following holds:
where denotes the larger solution to . See Fig. 5 for an illustration.
From the above discussions, it remains to prove (4.24). To this end, it is more convenient to express (4.18) using elliptic integrals discussed in Section 4.1:
where we introduced a new variable and the last step is derived using the identities in Lemma 4. Based on (4.25) we can now rewrite (4.24) as
To prove this, we first show that there exists such that is strictly increasing on and decreasing on , namely,
This can be seen from derived below:
Further noting that is strictly decreasing in while is increasing, we proved (4.27).
Based on the above discussions, we can finally turn to the proof of (4.26). From (4.25b), it is straightforward to verify that . Therefore, when , we have
Hence, the following equation admits a unique solution (denoted as below):
See Fig. 6 for an illustration. Also, from our above discussions on the monotonicity of it is straightforward to show that
which proves (4.26) by setting . This proves (4.24), which completes the proof.
Part (iii): We will prove a stronger result: . From (2.2b), is equivalent to
For and we have
which, similar to (4.29), can be proved by
Part (iv): We bound the partial derivative of for as:
Hence, there exists a unique solution (which we denote as ) to the following equation:
Finally, the property in (4.16) is a direct consequence of the fact that is a decreasing function of .
Part (v): In (4.25), we have derived the following:
where . From (4.25b), we see that is an increasing function of if the following holds:
Further, (4.27) implies that the maximum of happens at , i.e.,
where is the unique solution to
Clearly, immediately implies , which further guarantees that is monotonically increasing on . Finally, the strong global attractiveness of is a direct consequence of part (iv) of this lemma together with the monotonicity of . ∎
In this section we derive the main properties of the functions and introduced in Section 4.3.1. These properties play major roles in the results of the paper.
The following hold for and (for ):
and . Further, by choosing , we have is continuous on and strictly decreasing in ;
is a continuous function of and . , and for and .
Part (i): We first verify and . First, can be seen from the following facts: (a) for , see (2.2a); and (b) By definition, is the non-zero solution to . Then, by Lemma 9 (iii) and continunity of , we know is continuous on , and further since corresponds to a case where the non-negative solution to decreases to zero. Next, we prove the monotonicity of . Note that
Differentiation w.r.t. yields
where \partial_{2}\psi_{1}(F_{1}(\sigma^{2}),\sigma^{2})\overset{\scriptscriptstyle\Delta}{=}\frac{\partial\psi_{1}(\alpha,\sigma^{2})}{\partial\sigma^{2}}\Big{|}_{\alpha=F_{1}(\sigma^{2})} and \partial_{1}\psi_{1}(F_{1}(\sigma^{2}),\sigma^{2})\overset{\scriptscriptstyle\Delta}{=}\frac{\partial\psi_{1}(\alpha,\sigma^{2})}{\partial\alpha}\Big{|}_{\alpha=F_{1}(\sigma^{2})}. Hence,
We have proved in (4.14) that \frac{\partial\psi_{1}(\alpha,\sigma^{2})}{\partial\alpha}\Big{|}_{\alpha=0}<1 when . Together with the concavity of w.r.t. (cf. Lemma 9 (i)), we have
Further, from (2.2a), it is straightforward to see that is a strictly decreasing function of , and thus
Substituting (4.34) and (4.35) into (4.33), we obtain
Proof of (ii): By Lemma 10 (ii) and continuity of , it is straightforward to check that is continuous. Moreover, we have proved that is the unique solution to the following equation (for ):
which has two possible solutions (for ):
(For the special case , .) However, is invalid due to our constraint . This can be seen as follows. First, for and hence invalid. When , we have
Hence, . When , (4.36) becomes:
It is straightforward to verify that is a solution. Also, from Lemma 10 (ii), is a also the unique solution. Hence, .
3.4 Proof of Lemma 5
In Lemma 10, we have proved that is the unique globally attracting fixed point of in (for ), and from (4.15) we have
We now make some variable changes for (4.39). From (2.2a), in can be rewritten as the following for :
By definition, is the solution to , and hence the following holds:
At this point, it is more convenient to make the following variable change:
where the first equality is from (4.40) and the second step from (4.41). Using the relationship in (4.42), we can reformulate the inequality in (4.39) into the following equivalent form:
Substituting (4.41) and (2.2b) into (4.43) and after some manipulations, we can finally write our objective as:
In the next two subsections, we prove (4.44) for and using different techniques.
Using the variable , we can rewrite (4.44) into the following:
Notice that if we could prove (4.46a) for , we would have proved (4.44) for , since . For the ease of later discussions, we define
The following identities related to will be used in our proof:
We now prove (4.46a). First, it is straightforward to verify that equality holds for (4.46a) at , i.e.,
Hence, to prove that for , it is sufficient to prove that is an increasing function of on . To this end, we calculate the derivative of :
where step (a) follows from the identities listed in (4.47). Since , we have
It remains to prove that for . Our numerical results suggest that is a monotonically decreasing function for , and as However, directly proving the monotonicity of seems to be quite complicated. We use a different strategy here. We will prove that (at the end of this section)
As a consequence, the following hold true for any :
Hence, if we verify that , we will be proving the following:
To this end, we verify that hold for a sequence of and : , , , , , , , , , , , , . Combining all the above results proves
From the above discussions, it only remains to prove the monotonicity of and . Consider first:
Applying the Cauchy-Schwarz inequality yields:
Combining (4.49) and (4.50), we proved that , and therefore is monotonically increasing. For , we have
Combining the previous two equations leads to , which completes our proof.
Case II: We next prove (4.44) for , which is based on a different strategy. Some manipulations of the RHS of (4.44) yields:
From our reformulation in (4.51), the inequality in (4.44) for becomes
Note that and thus proving the above inequality for is sufficient to prove the original inequality for (note that , see (4.51b)).
With some further calculations, (4.52) can be reformulated as
The following inequality is due to [51, Eqn. (1)]
and to prove (4.53) it suffices to prove the following
To this end, we will prove that the LHS of (4.54) is a strictly increasing function of . If this is true, we would have
We next prove the monotonicity of . From the identities in Lemma 3, we derive the following
Hence, to prove that is monotonically increasing, it is sufficient to prove the following inequality:
Now, substituting into (4.55) and after some manipulations, we finally reformulate the inequality to be proved into the following form:
It can be verified that equality holds at . We next prove that is monotonically decreasing on . We differentiate once more:
Our problem boils down to proving for . We can verify that holds at . We finish by showing that is monotonically increasing in . To this end, we differentiate again:
We note that is a monotonically increasing function in (0,1) since is monotonically increasing and is monotonically decreasing. We verify that when . Hence,
Substituting (4.57) into (4.56), we prove that for , which completes the proof.
3.5 Proof of Lemma 7
First, we introduce a function that will be crucial for our proof.
where is the inverse functions of . The existence of follows from its monotonicity, which can be seen from its definition.
In the following, we list some preliminary properties of . The main proof for Lemma 7 comes afterwards.
The following lemma helps us clarify the importance of in the analysis of the dynamics of SE:
For any , and , the following holds:
where and are the SE maps defined in (2.2), and is defined in (4.58).
Define . Let be the image of under the SE map in (2.2). We will prove that the following holds for an arbitrary :
where satisfies the constraint
If (4.61) holds, we would have proved (4.60). To see this, consider arbitrary such that . Then, we have
where step (a) follows from (4.61) and , and step (b) holds since the choice and is feasible for the constraint . This is precisely (4.60).
Furthermore, from the definition of in (4.59a) we have
Similarly, from (2.2b), i.e. the definition of , and the definition of in (4.59b), we can express as
From (4.62), we see that fixing is equivalent to fixing . Further, for a fixed , is a quadratic function of , and the minimum happens at
and is
where the last step is from the definition of is (4.58). This completes the proof. ∎
To understand the implication of this lemma, let us consider the iteration of the SE:
Note that according to Lemma 12, no matter where is, will fall above the curve. This function is a key component in the dynamics of . Before we proceed further we discuss two main properties of the function .
is a strictly decreasing function of .
Recall from (4.58) that is defined as
From (4.59a), it is easy to see that is a decreasing function. Hence, to prove that is a decreasing function of , it suffices to prove that is strictly decreasing.
where step (a) is obtained through similar calculations as those in (4.6), and in the last step we defined . Hence, to prove that is a decreasing function of , it suffices to prove that is an increasing function of . Further, (form the definition of in (4.1)), our problem reduces to proving that is increasing. To this end, differentiation yields
where (a) is from the differentiation identities in Lemma 3, (b) is from (4.1), and follows from Lemma 3 (ii) together with the fact that . ∎
The next lemma compares the function with .
We prove by contradiction. Suppose that at some . If this is the case, then there exists a such that
Since is a decreasing function (see Lemma 11), the first inequality implies that \hat{\alpha}{\color[rgb]{1,0,0}\geq}F_{1}(\hat{\sigma}^{2}). Then, based on the global attractiveness property in Lemma 9 (iii), we have
where step (a) follows from the global attractiveness property in Lemma 10 (iv), step (b) is due to the hypothesis in (4.64), step (c) is from (4.65) together with the monotonicity of (see Lemma 13). Note that (4.66) shows that , which contradicts Lemma 12, where we proved that for any , and . Hence, we must have that for any . ∎
The following holds for any and ,
where is defined in (4.58).
From (4.58), proving (4.67) is equivalent to proving:
where and are defined as (see (4.59a) and (4.59b)):
Simple calculations show that (4.68) can be reformulated as the following
To this end, we can write the LHS of (4.72) into a quadratic form of :
Hence, to prove that this quadratic form is negative everywhere, it suffices to prove that the discriminant is negative, i.e.,
Hence, the result of Lemma 16 can be reformulated as proving the following:
In Lemma 15, we proved that the following holds for any :
We prove that is monotonically decreasing on for .
We prove that the following holds for :
Clearly, (4.73) follows from the above claims. Here, we introduce the function since has a simple closed-form formula and is easier to manipulate than . We next prove step (ii). From (4.27), it suffices to prove that
where and are defined in (4.32) and (4.31) respectively. To this end, we note that the following holds for :
where the inequality follows from the fact that in (4.74) is strictly decreasing in , and the last step is calculated from (4.74) and . Finally, numerical evaluation of (4.32) shows that . Hence, , which completes the proof.
We next prove step (iii). First, simple manipulations yields
where (a) is from the definition of in (4.75) and (b) is due to (4.74). Using (4.76), we further obtain
We prove (4.78) by showing that the following stronger result holds:
For convenience, we make a variable change:
With some straightforward calculations, we can rewrite (4.79) as
The following upper bound on is due to [52, Eqn. (1.2)]:
For any , is a strictly decreasing function of , where is defined in (4.58).
From the definition of in (4.58), we can write
where (note that is not the conjugate of )
A key observation here is that does not depend on . Clearly, Lemma 17 is implied by the following stronger result:
which we will prove in the sequel. For convenience, we define
where the last equality is from the definition of in (2.2b). It remains to prove that is an increasing function of . The partial derivative of w.r.t. is given by
where in step (a) we used the relationship (see (4.80)), and step (b) is from the identities in (4.6). From (4.81), we see that is a quadratic function of . Therefore, to prove , it suffices to show that the discriminant is negative:
Further, to prove (4.82), it is sufficient to prove that the following two inequalities hold:
We first prove (4.83a). It is sufficient to prove the following
Applying a variable change , we can rewrite (4.84) as
where the last equality is from the definition of in (4.1).
We next prove (4.83b). Again, applying the variable change and after some straightforward manipulations, we can rewrite (4.83b) as
Hence, we only need to prove for . First, we note that , from the fact that and (see Lemma 3 (i)). We finish the proof by showing that is strictly increasing in . Using the identities in (4.3), we can obtain
To prove , it is equivalent to prove
Hence, to prove (4.85), it suffices to prove
We now return to the main proof for Lemma 7. Notice that by Lemma 12, cannot fall below the curve for . Hence, for , we can focus on the region above (including ), which we denote as . See Fig. 7 for illustration.
We will first prove that if , then the next iterates and satisfy the following:
where and are defined as
Note that when is on (i.e., ), equalities in (4.87a) and (4.87b) can be achieved. Further, this is the only case when either of the equality is achieved. Also, it is easy to see that if is on , then cannot be on .
Since separates and , (4.88) can also be written as
As a concrete example, consider the situation shown in Fig. 7. In this case, for both point A and point B, and are given by the two dashed lines. This directly follows from (4.89) by noting that point A is in region and point B is in region . Let be a shorhand for . To prove the strict inequality in (4.87), we deal with and separately.
Assume that . Using (4.89), the inequality in (4.87) can be rewritten as
Since , we have . Then, applying (4.12) proves . Further, using Lemma 5, we have . Also, Lemma 6 guarantees that . Hence, and applying Lemma 10 (iv) yields .
We now consider the case where . Similar to (4.90), we need to prove
The inequality can be proved by the global attractiveness in Lemma 9 (iii) and the fact that when . The proof for is considerably more complicated and is detailed in Lemma 18 below.
where is the SE map in (2.2b) and is the inverse of defined in Lemma 9.
The following holds when :
We next prove (4.94). We consider the three different cases:
and .
Therefore, proving (4.98) reduces to proving
Finally, (4.95) follows from the global attractiveness property in Lemma 10 (iv) and the inequality in Lemma 5.
In the sequel, we first use (4.96) to prove (4.94), and the proof for (4.96) will come at the end of this proof.
and thus proving (4.98) reduces to proving
which follows from the same argument as that for (4.95).
Case (iii): Lemma 10 (iii) shows that for any . It is easy to see that , and thus
Further, Lemma 11 shows that is monotonically decreasing. Hence,
where the numerical constant is calculated from the closed form formula (see (4.42)) and (from (4.17)). Comparing (4.99) and (4.100) shows that (4.94) holds in this case.
It only remains to prove (4.96). We have shown in (4.25) that
where . Further, we have proved in (4.27) that is strictly increasing on and strictly decreasing on , where is defined in (4.32). Hence, when , there exist two solutions to
denoted as and , respectively. Also, from (4.101) and noting the definition , we have
where and . Hence, for fixed where , is a local maximum of and is a local minimum. Clearly, if
then the maximum of over can only happen at either or , which will prove (4.96). Further, for the degenerate case , only has a local minimum, and it is easy to see that (4.96) also holds. Thus, we only need to prove that (4.102) holds when . This can be proved as follows:
where (a) is from the fact that and (b) is from our assumption . On the other hand, since is a decreasing function of (see Lemma 13), and thus for we have
where the last step is from Definition 4.58. Based on (4.103) and (4.104), we see that for if
where the numerical constant is calculated based on the definition of in (4.31), the definition of in (4.32), and that of and in Definition 4.58. Hence, the condition is enough for our purpose. This concludes our proof. ∎
Now we turn our attention to the proof of part (i) of Lemma 7. Suppose that . Then, using (4.87) and based on the fact that is a strictly decreasing function, we know that . (See Definition 5.) Further, Lemma 8 shows that . Hence, . Applying this argument recursively shows that if , then for all . An illustration of the situation is shown in Fig. 7.
where and are defined in (4.88). Note that the definitions of and require , and such requirement is satisfied here due to part (i) of this lemma. Noting the SE update and , and recall the inequalities in (4.87), we obtain the following:
which together with (4.106), and the fact that and (since ), leads to the results we want to prove:
To prove (4.108), we only need to prove the following (based on the definition in (4.105))
where and are shorthands for and . From (4.88), the above inequalities are equivalent to
Note that (4.87) already proves the following
Hence, to prove (4.109) and (4.110), we only need to prove
To prove , we note that
where (a) is from (4.87b), (b) is from (4.88), and (c) is due to the fact that is strictly decreasing, and (d) from (4.87). Hence, since is strictly decreasing, we have
Further, it is straightforward to see that if both inequalities are strict in (4.87) then
This shows that equalities of (4.108) hold only when the equalities in (4.87) hold.
The proof for is similar and omitted.
3.6 Proof of Lemma 8
Suppose that . From Definition 5, we have
where the last inequality is due to Lemma 5. Combining (4.111) and (4.112) yields
By the global attractiveness property in Lemma 10 (iv), (4.113) implies
From the above analysis, we see that as long as (and also ), will be strictly smaller than :
Hence, there exists a finite number such that
Otherwise, will converge to a in . This implies that is a fixed point of for certain value of . However, we know from part (i) of Lemma 11 and Lemma 5 that this cannot happen.
Based on a similar argument, we also have and so for . Further, we can show that (i.e., ) for all . First, follows from our assumption. Further, from (2.2a) we see that if . Then, using a simple induction argument we prove that for all . Putting things together, we showed that there exists a finite number such that
(Recall that we have proved in Lemma 6 that .) From Definition 5, .
4 Proof of Theorem 3
where and (see (4.41) for the definition of ). Again, it is more convenient to express (4.115) using elliptic integrals (cf. (4.52))
where we made a variable change . To this end, we can verify that
where the last step is due to the facts that and . See Section 4.1 for more details. Hence, the above derivative is negative if or by noting the definition . ∎
We continue to prove the local convergence of the state evolution. We divide the region into the following sub-regions:
Similar to the proof of Lemma 7 discussed in Section 4.3.5, we will show that if then the new states can be bounded as follows:
Based on the strong global attractiveness of (Lemma 9-iii) and (Lemma 10-v) and the additional result (4.15), it is straightforward to show the following:
where . Hence, we have (note that )
Further, as discussed in the proof of Lemma 10-(i), is a continuous function of . Hence, there exists such that
and it is easy to see that is an increasing function of . Hence, together with (4.120) we get the following
which means that is a strictly increasing function of for . Hence,
This implies that moves away from in a neighborhood of the fixed point .
References
Appendix A Derivations of AMP.A
For the convenience of the readers (especially those who are not familiar with AMP), we provide a sketch of the derivations of the AMP.A algorithm in this appendix. Our derivations follow the approach proposed in . However, there are some differences specially in the last steps of our derivation.
For simplicity, we focus on the real-valued case. Consider the following optimization problem:
where is a penalization parameter. We now sketch the derivations of the AMP.A algorithm intended for solving (A.1). First, we construct the following joint pdf for (A.1):
where is a normalizing constant, and denote the -th entries of and , and is parameter (the inverse temperature). Define
Following [53, Chapter 5], we proceed in three steps:
Derive the sum-product belief propagation (BP) algorithm for (A.2).
Find the message update rules in the limit of .
The above procedure is slightly different from the original derivations in (which is derived directly from the max-sum belief propagation algorithm) but equivalent. The sum-product BP algorithm reads
We next simplify the above BP update rules.
Based on this approximation, the message in (A.4a) can be expressed as follows
Using this definition, we can write in (A.6) as
Noting , following we apply a second order Taylor expansion to (amounts to a Gaussian approximation of ) :
where we have omitted constant terms (relative to ), and , and are short-hands for
A.2 Messages from variable nodes to factor nodes
From the Gaussian approximation in (A.8), is also Gaussian. Consider the following term:
where the second approximation comes from (A.8). Comparing (A.10) with the exponent of a Gaussian pdf, we find that its variance (which we denote by ) and mean are respectively given by
A.3 From BP to AMP
We assume that the message has the following structure [53, Chapter 5.2.4]:
where and . From (A.12), we can identify and (which is the term that depends on the index ) to be the following
We further simplify (i.e., the first term in the above equation) as follows
The approximation error in the above is since
where we used in the previous equation. Ignoring the term, the update in (A.14) becomes
We now return to the update of defined in (A.5):
where step (a) is due to (A.13) and step (b) is from the definition in (A.5).
A.4 Large β𝛽\beta Limit
Putting (A.5), (A.16), (A.11), (A.15), we obtain the following simplified BP update rules ( and ):
where and are shorthands for and respectively. The algorithm summarized above is a special form the generalized AMP (GAMP) algorithm derived in (see Algorithm 1).
We further approximate the variance updates in (A.17a) and (A.17c) by averaging over (based on some heuristic concentration arguments). After this approximation, becomes invariant to the index (denoted as below). We can then write (A.17) into the following vector form:
We next consider the zero-temperature limit, i.e., . From the definition of in (A.7), it can be verified that :
which has the following closed-form expression (for ):
A.5 Summary of AMP.A
After some algebra, we can finally express (A.18) using (instead of , see (A.20)) as the following:
There are a couple of points we want to emphasize:
When , the update of and are independent of the parameter . This is why we prefer to use instead of , see (A.20).
A.6 Heuristic derivations of the state evolution
According to (1.6), the complex-valued version of AMP.A proceeds as follows
Suppose that at each iteration the elements of are distributed as
where represents the th entry of the true signal vector and is independent of . Rigorous proof of the state evolution framework is based on the conditioning technique developed in . Here, our goal is show the reader how to heuristically derive the state evolution (SE) recursion, namely, given and , how to derive and . Following , we make the following heuristic assumptions to derive the SE:
We ignore the Onsager correction term, i.e., we assume that is generated as (cf. (1.6)):
We assume that is independent of .
We derive and separately in the following two subsections.
𝑡1\alpha_{t+1} To derive , we will calculate the expectation of the term in (A.22a) by treating and as constants. In other words, the expectations in this section are conditioned on and . We now consider the expectation of a single entry in :
where the last step is from Stein’s lemma (for complex Gaussian random variables) [54, Lemma 2.3], and and are defined as
where and are the phases of and respectively. Note that in rigorous calculations we should be careful about the discontinuity of . In this heuristic calculations we have ignored this issue. We will discuss this issue in our forthcoming paper . Substituting (A.24) into (A.22a) yields
Finally, when and are independent of , and by central limit theorem we can assume that both and are Gaussian, and their joint distribution is specified by the relationship where and are independent.
𝑡1\sigma^{2}_{t+1} From (A.23), can be derived as
where and are shorthands for and respectively, and step (a) follows from the heuristic assumption that the correlation between and , and the correlation between and can be ignored. Hence, combining (A.27) and (A.28) we obtain
where as argued below (A.26) the joint distribution of and are specified by where and are independent.
Appendix B Simplifications of SE maps
Here we collect some auxiliary results that will be used in the simplification of the state evolution equation.
where step (a) is from the integral ( denotes the CDF of the standard Gaussian distribution):
step (b) is from the variable change , step (c) is from the fact that , and step (d) is from
The identity in (B.1b) can be derived based on similar calculations:
where and are, respectively, PDF and CDF functions of the standard Gaussian distribution.
where (a) is from the symmetry of and (b) from the definition . Further,
where the last equality is from (B.4). Hence,
Finally, the third identity in (B.3) can be derived as follows:
where (a) is from the identity and (b) from our previously derived identities in (B.4) and (B.6). ∎
B.2 Complex-valued AMP.Aformulae-sequenceAMPA\rm AMP.A
From Definition 2, the SE equations are given by
In the above, , where is independent of , and where independent of both and . We first consider a special case (). When , we have , and therefore
We next turn to the general case where . Later, we will see that our formulas derived for positive covers the special case as well. Lemma 22 can simplify our derivations.
.
Note that Lemma 22 also holds for if we define .
In the following, we will derive and for the case where is real and nonnegative. The results for complex-valued can be easily derived from those for nonnegative , based on Lemma 22.
where the last step follow the following two identities together with some straightforward manipulations:
The above identities are proved in Lemma 20 in Appendix B.1. Using (B.9) and noting that , we further average our result over :
We next derive . From (B.8), we have
where in the last step we used the following indentity
where (a) is from (B.13), and the derivations of step (b) is more involved and are given in Lemma 4.
B.3 Real-valued AMP.Aformulae-sequenceAMPA\rm AMP.A
For the real-valued case, the SE maps are given by
where (a) is derived using the identities in (B.3). Finally, combining (B.15), (B.17) and (B.18), and after some calculations, we finally obtain the following
Our goal in this section is to show that . From the definition of the partial derivative, we have
To obtain Equality (a) we have used (4.6). By employing Lemma 3 (i) we have
Again we emphasize that we have also shown in the proof of Lemma 10 that . Hence, is continuous at .
Appendix D Asymptotic analysis of real-valued AMP.Aformulae-sequenceAMPA\rm AMP.A
The proof of Theorem 5 is in parallel to that for Theorem 2. For this reason, we will only report the discrepancies. For intuition and more discussions, please refer to Section 4.3.
Again, we define to be the non-negative fixed point of and to be the fixed point of , where and are now defined in (3.3). Different from the complex-valued case, now has a unique fixed point. Properties of and are detailed in Section D.1.2. Similar to complex-valued case, and satisfy the following property:
This lemma is a direct consequence of Lemma 29-ii proved in Section D.1.2. Hence, we skip its proof. Similar to (4.58), the following function characterizes the lower boundary of the region that () can fall into.
For any and , define
For the intuition about the reader may refer to Section 4.3. As in the complex-valued signals case, the following properties of this function play critical roles in the dynamics of the SE:
defined in (D.1) is a strictly decreasing function of .
This is straightforward to see and hence the proof is skipped.
We skip the proofs of this Lemma. The arguments are similar to Lemma 5 and the calculations are straightforward too. Similar to Definition 8, we divide into four subregions.
We divide into the following four sub-regions:
Note that there are two differences between Definition 8 and Definition 5. First, the upper limit of for is changed from to . Second, in Definition 5, for , but in Definition 8, the value of for is not upper bounded. Our next lemma shows that for any , the states of the dynamical system (3.2) will eventually move to or .
Starting from , cannot be in for any and .
Let be an arbitrary point in . Then, there exists a finite number such that .
The proof of Lemma 27 is very similar to that of Lemma 8 and therefore skipped here. Finally, we complete the proof by proving the following lemma.
remains in for all ;
The proof of this lemma is presented in Section D.1.5.
In this section, we discuss several properties of and .
in (3.3a) has the following properties (for ):
is a concave and strictly increasing function of , for any given .
, for and .
If , then there are two nonnegative solutions to : and . Further, is strongly globally attracting. On the other hand, if then is the unique nonnegative fixed point and it is strongly globally attracting.
The proof strategy is similar to the one given in Section 4.3.2. Also, the calculations are straightforward. Hence, to save some space we skip the proof of this lemma. ∎
has the following properties:
If , then is a locally unstable fixed point to for any , meaning that
For any , has a unique fixed point, denoted as , in for any . Further, the fixed point is weakly globally attracting in .
For any , is an increasing function of if
Further, in this case is strongly globally attracting in .
Recall from (3.3b) that is defined as
Proof of (i): The partial derivative of w.r.t. is
The claims follows from the following fact:
Proof of (ii): From (D.20), we see that the following holds for any and :
Further, using similar arguments as those in the proof of Lemma 10, we can prove that is globally attracting in .
Proof of (iii): When is an increasing function of in , we have
It is easy to show that the maximum of the RHS over is . Hence, is a strictly increasing function of in if . ∎
In this section we derive the main properties of the functions and .
The following hold for and (for ):
and . Further, by defining , we have is continuous on and strictly decreasing in ;
and .
The proof is similar to the proof of Lemma 11. ∎
D.1.4 Proof of Lemma 23
where is defined as (with some abuse of notations)
Based on this re-parameterization, (D.5) becomes
Substituting the definition of in (3.3b) into (D.7) and after some straightforward calculations, it can be shown that (D.7) is implied by the following:
Similar to the treatment in Section 4.3.4, we consider two different cases: (1) and (2) .
Case I: . From (D.11), is a strictly increasing function of , and thus is strictly decreasing.
We next show that is an increasing function of . The derivative of is given by:
The following proof is based on the idea introduced in Section 4.3.4: since is an increasing function and is a decreasing function, the following holds for any :
We verified that holds for a sequence of intervals: , , , , , . Altogether, we proved for .
Case II: . From the definitions in (D.12), (D.10) and (D.11), and based on some calculations not shown here, we write the LHS of (D.12) as
Hence, to prove for , it suffices to show that in (D.13) is strictly decreasing on . To this end, we calculate below:
Hence, to prove for , we only need to prove
which, after some straightforward manipulations, reduces to
We can verify that the above inequality holds for . We complete our proof by showing that the LHS of the above inequality is decreasing in :
where and , and the last inequality can be easily proved since is a strictly decreasing function of and .
D.1.5 Proof of Lemma 28
For any and , satisfies
Then, the inequality is equivalent to
which is clear from the Cauchy-Schwartz Inequality. ∎
In Lemma 30, we proved that is strictly increasing on for . Hence, we only need to consider the case . From the expression of in (3.3), it is straightforward to see that is increasing on (for ), where
Lemma 32 shows that is a lower bound of for any . Hence, it suffices to prove that
which holds since the LHS is lower bounded by while the RHS is upper bounded by . ∎
is a decreasing function of for any .
Note that we can represent as , where is a number that does not depend on . Hence, we will prove that is a decreasing function of for any fixed and . From the definition of in (3.3b), we have
We then calculate the derivative of w.r.t. :
where in the last step we defined . It suffices to prove that
We prove by showing that the discriminant of the above quadratic function (of ) is negative:
We next prove that the following two inequalities hold:
Hence, to prove (D.19), it is sufficient to prove
which holds since (i) LHS is an increasing function of while the RHS is a decreasing function, and (ii) equality holds when .
The proof is similar to that of Lemma 18. We consider three different cases:
and ,
where .
Case (i): In Lemma 30, we proved that is strictly increasing on for . Since in , the proof of
on reduces to the proof of
The last equality is clear from the global attractiveness of in that is proved in Lemma 30-ii and the fact that that is proved in Lemma 23.
Hence, has two stationary points if :
where is a local maximum and is a local minimum. Then, the maximum of over can only happen at either or if the following holds:
Since is a decreasing function of (which can be confirmed with a straightforward calculation of the derivative), then the following holds for :
Further, is an increasing function of and is upper bounded by
Hence, when
Now, suppose that . Then, proving that is equivalent to proving:
The rest of the argument is similar to the ones used in the proof of Lemma 18. Since according to Lemma 34 is a decreasing function of , and trivially is a decreasing function of we need to prove that
Also, since according to Lemma 33, we have , (D.21) simplifies to:
which is a simple implication of the global attractiveness of in that is proved in Lemma 30-ii.
Further, if the following holds for we would have proved that when :
Noting that for . Comparing this result with (D.22) proves that
where the last inequality is due to the fact that and hence . ∎
Main part The proof is similar to that of Lemma 7. The only noticeable difference is the proof for the following inequality (cf. (4.91))
where is now defined in Definition 8. We have dedicated Lemma 35 to the proof of the above inequality, which is in parallel to Lemma 18 for the complex-valued case.
D.2 Proof of Theorem 6
The proof is similar to that of Lemma 3. Hence, we only focus on the discrepancies.
Taylor expansions of the LHS and the RHS are respectively given by
and in this case there exists a constant such that
Since the rest of the proof is exactly similar to the proof of Lemma 3 for the sake of brevity we skip it here.
It is straightforward to use an argument similar to the one presented in Section 4.4.2 and show that there exists a neighborhood of in which . Hence, the state evolution moves away from .
Appendix E Proofs of Theorems 4 and 7
In light of Lemma 1, we assume that throughout this Appendix.
In the noisy setting the dynamic of SE becomes more challenging. In fact can move in any direction around the fixed point. That makes the proof of convergence of more complicated.
In the noiseless setting the location of the fixed point of SE was . This is not the case for the noisy settings where the location of the fixed point depends on the noise variance.
In the sections below we go over the entire proof, but will skip the parts that are similar to the proof of the noiseless setting which was discussed in Section 4.3.
E.2 Complex-valued case
In the noisy setting, remains unchanged, and is replaced by below:
Before we proceed to the analysis of and , we list a few identities for and which will be used in our proofs later.
and satisfy the following properties:
where and are shorthands for and respectively in the last two identities.
The proof of this lemma is a simple application of the identities we derived in Section 4.1, and is hence skipped.
Our next lemma summarizes the main properties of and in the noisy phase retrieval problem.
The equation has a unique nonzero solution in . Let be that unique solution. Then, for and for .
There exists , such that is strictly decreasing on and strictly increasing on . Further, .
Define , where is defined in (4.58). Then, for all , where is defined in (4.17).
For any and , we have .
.
In the following, we will prove that each part of the lemma holds when is smaller than a constant. Hence, the statements hold simultaneously when is smaller than the minimum of those constants.
Part (c): It is more convenient to introduce a variable change:
From the definition of in (E.1) and after straightforward manipulations, we can write (E.4) into
Applying the identities listed in (E.3), we obtain
Further, is a continuous function at , and thus there exists such that
The above result shows that is monotonically increasing in . Further, from (E.6) we have
It is straightforward to show that . Hence, has a unique solution if the following holds:
Part (d): From the fixed point equation where ( denotes ), we can derive the following (cf. (4.33))
Similar to the proof of part (b), when is sufficiently small. Hence, proving is simplified to proving that there exists such that
From (2.2) and after some calculations, we obtain the following
where . Then, we can reformulate (E.9) as
Similar to (E.4) and (E.5), (E.11) can be re-parameterized as
Finally, to show , we will prove that for . See the plot in Fig. 9. Since and , we only need to prove . Noting that both and are monotonically decreasing functions, it suffices to prove for , which directly follows from their definitions (cf. (E.10) and (4.59a)):
Part (e): First note that . Hence, the proof for the claim is straightforward if the inequality is strict for . This is the case since Lemma 14 shows that for , but equality only happends at .
Part (f): In Lemma 18, we have proved the following result in the case of :
(In fact, the above inequality holds for up to one.) In the noisy case, increases a little bit: . Hence, when is sufficiently small, we still have
Clearly, the inequality in (E.14) also holds for , since .
Part (g): Note that does not depend on . Further, and is a continuous function of . Hence, for small enough .
E.2.2 Convergence of the SE
where is the unique positive solution to and .
where was defined in (4.17). Notice that , and therefore it is guaranteed that for small enough . See Fig. 10 for illustration. To prove the lemma, we will prove the following arguments:
If , then there exists a finite such that .
If for (i.e., after one iteration), then there exists a finite such that .
We show that if for , then for all , and converges to .
The proof of (i) is similar to that of Lemma 8 and therefore omitted here.
Proof of (ii): Following the proof of Lemma 7, we argue that if then the following holds
where and . Then, it is easy to show that . Applying this recursively, we see that either moves to at a certain time or stays in . We next prove that the latter case cannot happen. Suppose that for . If this is the case, then it can be shown that
On the other hand, since we assume for , is upper bounded by and lower bounded by . Hence, this means the sequences and converges to and , respectively. This cannot happen since there is no fixed point in .
The proof for (E.16) and (E.17) are basically the same as those for the noiseless counterparts and hence skipped here. Please refer to the proof of Lemma 7. We only need to show that some of the key inequalities used in the proof of Lemma 7 still hold in the noisy case, which have been listed in Lemma 38 (e) and (f).
Proof of (iii): Lemma 38-(c), (d) and (g) imply that for all . Then, based on the strong global attractiveness of and , it is easy to show that if then for all . We have proved in Lemma 38-(d) that is a decreasing function of on and increasing on , where . Then, the maximum of on can only happen at either or . We assume that the latter case happens; it will be clear that our proof for the former case is a special case of the proof for the latter one. See the right panel of Fig. 10.
As discussed above, we assume that . Hence, by Lemma 38-(d), there exists a unique number such that . See the plot in the right panel of Fig. 10. We further divide into four regions:
Based on the strong global attractiveness of and (and similar to the proof of part (i) of this lemma), we can show the following:
if , then can only be in ;
if , then can be in , or ;
if , then can be in or .
Putting things together, and similar to the treatment of , it can be shown that there exists a finite such that for all .
It only remains to prove that if at a certain , then converges to . To this end, define
See examples depicted in Fig. 10. Using the strong global attractiveness of and and noting that for and for , it can be proved that
which implies that and . We skip the proofs for the above statements since similar arguments have been repeatedly used in this paper. ∎
E.2.3 Proof of Theorem 4
According to Lemma 39, we know that converges to the unique fixed point of the state evolution equation. We now analyze the location of this fixed point and further derive the noise sensitivity. Applying a variable change , we obtain the following equations for this unique fixed point:
where and are defined in (E.2). Using (E.18a) and , and after some algebra, we can write (E.18b) as
Differentiating with respect to yields
Using the identities listed in (E.3), we have
Also, it is straightforward to see that . Note that we have an implicit relation between and , and by the implicit function theorem we have
Further, is a continuously differentiable function of . Hence, by the mean value theorem we know that
To derive the noise sensitivity, we notice that
As shown in (E.3), can be expressed using elliptic integrals as:
From Lemma 3-(i), , hence . Further, since , we have . Therefore, . Hence, and so
E.3 Proof of Theorem 7
In the noisy setting, the state evolution of AMP.A becomes
Similar to the complex-valued case, the SE of real-valued AMP.A still converges to the nonzero fixed point, as stated in Lemma 40 below. We skip the proof since it is very similar to the proof of Lemma 39.
where is the unique positive solution to and .
Using (E.22a) and with simple manipulations we can rewrite (E.22b) as
From (E.22a) and the definition of , we have
Note that we have an implicit relation between and . By the implicit function theorem we have
Furthermore, from (E.25), we see that when and hence
Appendix F Spectral initialization
As shown in Section 2.2, to achieve successful reconstruction, the initial estimate cannot be orthogonal to the true signal , namely,
In many important applications (e.g., astronomic imaging and crystallography ), the signal is known to be real and nonnegative. In such cases, the following initialization of meets the non-orthogonality requirement:
(At the same time, we set .)
However, note that finding initializations that satisfy (F.1) is not straightforward in general settings. For instance, the above initialization may not work for generic complex-valued signals. Also, random initialization does not necessarily work either, since asymptotically speaking a random vector will be orthogoanl to . One promising direction to alleviate this issue is the spectral initialization method that was introduced in for phase retrieval and subsequently studied in . Specifically, the “direction” of the signal is estimated by the principal eigenvector () For the spectral method proposed in , the eigenvalues can be negative and the eigenvector associated with the largest eigenvalue (not the largest eigenvalue in magnitude) is picked. of the following matrix:
The above discussions suggest that the spectral method can provide the required non-orthogonal initialization for . However, the naive combination of the spectral estimate with will not work: performance of the that is initialized with the spectral method will not follow the state evolution. This is due to the fact that is heavily dependent on the matrix and violates the assumptions of SE. A trivial remedy is data splitting, i.e, we generate initialization and apply on two separate sets of measurements . However, this simple solution is sub-optimal in terms of sample complexity. To avoid such loss, we propose the following modification to the spectral initialization method, that we call decoupled spectral initialization:
Decoupled spectral initialization: Let . Set to be the eigenvector of corresponding to the largest eigenvalue defined in (F.2). Let , where is a fixed number which will be discussed later. Define
where denotes entry-wise product and is the unique solution of The uniqueness of solution in (F.4) and (F.5) is guaranteed for our choice of in (F.7). For the noisy case, we assume that the variance of the noise is known so that (F.4) and (F.5) can be calculated offline.
and is the unique solution of
The expectations above are over and , where is independent of . Now we use and as the initialization for . So far, we have not discussed how we can set and . In this paper, we use the following derived by :
In summary, our initialization in (F.3) intuitively satisfies “enough independency” requirement such that the SE for still holds. We have clarified this intuition in Section F.2. Our numerical experiments (see below) suggest that the intuition is correct. Our empirical finding is summarized below.
Let and be generated according to (F.3), and and generated by the algorithm as described in (1.6). The AMSE converges to
where is the solution to (F.3) and are defined as ( is defined in (F.6))
Fig. 11 shows a numerical example. The true signal is generated as . We measure the following two quantities (averaged over 10 runs):
We expect and to converge to their deterministic counterparts and (as described in Finding 1). Indeed, Fig. 11 shows that the match between the simulated and (solid curves) and the SE predictions (dotted curves) is precise. For reference, we also include the simulation results for the “blind approach” where the spectral initialization is incorporated into without applying the proposed correction (i.e., we use instead of (F.3)). From Fig. 11, we see that this blind approach deviates significantly from the SE predictions. Note that the blind approach still recovers the signal correctly for the current experiment. However, we found that (results are not shown here) the blind approach can perform rather poorly for other popular choices of (such as the orthogonality-promoting method proposed in ).
F.2 Intuition of our initialization
Note that in conventional , we set initial and therefore . Hence, our modification in (F.3) appears to be a rescaling procedure of . Note that solving the principle eigenvector of in (F.2) is equivalent to the following optimization problem:
Following the derivations proposed in , we obtain the following approximate message passing algorithm for spectral method (denote as ):
The optimizer of (F.10) can be regarded as the limit of the estimate under correct initialization of . Note that acts as a proxy and we do not intend to use it for the eigenvector calculations. (There are standard numerical recipes for that purpose.) But, the correction term used in (F.3) is suggested by the Onsager correction term in AMP.S. To see that let , , represent the limits of , , respectively. Then, from (F.11a) and (F.11b), we obtain the following equation
By solving (F.12), we obtain (F.3) with rescaling of (since and ). Further, (F.4) and (F.5) that determine the value of can be simplified through solving the fix point of the following state evolution of :
where are defined in (F.6) and is defined in (F.9).