Statistical mechanics approach to 1-bit compressed sensing
Yingying Xu, Yoshiyuki Kabashima
Introduction
Compressed (or compressive) sensing (CS) is a technique for recovering a high-dimensional signal from lower-dimensional data, whose components represent partial information about the signal, by utilizing prior knowledge on the sparsity of the signal . The research field of CS is one of the main topics in information science nowadays and has been intensively investigated from the theoretical point of view . This technique has also been used in various engineering fields .
Let us suppose a situation that an -dimensional vector is linearly transformed into an -dimensional vector by an measurement matrix , where \textrm{\boldmathy}=\textrm{\boldmath\Phi x^{0}}. The signal is recovered from by determining the sparsest signal that is consistent with the measurements. When , the measurement usually loses some information and the inverse problem has an infinite number of solutions. However, when the -dimensional signal is guaranteed to have only nonzero entries in some convenient basis and the measurement matrix is incoherent with that basis, there is a high probability that the inverse problem has a unique and exact solution. For example, smooth signals and piecewise-smooth signals, like natural images or communications signals, typically have a representation in a sparsity-inducing basis such as a Fourier or wavelet basis .
This paper is organized as follows. The next section sets up the problem that we will focus on when explaining the 1-bit CS scheme. Section 3 uses the replica method to examine the signal recovery performance achieved by the scheme. In section 4 an approximate signal recovery algorithm based on the cavity method is developed and evaluated, and the final section is devoted to a summary.
Problem setup
where for simplicity we assume that each entry of measurement matrix is provided as an independent sample from an identical Gaussian distribution of zero mean and variance .
based on the -recovery method widely used and studied for standard CS problems . Here and denote the - and -norms of , respectively. The measurement process of (2) completely erases the information of length , which makes it impossible to recover the signal uniquely. We therefore introduce an extra normalization constraint for the recovered signal , and we consider the recovery successful when the direction cosine is sufficiently large.
Unlike the standard CS problem, finding a solution of (3) is non-trivial because the norm constraint keeps it from being a convex optimization problem (Figures 1 (a) and (b)). The authors of also developed, as a practically feasible solution, a double-loop algorithm called Renormalized Fixed Point Iteration (RFPI) that combines a gradient descent method and enforcement to a sphere of a fixed radius. It is summarized in Figure 2.
The practical utility of RFPI was shown by numerical experiments, but how good solutions are actually obtained is unclear because in general the algorithm can be trapped at various local optima. One of our main concerns is therefore to theoretically evaluate the typical performance of the global minimum solution of (3) for examining the possibility of performance improvement.
Performance assessment by the replica method
where and for and , respectively, offers the basis for our analysis. As tends to infinity, the integral of (4) is dominated by the correct solution of (3). One therefore can evaluate the performance of the solution by examining the macroscopic behavior of equation (4) in the limit of .
A characteristic feature of the current problem is that (4) depends on the predetermined random variables and , which requires us to assess the average of free energy density when evaluating the performance for typical samples of and . Here, denotes the configurational average concerning and . Because directly averaging the logarithm of the partition function is technically difficult, we here resort to the replica method .
In particular, under the replica symmetric (RS) ansatz where the dominant saddle point is assumed to be of the form of
in the limit of . Here , denotes extremization of a function with respect to , , is a Gaussian measure, and
The derivation of is provided in A.
The extremization problem of (12) yields the following saddle point equations:
where . The value of determined by these equations physically means the typical overlap between the original signal and the solution of (3). Therefore the typical value of the direction cosine between and , which serves as a performance measure of the current recovery problem, is evaluated as . Alternatively, we may also use as a performance measure the mean square error (MSE) between the normalized vectors:
We solved the saddle point equations for various sets of and . The curves in figures 3 (a)–(d) show the theoretical prediction of MSE evaluated by (19) plotted against the measurement bit ratio for , and . To examine the validity of the RS ansatz, we also evaluated the local stability of the RS solutions against the disturbances that break the replica symmetry , which offers
as the stability condition. A brief sketch of the derivation of this condition is shown in B. Unfortunately, (21) is not satisfied for any regions in figures 3 (a)–(d). This is presumably because the optimization problem for (3) has many local optima reflecting the fact that the constraint of loses the convexity. This indicates that taking the replica symmetry breaking (RSB) into account is necessary for evaluating the exact performance of the signal recovery scheme defined by (3).
We nonetheless think that the RS analysis offers considerably accurate approximates of the exact performance in terms of MSE. The () symbols in figures 3 (a)–(d) stand for MSE experimentally achieved by RFPI, which were assessed as the arithmetic averages over samples for each condition of systems. Excellent consistency between the curves and symbols suggests that even if (3) has many local optima, they are close to one another in terms of the -norm yielding similar values of MSE. This also implies that RFPI, which is guaranteed to find one of the local optima, performs nearly saturates as well (as measured by the MSE) as the signal recovery scheme based on (3).
Of course, we have to keep in mind that the consistency between the theory and experiments depends highly on the performance measure used. Figures 4 (a)–(d) show the probabilities of wrongly predicting sites of nonzero and zero entries, which are sometimes referred to as false positive (FP) and false negative (FN), respectively. These indicate that there are considerably large discrepancies between the theory and experiments in terms of these performance measures, which is probably due to the influence of RSB. Nevertheless, the RS-based theoretical predictions are still qualitatively consistent with the experimental results in the way that the probability of a FP remains finite even when the measurement bit ratio tends to infinity for any values of . This implies that the -based scheme is intrinsically unable to correctly identify sites of nonzero and zero entries.
Cavity-inspired signal recovery algorithm
The analysis so far indicates that the performance of RFPI is good enough in the sense that there is little room for improvement in achievable MSE. RFPI requires tuning of two parameters and , however, which is rather laborious. In addition, the convergence of the inner loop of Figure 2 is relatively slow, which may limit its application range to systems of relatively small sizes. We therefore developed another recovery algorithm following the framework of the cavity method of statistical mechanics , or equivalently, the belief propagation of probabilistic inference .
For simplicity of notations, let us first convert all the measurement results to by multiplying to each row of the measurement matrix as , and newly denote the resultant matrix as . In the new notation, introduction of Lagrange multipliers and surplus variables converts (3) to an unconstrained optimization problem:
where means that each entry of is restricted to be positive.
Coupling terms make the optimization of (23) a nontrivial problem. In statistical mechanics, a standard approach to resolving such a difficulty is to approximate (23) with a bunch of optimizations for single-body cost functions parameterized as
where , and are parameters to be determined in a self-consistent manner.
In the cavity method this is done by introducing virtual systems that are defined by removing a single variable or a single pair of variables from the original system . When is sufficiently large, the law of large numbers allows us to assume that the values of and are constant independently of their indices; that is, that and are constants. Under this simplification, this method yields a set of self-consistent equations:
where , , , and . is evaluated using and as
is determined so that holds in (29), which provides
on the right-hand side of (28) is often referred to as the Onsager reaction term . Equation (29) offers the recovered signal. The derivations of these equations are provided in C.
A distinctive feature of the above set of equations is that they are free from tuning parameters such as and in RFPI, which is highly beneficial in practical use. It is therefore unfortunate that in most cases the naive iterations of (26)(27), (30)(28)(29), (31)(26) hardly converge, which is considered a consequence of RSB , while a similar approach offers successful results for various other problems of compressed sensing .
We found, however, that instead of updating by (31) at each iteration, handling as a parameter to be controlled in the outer loop, in conjunction with modifying (26) and (27) to
results in a fairly good approximate signal recovery algorithm.
The necessity of controlling in the outer loop, which is essential for having good convergence in the inner loop, means that our algorithm still requires one tuning parameter. Nonetheless, the reduction in the number of the tuning parameters from two to one is considerably advantageous for practical use. In practice, the initial value of should be set so that only a single entry becomes nonzero. This is easily done by the Binary Iterative Hard Thresholding algorithm , which requires the number of nonzero entries as extra prior knowledge. After the initial value is set, is reduced as with an appropriate constant , where is the counter of the outer loop. The algorithm terminates when the difference between the convergent solutions of two successive outer loops is sufficiently small.
The resultant algorithm is somewhat similar to RFPI as the combination of (28) and (32) roughly acts as the One-sided quadratic gradient descent step in Figure 2. However, as the length of is not restricted to a fixed value, the current algorithm does not need a small step size for the convergence. Another significant difference from RFPI is the existence of the Onsager reaction term in (28). This term effectively cancels the self-feedback effects included in of (28), and this is expected to accelerate the convergence of the algorithm. A pseudocode for the inner loop is summarized in Figure 5.
The MSE results obtained in numerical experiments with the cavity-inspired signal recovery (CISR) algorithm are shown in Figures 6 (a)–(d). They indicate that except in the case in which the nonzero density of the original signals is significantly low, CISR provides MSE values almost equal to or lower than those of RFPI. Figures 7 (a)–(d) show the FP and FN probabilities for CISR. The discrepancies from the theoretical prediction are not unexpected because the modification of (27) to (32) means that CISR is no longer based on (3) or (23). The FN probabilities for CISR are higher than those for RFPI, while the FP probabilities are lower. This implies that CISR has a capability of yielding sparser signals than RFPI, which is presumably because parameter of CISR is initially set so that only a single entry of is nonzero while such a tuning is not taken into account in RFPI.
The run times actually required for performing the experiments in a MATLAB® environment for the cases of and are listed in Table 1. Although the run times of RFPI may be reduced by optimally tuning the descent step size , CISR is several hundreds of times faster than RFPI. This shows the significant computational efficiency of CISR. The NORT values in Table 1 are the run times when the Onsager reaction term in (28) was removed from CISR. Their being 1.13–2.37 times longer than those for CISR indicates that the cancellation of the self-feedback effects by adding the Onsager reaction term speeds the convergence of CISR significantly.
Summary
In summary, we have examined typical properties of 1-bit compresses sensing (CS) proposed in utilizing methods of statistical mechanics. Signal recovery based on the -norm minimization is a standard approach in CS research. Unlike the normal CS scheme, however, the -based signal recovery cannot be formulated as a convex optimization problem, which makes practically performing it nontrivial.
We have shown that the theoretical prediction of the performance of the -based scheme, which is obtained by the replica method under the replica symmetric (RS) ansatz, exhibits a fairly good accordance (in terms of MSE) with experimental results obtained using for an approximate signal recovery algorithm, RFPI, proposed in . The replica symmetry of the RS solution turned out to be broken, however, which implies that there are many local optima for the optimization problem of the signal recovery. Our results suggest that the local optima, which can be searched by RFPI, yield similar values of MSE representing the potential performance limit of -based recovery scheme.
We have also developed an approximate signal recovery algorithm utilizing the cavity method. Naive iterations of self-consistent equations derived directly from the cavity method hardly converge in most cases, which can be regarded as a consequence of the replica symmetry breaking. However, we have shown that modification of one equation in an appropriate manner, in conjunction with controlling a macroscopic variable in the outer loop, results in a fairly good signal recovery algorithm. Compared with RFPI, the resultant algorithm is beneficial in that the number of tuning parameters is reduced from two to one. Numerical experiments have also shown that whenever the density of nonzero entries of the original signal is not considerably small the cavity-inspired algorithm performs as well as or better than RFPI (in terms of MSE) and has a lower computational cost.
We here focused on the -based recovery scheme since it was proposed and examined in the seminal paper on 1-bit CS . However, the significance of the -based scheme may be rather weak for 1-bit CS because the loss of convexity it entails keeps it from leading to the development of mathematically guaranteed and practically feasible algorithms. Therefore, much effort should be devoted to developing recovery algorithms following various principles. For example, the idea based on the Bayesian inference and matrix design that was proposed for standard CS may also be a promising approach for 1-bit CS.
Appendix A Derivation of (12)12(\ref{eq:free energy})
Averaging (6) with respect to and offers the following expression of the -th moment of the partition function:
where , into (34). Furthermore, we define a joint distribution of vectors as
where and
Equation (39) can be regarded as the average of with respect to and over distributions of and . In computing this, it is noteworthy that the central limit theorem guarantees that can be handled as zero-mean multivariate Gaussian random numbers whose variance and covariance are provided by
when and are generated independently from and , respectively. This means that (39) can be evaluated as
Here and is an symmetric matrix whose and other diagonal components are given as and , respectively, while off-diagonal entries are offered as . Equations (42) and (46) indicate that is correctly evaluated by using the saddle point method with respect to in the assessment of the right-hand side of (38) when and tend to infinity keeping finite.
A.2 Treatment under the replica symmetric ansatz
Let us assume that the relevant saddle point in assessing (38) is of the form of (11) and, accordingly,
dimensional Gaussian random variables whose variance and covariance are provided as (11) can be expressed as
utilizing independent standard Gaussian random variables and . This indicates that (42) is evaluated as
On the other hand, substituting (51) into (46), in conjunction with the identity
In the limit of , a nontrivial saddle point is obtained only when is kept finite. Accordingly, we change the notations of the auxiliary variables as , , and . Furthermore, we use the asymptotic forms
Using these in the resultant expression of offers (12).
Appendix B Stability of the RS solution
The 1-step replica symmetry breaking (1RSB) ansatz means that, at the relevant saddle point, replica indices are classified into groups of an equal size , and holds if and belong to an identical group and , otherwise. This yields the following expression of the average free energy of finite temperature:
where , , , and . The RS solution is regarded as a special case of the 1RSB solution for which holds. Therefore one can check the thermodynamical validity of the RS solution by examining the stability of the solution of under the 1RSB ansatz.
The extremization condition of (65) indicates that
hold for and irrespectively of the value of . Here and represent assessments of and under the assumptions of and , respectively. In (69) and (74) we used the Taylor expansion expressions and , and the fact that the variances of for the measures and become unity as and vanish, irrespectively of the value of .
To examine the stability of the RS solution in the limit of , let us change the variable notations as , , , , and and set and . This yields expressions of and for . Substituting these into (69) and (74) leads to
where we set and , and used . The condition that (75) and (76) allow a solution of offers (21).
Appendix C Derivation of the cavity equations
We refer to the system in which and are kept out as the -cavity and -cavity systems, respectively. In addition, we denote , and as the single-body cost function for the -cavity system and its parameters, respectively, and similarly for , and . Self-consistent equations are derived from the following arguments.
Vertical step: Let us suppose that is put into the -cavity system, which yields an approximation of the cost function of (23) as . From this function we remove all terms that are related to of a certain index , which leads to an approximate cost function of the -cavity system. must be obtained by partially optimizing the resulting -cavity cost function with respect to
of the remaining indices , where generally denotes the set provided by removing an element from a set . This offers the relation
This relation and the fact that is a negligibly small independent sample from an identical Gaussian distribution with zero mean and variance yield the following equations evaluating and from a set of and :
Horizontal step: Similarly, putting into the -cavity system and removing yields another relation,
Recovery step: and are evaluated from (81) and (82) as
This means that the recovered signal is provided as
where is determined in such a way that holds. Similarly,
are obtained from (78) and (79). These offer the (approximate) optimal value of the Lagrange multiplier as
Equations (79) and (84) indicate the difference between
is vanishingly small for as scales as , and similarly for
This also allows us to handle and as a single site-independent parameter , and we similarly deal with and as . These considerations, in conjunction with (83) and (86), offer
where we replaced in (83) and (86) with its expectation by utilizing the law of large numbers. Furthermore, inserting and into (84) and (87), respectively, yields
where we set . Equations (85), (88), and (89)–(96) lead to (26)–(29).