Phase Retrieval: Stability and Recovery Guarantees
Yonina C. Eldar, Shahar Mendelson
Introduction
where is noise, and are a set of known vectors. Since only the magnitude of is measured, and not the phase (or the sign, in the real case), this problem is referred to as phase retrieval. Phase retrieval problems arise in many areas of optics, where the detector can only measure the magnitude of the received optical wave. Several important applications of phase retrieval include X-ray crystallography, transmission electron microscopy and coherent diffractive imaging .
Many methods have been developed for phase recovery which often rely on prior information about the signal, such as positivity or support constraints. One of the most popular techniques is based on alternating projections, where the current signal estimate is transformed back and forth between the object and the Fourier domains. The prior information and observations are used in each domain in order to form the next estimate. Two of the main approaches of this type are Gerchberg-Saxton and Fienup . In general, these methods are not guaranteed to converge, and often require careful parameter selection and sufficient prior information.
To circumvent the difficulties associated with alternating projections, more recently, phase retrieval problems have been treated using semidefinite relaxation, and low-rank matrix recovery ideas . In several masks where used in the measurement process in order to ensure the ability to retrieve the phase. Another approach to generate robust solutions is to assume that the input signal is sparse, namely, that it contains only a few non-zeros values in an appropriate basis expansion. Sparsity has long been exploited in signal processing, applied mathematics, statistics and computer science for tasks such as compression, denoising, model selection, image processing and more. Despite the great interest in exploiting sparsity in various applications, most of the work to date has focused on recovering sparse or low rank data from linear measurements . Recently, the basic sparse recovery problem has been generalized to the case in which the measurements are quadratic , or given by a more general nonlinear transform of the unknown input . The first paper to consider sparse phase retrieval was , based on semidefinite relaxation combined with a row-sparsity constraint on the resulting matrix. An iterative thresholding algorithm was then proposed that approximates the solution. Similar approaches were later used in . An alternative algorithm was recently designed in using a greedy search method which is far more efficient than the semidefinite relaxation, and often yields more accurate solutions.
Despite the vast interest in phase retrieval, there has been little theoretical work on the fundamental limits of this problem. One important question in this context is how many measurements are needed in order to ensure robust recovery of the input , regardless of the specific recovery method used. Several recent works treat this problem. Most of the papers discuss the case in which is a general input, namely, there is no sparsity (or other) constraint on . The first result of this kind was obtained in , where it is shown that with probability one randomized equations are sufficient for recovery using a brute force (intractable) method, when there is no noise. However, it is not clear whether a stable recovery method exists with this number of measurements. In the authors consider the case in which are real or complex vectors that are either uniform on the sphere of radius , or iid zero-mean Gaussian vectors with unit variance. Under these assumptions they show that on the order of measurements are needed in order to recover a generic using a semidefinite relaxation approach. In the presence of noise, it is shown in that one can find an estimate satisfying
for some , where is a constant and is the noise vector that is assumed to be bounded so that is finite.
The paper treats the case in which the input is -sparse and are iid zero-mean normal vectors. When there is no noise, they show that in the real case measurements are needed for uniqueness and in the complex case, measurements are required. They further prove that if is on the order of then the solution can be obtained using a sparse semidefinite relaxation approach as in .
It turns out that the natural complexity parameter for this problem is the same as the one used for analyzing stability in linear measurements, as we will discuss in Section 5. Thus, in a rather general sense, the number of measurements required for stable recovery in the quadratic setting we treat here is of the same order of magnitude as the one needed to ensure stability under linear sampling. In that sense there is no substantial price to be paid for not knowing the phase of the measurements, and for very general choices of input sets .
The second main result of this article deals with the noisy phase retrieval problem. More specifically, we consider recovering an input in a set from noisy measurements of the form (1.1). A straightforward approach is to seek the value of that minimizes the empirical risk (or a least-squares approach). Since this leads to a nonconvex problem, finding its global solution is in general not possible. Nonetheless, we show that if one can find a value for which the empirical risk is bounded by a given, computable constant (which depends on the set ), then is bounded above by an expression that once again depends on the complexity parameter of the set, and which converges to faster than for any . Here is the true (unknown) input. The complexity parameter that determines the rate in the noisy setting is essentially the same as in the stability analysis. Moreover, the resulting sample complexity is of the same order of magnitude as in the linear case in the examples of sets we consider. An exact formulation of both main results is presented in the next section.
The reminder of the article is organized as follows. The problem and the main results are formulated in Section 2. Stability results in the noise-free setting are developed in Section 3, while the noisy setting is treated in Section 4. In Section 5 the relation between the results in the quadratic case and those in the linear setting are discussed.
Problem Formulation and Main Results
Our goal is to study conditions under which stable recovery is possible irrespective of the specific recovery method used, and to develop guarantees that ensure that empirical minimization or approximate empirical minimization (namely, least-squares recovery) lead to an estimate that is close to in a squared-error sense.
If is a class of functions on a probability space , then it is -subgaussian if for every and every
where is distributed according to .
Our goal is to study when the mapping is both invertible and stable first, when (the noise-free case) and second, in the presence of noise.
2 Stability Results
The mapping is stable with a constant in a set if for every ,
Note that stability in a set is a much stronger property than invertibility. Indeed, for the latter it suffices that if then , but without any quantitative estimate on the difference.
Let be independent Gaussian random variables, that have mean zero and variance . Set
Throughout this section, we will refer to as the complexity measure of .
The main result in the noise free case is the following:
For every there exist constants and that depend only on for which the following holds. Let be an isotropic, -subgaussian measure. Then, for , with probability at least , for every ,
To put Theorem 2.4 in the right perspective, one has to obtain lower bounds on and upper bounds on . Since the latter depends on the number of measurements , its behavior provides insight into the number of measurements that are needed for stability.
In particular, the result is true for a random Gaussian matrix , where and are absolute constants.
Interestingly, it can be shown that in the case of linear measurements, stable recovery is guaranteed as long as . Thus, the number of measurements needed for stable recovery in the nonlinear and linear settings is the same up to multiplicative constants – at least for ensembles that have a well behaved . As mentioned in the introduction, this observation is not a coincidence and will be explained in more detail in Section 5.
In Section 3.2 we study other choices of , and the number of measurements needed in order to guarantee stability.
3 Noisy Recovery Results
Section 4 is devoted to the case in which the measurements are contaminated with iid noise. The goal is to find a point for which is small, using the data and the fact that is generated according to (1.1) for some .
A natural approach is to recover from by minimizing the empirical risk:
be the Gaussian complexity of , where are iid standard Gaussian variables, and put
Suppose that for a given , and (which will later on govern our probability estimates), one produces satisfying
Here , is a measure of the decay properties of the noise, and will be defined formally in (4.1), is a complexity measure similar to (defined formally in (4.8)), and . Our main result shows that, with high probability, such a point is close to either or to . To find an appropriate , it is possible, for example, to use the greedy method of with different starting points and stop once a solution that satisfies the bound is found.
For every and every there exists constants and that depend only on and , for which the following holds. Let be distributed according to an isotropic, -subgaussian measure, and assume that where . Assume further that . For every integer set
Let be chosen to satisfy (2.11). Then, for , with probability at least ,
4 Technical Tool
The main technical tool needed in the proof of both main results is a general estimate on properties of empirical processes indexed by . Although the result is true in a far more general situation than needed here, for the sake of simplicity we will present it only in the cases required. We refer the reader to for the more general statement and precise results.
Stability Results
In this section we present the proof of Theorem 2.4, followed by estimates on the values of and appearing in the theorem.
Therefore, to establish the desired stability result, it suffices to show that
By Theorem 2.8 for F=\{|\bigl{<}v,\cdot\bigr{>}|:v\in T_{+}\} and H=\{|\bigl{<}w,\cdot\bigr{>}|:w\in T_{-}\}, it follows that if and then with probability at least ,
The claim now follows immediately from the definition of and of .
Let be a compact metric space. For every , let be the smallest number of open balls of radius needed to cover . The numbers are called the -covering numbers of relative to the metric .
The upper bound is due to Dudley and the lower to Sudakov . The proof of both bounds may be found, for example, in .
It is straightforward to verify that the gap between the upper and lower bounds in Proposition 3.3 is at most , and in all the examples we study below, the resulting estimate is sharp.
2.2 Bounding κ𝜅\kappa
Here, we present two simple methods for bounding from below. These methods are not the only possibilities by which one may obtain such a bound; rather, they serve as an indication that the assumption on is less restrictive than may appear at first glance.
If satisfies the small ball assumption with constant then
Proof. Consider for which . Then for every , there is an event of measure at least on which |\bigl{<}v,a\bigr{>}|\geq\varepsilon. Hence, for two fixed vectors ,
The following lemma is standard (see e.g. ).
The desired small-ball estimate clearly follows from the lemma, since
The second method, which we only outline, is based on the Paley-Zygmund argument.
Let be a random variable, set and put . Then, for every ,
We will use the lemma for and . Assume that are iid copies of a symmetric, variance random variable and set . If , then a straightforward computation shows that
Using the fact that , (3.2.2) reduces to
On the other hand, if the reverse inequality holds, then using
Let be a symmetric, variance random variable, with a finite moment for some . If then
where depends on and on .
Proof. Assume that for some . Observe that if then for every , \|\bigl{<}a,v\bigr{>}\|_{L_{r}}\leq c_{r}\|X\|_{L_{r}}. Indeed, by a Rosenthal type inequality (see, e.g. , Section 1.5),
Since and , the claim follows.
Therefore, \sup_{v\in S^{n-1}}\|\bigl{<}a,v\bigr{>}\|_{L_{2q}}\leq c_{q}\|X\|_{L_{2q}}, and thus,
3 Examples
Let us turn to a few special cases of Theorem 2.4. To that end, explicit expressions for are required for the sets of interest.
The corollary follows from the fact that with this choice of , is proportional to .
When is given by a constant, independent of the dimension , Corollary 3.8 implies that it is sufficient to choose to ensure stable recovery with high probability.
3.2 Sparse Vectors
where is a monotone rearrangement of . It is standard to check (see, e.g., ) that there is an absolute constant such that for every ,
For every there are constants , and that depend only on and for which the following holds. If , and , then with probability at least , for every ,
When is an absolute constant, Corollary 3.8 implies that it is sufficient to choose to ensure stable recovery with high probability.
3.3 Finite Set
Therefore, , implying that
For every there are constants , and that depend only on and for which the following holds. If , and , then with probability at least , for every ,
In this case, with constant , measurements ensure stable recovery with high probability.
3.4 Block Sparse Vectors
There exist absolute constants and for which the following holds. For every ,
Proof. Let and observe that
where for every , is the Euclidean sphere on the coordinates . Clearly, there are at most such subsets . Using a standard volumetric estimate (see, e.g., ), for every fixed set and every , one needs at most Euclidean balls of radius to cover . Therefore, for every ,
The second part of the claim is an immediate consequence of Proposition 3.3 and the fact that is a decreasing function of .
For every there are constants , and that depend only on and for which the following holds. If , and , then with probability at least , for every ,
When is constant we conclude that measurements are needed for stability. This result is consistent with that of which shows that the same value ensures that a random Gaussian matrix satisfies the block restricted isometry constant.
Noisy Measurements
Next, consider the phase retrieval problem in the presence of noise. The goal is to find an estimate of the true signal that is close to (or ) in a squared error sense.
for some . Let be an isotropic, -subgaussian random vector and assume that the noise is independent of , symmetric, and of reasonable decay properties, which will be specified in Assumption 4.1 below.
Given , combined with the information that the noisy data is generated by a point via (4.1), is it possible to produce an estimate for which is small?
Note that the error is measured by the product , since it is impossible to distinguish between and .
The answer to this question is affirmative, as shown in Theorem 4.8.
Throughout our analysis we assume that the noise decays properly. In order to quantify this decay we rely on the notion of random variables, which are defined below (see as general references for properties of random variables).
Let be a random variable. For let
and denote by the set of random variables for which .
The norm can be characterized using information on the tail of . Indeed, there exists an absolute constant , for which, if , then . The reverse direction is also true, that is, if , then for an absolute constant .
It is well known that is a norm on , and that
In the language of the previous section, is -subgaussian if and only if . Since the norms have a natural hierarchy, it follows that if is -subgaussian then
Therefore, if is -subgaussian and mean-zero then , where is the standard deviation of .
A straightforward application of the tail behavior of a random variable implies that if are independent copies of and , then
From the definition of the norm it is evident that if then
and in particular, for if and only if .
Although there are versions of the following theorem (and of Definition 4.2) for any , for the sake of simplicity, we shall restrict ourselves to the case , which is the setting needed in the proofs below.
There exists an absolute constant for which the following holds. If and are independent copies of , then for every ,
Combining Theorem 4.3 and (4.4) leads to the following corollary:
Let and assume that is a random variable for which (or ). Then, with probability at least ,
The corollary follows immediately from Theorem 4.3 by taking for , and since .
For every there exist constants , and that depend only on and for which the following holds. If , then with probability at least ,
2 The Recovery Algorithm
The assumptions we make throughout this section are as follows:
Assume that is isotropic and subgaussian, and that the noise in (4.1) is a symmetric, random variable that is independent of .
Recall that the goal is to find an estimate of that is close to or to . Given the measurements , a reasonable approach is to seek a value of that minimizes the empirical risk function:
for some . Here we will consider values of in the regime ; the exact choice of will become clear later on. Note that for every ,
Let be given, and choose a value of . Given the data , is called a good estimate if it satisfies that
To motivate the choice of in Definition 4.6, observe that captures the “statistical complexity” of the problem – namely, the sum of the “gaussian complexity” of , , and the influence of the noise, . The parameter tunes the probability estimate, for the moment is of secondary importance. The exact choice of and will be specified in Theorem 4.8.
Observe that the value on the left hand side of (4.9) is the empirical excess risk where
is the excess loss functional. The definition implies that the empirical excess risk at is of the same order of magnitude as the “statistical error” and thus
Unfortunately, it is impossible to estimate the empirical excess risk since one does not have access to the sampled noise , and therefore, nor to – which is the reason for the second modification. By Assumption 4.1, and consequently . From Corollary 4.4, if , then with probability at least ,
Therefore, if satisfies (4.7), then it also satisfies (4.9), meaning that its empirical excess risk is bounded above by the desired quantity. This leads to the following proposition.
There exists an absolute constant for which the following holds. Let be a point that satisfies (4.7) and let . If , then with probability at least , .
To see that there is always a point that satisfies (4.7), observe that for and , with probability at least (see Corollary 4.4),
Moreover, unless is very small and is very large, is the dominant term in . For example, consider the case in which is a centered Gaussian with variance and is the set of -sparse vectors on the unit sphere. Then, , while which clearly is larger than , as long as is large relative to .
We are now ready to state our main result. To this end recall the definition of given by (2.6), and let .
For every and every there exists constants and that depend only on and , for which the following holds. Let be distributed according to an isotropic, -subgaussian measure, and assume that . Assume further that . For every integer set
Let be chosen to satisfy (4.7). Then, for , with probability at least ,
Note that implies that for any .
If (which is the reasonable range, as one expects up to logarithmic factors), then , and by Theorem 4.8,
To proceed, and as will be noted in Section 5, in the case of linear measurements, with high probability,
To compare the “quadratic” estimate with the linear one, note that if for and some constant , then recalling that for every , , it is evident that
where is an absolute constant. Therefore, with this choice of ,
and up to logarithmic factors scales as the estimate in the linear case.
Clearly, it suffices to take to ensure that , which is off only by a factor from the optimal estimate in the linear case.
For every and there exist constants , that depend only on and and for which the following holds. Let be the set of -sparse vectors on the sphere, set to be distributed according to an isotropic, -subgaussian measure and assume that . If the noise is -subgaussian, for and , then with probability at least ,
In particular, if then with probability at least .
3 Proof of Theorem 4.8
The proof of the theorem requires several preliminary facts about empirical and Bernoulli processes. We refer the reader to for more details on these processes.
Throughout this section, is a probability space and are iid, distributed according to . Let be independent, symmetric, -valued random variables, that are independent of .
The first result we require is the contraction inequality for Bernoulli processes.
The following symmetrization argument allows one to bound an empirical process using the Bernoulli process indexed by the random set .
We will use Theorem 4.10 and Theorem 4.11 with for .
The final result we require is the Kahane-Khintchine inequality , on the moments of Bernoulli processes.
For every there exist constants and that depend only on for which the following holds. If is chosen as in Theorem 4.8, then for , with probability at least , for every ,
Proof. Fix . By the symmetrization theorem (Theorem 4.11) and the independence of and ,
and observe that for every realization of , the functions vanish at and are Lipschitz on with a constant . For b\leq\max_{1\leq i\leq N}\sup_{x\in T}|\bigl{<}a_{i},x-x_{0}\bigr{>}\bigl{<}a_{i},x+x_{0}\bigr{>}| this constant is proportional to , since . Applying the contraction inequality (Theorem 4.10), conditioned on and ,
By the Kahane-Khintchine inequality, the Cauchy-Schwarz inequality, and Jensen’s inequality combined with reverse symmetrization (the other direction of Theorem 4.11),
where is taken with respect to the -product measure .
and it remains to bound and .
Turning to , observe that pointwise
Set . With this choice, combined with the moment characterization of the norm (4.2), it is evident that
for a suitable absolute constant . Indeed,
With these two estimates, it is evident that there exists a constant that depends only on for which, for every ,
With this estimate at hand, it is standard to show (see, e.g., for a similar argument), that for , with probability at least ,
where and depend only on .
Finally, in this case, for every
With Lemma 4.13 in mind, the choice of becomes clearer. One would like to find any point in for which is, at most, of the same order of magnitude as the combined complexity term of the set and the noise
Given set h_{x}=\bigl{<}a,x-x_{0}\bigr{>}\bigl{<}a,x+x_{0}\bigr{>} and recall that . Since is a symmetric random variable, it is distributed as , where is a symmetric -valued random variable, independent of and of . Therefore,
Observe that the function is increasing for and that . Hence, for every ,
By the definition of , for every and ,
Combining this lower bound with (4.13), and recalling that completes the proof of the theorem.
4 Examples
Let us present some of the examples seen in Section 3.2, in the noisy setting. Other examples may be obtained with similar ease.
In all the examples below we will assume that and so . Since is symmetric and -subgaussian, then , where is the noise variance. Also, since , .
for the regime of we are interested in, and
Suppose that . Then, by Theorem 4.8,
where is an absolute constant. If for , then
for a suitable absolute constant . Therefore, with this choice of ,
and it suffices to take to ensure that with probability at least .
For every and there exist constants , that depend only on and and for which the following holds. If , and are as above, and for , then for with probability at least ,
4.2 Sparse Vectors
We already treated the case of sparse vectors in Corollary 4.9. The block-sparse setting can be treated in a similar manner, leading to the following corollary.
For every and there exist constants , that depend only on and and for which the following holds. If and are as above, is the set of -block sparse vectors of length on the sphere and for , then for with probability at least ,
In particular, if then with probability at least .
Connection with Results on Linear Estimation
It should come as no surprise that the methods used here are very similar in nature to the analogous “linear questions”. Both stability and noisy recovery are well understood in the linear case, and in a sharp way, as we will explain below.
Stability in a set for a random ensemble depends on the way in which a typical operator acts on the set
The study of the process (5.2), both for and for an arbitrary subset of the sphere has been extensive in recent years. A good starting point for the interested reader would be for subgaussian ensembles, for log-concave ensembles, and for ensembles with heavy tails (though this does not begin to cover the extensive literature on the topic).
In the context of this paper, subgaussian ensembles, the best estimate on (5.2) follows from Theorem 2.8, applied to the class F=H=\{\bigl{<}v,\cdot\bigr{>},\ v\in T_{-}\}. Moreover, in it was shown that under very mild assumptions on the set , the estimate is sharp.
The best results to-date on linear regression that take into account the complexity of the indexing set can be found in . One may show that these estimates are sharp under very mild assumptions on , and it turns out that these assumptions are satisfied in the examples that were presented here. Since our bounds in the “quadratic” case are of the same order of magnitude as in the easier, linear case, and since these bounds are optimal in the linear case, it is reasonable to expect that they are optimal in the quadratic scenario as well. Unfortunately, the methods required to prove this optimality are rather involved, and we will not explore this issue here. Rather we refer the reader to , in which the linear case is explored.