Professional Documents
Culture Documents
Statistical-Physics-Based Reconstruction in Compressed Sensing
Statistical-Physics-Based Reconstruction in Compressed Sensing
Statistical-Physics-Based Reconstruction in Compressed Sensing
DOI: 10.1103/PhysRevX.2.021005 Subject Areas: Complex Systems, Computational Physics, Statistical Physics
FIG. 1. Two illustrative examples of compressed sensing in image processing. Top: The original image, known as the Shepp-Logan
phantom, of size N ¼ 1282 , is transformed via one step of Haar wavelets into a signal of density 0 0:15. With compressed sensing,
one is thus in principle able to reconstruct the image exactly with M 0 N measurements, but practical reconstruction algorithms
generally need M larger than 0 N. The five columns show the reconstructed figure obtained from M ¼ N measurements, with
decreasing acquisition rate . The first row is obtained with the ‘1 -reconstruction algorithm [3,4] for a measurement matrix with
independent identically distributed (iid) elements of zero mean and variance 1=N. The second row is obtained with belief propagation,
using exactly the same measurements as used in the ‘1 reconstruction. The third row is the result of the seeded belief propagation,
introduced in this work, which uses a measurement matrix based on a chain of coupled blocks (see Fig. 4). The running time of all
three algorithms is comparable. (Asymptotically, they are all quadratic in the limit of the large signal size.) In the lower set of images,
we took as the sparse signal the relevant coefficients after two-step Haar transform of the picture of Lena. Again for this signal, with
density 0 ¼ 0:24, the s-BP procedure reconstructs exactly down to very low measurement rates. (Details are in Appendix G; the data
is available online [6].)
M ¼ N and K ¼ 0 N. The enumeration method solves continuous distribution. A general and detailed discussion
the compressed-sensing problem in the regime where of information-theoretically optimal reconstruction has
measurement rates, are at least as large as the signal been developed recently in [8–10].
density, 0 , but in a time which grows exponentially In order to design practical ‘‘low-complexity’’ recon-
with N, making it totally impractical. Therefore, ¼ 0 struction algorithms, it has been proposed
P[3,4] to find a
is the fundamental reconstruction limit for perfect vector x which has the smallest ‘1 norm, N i¼1 jxi j, within
reconstruction in the noiseless case, when the nonzero the subspace of vectors which satisfy the constraints
components of the signal are real numbers drawn from a y ¼ Fx, using efficient linear-programming techniques.
021005-2
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
In order to measure the performance of this strategy, one innovative design of the measurement matrix inspired
can focus on the measurement matrix F generated ran- by the theory of crystal nucleation in statistical physics
domly, with independent Gaussian-distributed matrix and from recent developments in coding theory [18–21].
elements of mean zero and variance 1=N, and the signal Some previous works on compressed sensing have
vector s having density 0 < 0 < 1. The analytic study of used these ingredients separately. In particular, adapta-
the ‘1 reconstruction in the thermodynamic limit N ! 1 tions of belief propagation have been developed for the
can be done using either geometric or probabilistic meth- compressed-sensing reconstruction, both in the context
ods [5,11], or with the replica method [12–14]. The analy- of ‘1 reconstruction [11,22,23] and in a probabilistic
sis shows the existence of a sharp phase transition at a approach [24]. It is the combined use of these three
value ‘1 ð0 Þ. When is larger than this value, the ‘1 ingredients that allows us to reach the ¼ 0 limit.
reconstruction gives the exact result x ¼ s with probability
going to 1 in the large N limit; when < ‘1 ð0 Þ, the
III. A PROBABILISTIC APPROACH
probability that it gives the exact result goes to zero as N
grows. As shown in Fig. 2, ‘1 ð0 Þ > 0 and therefore the For the purpose of our analysis, we consider the case
‘1 reconstruction is suboptimal: It requires more measure- where the signal s has independent
Q identically distributed
ments than would be absolutely necessary, in the sense (iid) components: P0 ðsÞ ¼ N i¼1 ½ð1 0 Þðsi Þ þ 0 0 ðsi Þ,
that, if one were willing to do brute-force combinatorial with 0 < 0 < 1. In the large-N limit, the number of non-
optimization, no more than 0 N measurements would be zero components is 0 N. Our approach handles general
necessary. distributions 0 ðsi Þ.
We introduce a new measurement and reconstruction Instead of using a minimization procedure, we shall
approach, designated the seeded belief propagation adopt a probabilistic approach. We introduce a probability
(s-BP), that allows one to reconstruct the signal by a ^
measure PðxÞ over vectors x 2 RN which isQthe restriction
practical method that needs only 0 N measurements. of the Gauss-Bernoulli measure PðxÞ ¼ N i¼1 ½ð1 Þ
We shall now discuss s-BP’s three ingredients: (1) a ðxi Þ þ ðxi Þ to the subspace jy Fxj ¼ 0 [25]. In
probabilistic approach to signal reconstruction, (2) a this paper, we use a distribution ðxÞ, which is a
message-passing algorithm adapted from belief propaga- Gaussian with mean x and variance 2 , but other choices
tion [15], which is a procedure known to be efficient in for ðxÞ are possible. It is crucial to note that we do not
various hard computational problems [16,17], and (3) an require a priori knowledge of the statistical properties of
FIG. 2. Phase diagrams for compressed-sensing reconstruction for two different signal distributions. On the left, the 0 N nonzero
components of the signal are independent Gaussian random variables with zero mean and unit variance. On the right, they are
independent 1 variables. The measurement rate is ¼ M=N. On both sides, we show, from top to bottom: (a) The phase transition
‘1 for ‘1 reconstruction [5,11,12] (which does not depend on the signal distribution). (b) The phase transition EM-BP for EM-BP
reconstruction based for both sides on the probabilistic model with Gaussian . (c) The data points which are numerical-reconstruction
thresholds obtained with the s-BP procedure with L ¼ 20. The point gives the value of where exact reconstruction was obtained in
50% of the tested samples; the top of the error bar corresponds to a success rate of 90%; the bottom of the bar to a success of 10%. The
shrinking of the error bar with increasing N gives numerical support to the existence of the phase transition that we have studied
analytically. These empirical reconstruction thresholds of s-BP are quite close to the ¼ 0 optimal line, and they move closer to it
with increasing N. The parameters used in these numerical experiments are detailed in Appendix E. (d) The line ¼ 0 that is the
theoretical reconstruction limit for signals with continuous 0 . An alternative presentation of the same data using the convention of
Donoho and Tanner [5] is shown in Appendix F.
021005-3
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
the signal: We use a value of not necessarily equal to 0 , achieved with probability 1 in the large-N limit if and
and the that we use is not necessarily equal to 0 . The only if > EM-BP . Using the asymptotic replica
important point is to use < 1 (which reflects the fact that analysis, as explained below, we have computed the
one searches for a sparse signal). line EM-BP ð0 Þ when the elements of the M N
Assuming that F is a random matrix, either where all the measurement matrix F are independent Gaussian ran-
elements are drawn as independent Gaussian random var- dom variables with zero mean and variance 1=N and
iables with zero mean and the same variance or are of the the signal components are iid. The location of this
carefully designed type of ‘‘seeding matrices’’ described transition line does depend on the signal distribution
below, we demonstrate in Appendix A that, for any (see Fig. 2), contrary to the location of the ‘1 phase
0 -dense original signal s and any > 0 , the probability transition.
^
PðsÞ of the original signal goes to 1 when N ! 1. This Notice that our analysis is fully based on the case when
result holds independent of the distribution 0 of the the probabilistic model has a Gaussian . Not surprisingly,
original signal, which does not need to be known. In EM-BP performs better when ¼ 0 (see the left panel of
practice, we see that s also dominates the measure when Fig. 2), where EM-BP provides a sizable improvement
N is not very large. In principle, sampling configurations x over ‘1 . In contrast, the right panel of Fig. 2 shows an
proportionally to the restricted Gauss-Bernoulli measure adversary case when we use a Gaussian to reconstruct a
^
PðxÞ thus gives asymptotically the exact reconstruction in binary 0 ; in this case, there is nearly no improvement
the whole region > 0 . This idea stands at the root of our over the ‘1 reconstruction.
approach and is at the origin of the connection with statis-
tical physics (where one samples with the Boltzmann
measure). V. DESIGNING SEEDING MATRICES
IV. SAMPLING WITH EXPECTATION- In order for the EM-BP message-passing algorithm to be
MAXIMIZATION BELIEF PROPAGATION able to reconstruct the signal down to the theoretically
optimal number of measurements ¼ 0 , one needs to
The exact sampling from a distribution such as PðxÞ,^ use a special family of measurement matrices F that we
Eq. (A2), is known to be computationally intractable [26]. call ‘‘seeding matrices.’’ If one uses an unstructured F, for
However, an efficient approximate sampling can be per- instance, a matrix with independent Gaussian-distributed
formed using a message-passing procedure that we now random elements, EM-BP samples correctly at large . At
describe [24,27–29]. We start from the general belief- small enough , however, a metastable state appears in the
propagation formalism [17,30,31]: For each measurement ^
measure PðxÞ. The EM-BP algorithm is trapped in this
¼ 1; . . . ; M and each signal component i ¼ 1; . . . ; N, state and is therefore unable to find the original signal
one introduces a ‘‘message,’’ mi! ðxi Þ, which is the
(see Fig. 3), just as a supercooled liquid gets trapped in a
probability of xi in a modified measure where measure- glassy state instead of crystallizing. It is well known in
ment has been erased. In the present case, the canonical
crystallization theory that the crucial step is to nucleate a
belief-propagation equations relating these messages can
large enough seed of crystal. This is the purpose of the
be simplified [11,22–24,32] into a closed form that uses
ðtÞ following design of F.
only the expectation ai! and the variance vðtÞ
i! of the We divide the N variables into L groups of N=L
ðtÞ
distribution mi! ðxi Þ (see Appendix B). An important variables, and the M measurements into L groups. The
ingredient that we add to this approach is the learning number of measurements in the p-th group is Mp ¼
P
of the parameters in PðxÞ: the density , and the mean x p N=L, so that M ¼ ½ð1=LÞ Lp¼1 p N ¼ N. We
and variance 2 of the Gaussian distribution ðxÞ. These then choose the matrix elements Fi independently in
are three parameters to be learned using update equations
such a way that, if i belongs to group p and to group
based on the gradient of the so-called Bethe free entropy,
q, then Fi is a random number chosen from the normal
in a way analogous to the expectation maximization
[33–35]. This leads to the expectation-maximization- distribution with mean zero and variance Jq;p =N (see
belief-propagation (EM-BP) algorithm that we will use Fig. 4). The matrix Jq;p is an L L coupling matrix
in the following discussion for reconstruction in com- (and the standard compressed-sensing matrices are ob-
pressed sensing. It consists of iterating the messages tained using L ¼ 1 and 1 ¼ ). Using these new ma-
and the three parameters, starting from random messages trices, one can shift the BP phase transition very close to
ð0Þ
ai! and vð0Þi! , until a fixed point is obtained. Perfect the theoretical limit. In order to get an efficient recon-
reconstruction is found when the messages converge to struction with message passing, one should use a large
the fixed point where ai! ¼ si and vi! ¼ 0. enough 1 . With a good choice of the coupling matrix
Like the ‘1 reconstruction, the EM-BP reconstruction Jp;q , the reconstruction first takes place in the first block
also has a phase transition. Perfect reconstruction is and then propagates as a wave in the following blocks
021005-4
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
p ¼ 2; 3; . . . , even if their measurement rate p is small. nucleation, where a crystal is growing from its seed (see
In practice, we use 2 ¼ . . . ¼ L ¼ 0 , so that the total Fig. 5). Similar ideas have been used recently in the
measurement rate is ¼ ½1 þ ðL 1Þ0 =L. The design of sparse coding matrices for error-correcting
whole reconstruction process is then analogous to crystal codes [18–21].
FIG. 4. Construction of the measurement matrix F for seeded FIG. 5. Evolution of the mean-squared error at different times
compressed sensing. The elements of the signal vector are split as a function of the block index for the s-BP algorithm. Exact
into L (here, L ¼ 8) equal-sized blocks; the number of mea- reconstruction first appears in the left block whose rate 1 >
surements in each block is Mp ¼ p N=L (here, 1 ¼ 1, p ¼ EM-BP allows for seeded nucleation. It then propagates gradu-
0:5 for p ¼ 2; . . . ; 8). The matrix elements Fi are chosen as ally block by block, driven by the free-entropy difference be-
random Gaussian variables with variance Jq;p =N if variable i is tween the metastable and equilibrium states. After little more
in the block p and measurement is in the block q. In the s-BP than 300 iterations, the whole signal of density 0 ¼ 0:4 and size
algorithm, we use Jp;q ¼ 0, except for Jp;p ¼ 1, Jp;p1 ¼ J1 , N ¼ 50 000 is exactly reconstructed well inside the zone forbid-
and Jp1;p ¼ J2 . Good performance is typically obtained with den for BP (see Fig. 2). Here, we used L ¼ 20, J1 ¼ 20, J2 ¼
relatively large J1 and small J2 . 0:2, 1 ¼ 1:0, ¼ 0:5.
021005-5
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
VI. ANALYSIS OF THE PERFORMANCE OF THE the seeding matrix F by choosing 1 , L, and Jp;q in such a
SEEDED BELIEF PROPAGATION PROCEDURE way that the convergence to exact reconstruction is as fast
as possible. In Fig. 3, we show the convergence time of
The s-BP procedure is based on the joint use of seeding- s-BP predicted by the replica theory for different sets of
measurement matrices and of the EM-BP message-passing parameters. For optimized values of the parameters, in the
reconstruction. We have studied it with two methods: limit of a large number of blocks L and large system sizes
direct numerical simulations and analysis of the perform- N=L, s-BP is capable of exact reconstruction close to the
ance in the large-N limit. The analytical result is obtained smallest possible number of measurements, ! 0 . In
by a combination of the ‘‘replica’’ method and the practice, finite-size effects slightly degrade this asymptotic
‘‘cavity’’ method (also known as ‘‘density evolution’’ or threshold saturation, but the s-BP algorithm nevertheless
‘‘state evolution’’). The replica method is a standard reconstructs signals at rates close to the optimal one re-
method in statistical physics [7], which has been applied gardless of the signal distribution, as illustrated in Fig. 2.
successfully to several problems of information theory We illustrate the evolution of the s-BP algorithm in
[17,28,36,37], including compressed sensing [12–14]. It Fig. 5. The nonzero signal elements are Gaussian with
can be used to compute the free-entropy function asso- zero mean, unit variance, density 0 ¼ 0:4, and measure-
ciated with the probability PðxÞ^ (see Appendix D). The ment rate ¼ 0:5, which is deep in the glassy region
cavity method shows that the dynamics of the message- where all other known algorithms fail. The s-BP algorithm
passing algorithm is a gradient dynamics leading to a first nucleates the native state in the first block and then
maximum of this free entropy. propagates it through the system. We have also tested the
When applied to the usual case of the full F matrix with s-BP algorithm on real images where the nonzero compo-
independent Gaussian-distributed elements (case L ¼ 1), nents of the signal are far from Gaussian, and the results are
the replica computation shows that the free-entropy func- nevertheless very good, as shown in Fig. 1. This illustration
tion ðDÞ for configurations constrained to be at a mean- shows that the quality of the result is not due to a good
squared distance D has a global maximum at D ¼ 0 when guess of PðxÞ. It is also important to mention that the gain
> 0 , which confirms that the Gauss-Bernoulli probabi- in performance in using seeding-measurement matrices is
listic reconstruction is in principle able to reach the optimal really specific to the probabilistic approach: We have
compression limit ¼ 0 . However, for EM-BP > > computed the phase diagram of ‘1 minimization with these
0 , where EM-BP is a threshold that depends on the signal matrices and found that, in general, the performance is
and on the distribution PðxÞ, a secondary local maximum slightly degraded with respect to the phase diagram of
of ðDÞ appears at D > 0 (see Fig. 3). In this case, the the full measurement matrices, in the large N limit. This
EM-BP algorithm converges instead to this secondary result demonstrates that it is the combination of the proba-
maximum and does not reach exact reconstruction. The bilistic approach, the message-passing reconstruction with
threshold EM-BP is obtained analytically as the smallest parameter learning, and the seeding design of the measure-
value of such that ðDÞ is decreasing (Fig. 2). This ment matrix that is able to reach the best possible
theoretical study has been confirmed by numerical mea- performance.
surements of the number of iterations needed for EM-BP to
reach its fixed point (within a given accuracy). This con-
vergence time of EM-BP to the exact reconstruction of the VII. PERSPECTIVES
signal diverges when ! EM-BP (see Fig. 3). For < The seeded compressed-sensing approach introduced
EM-BP the EM-BP algorithm converges to a fixed point here is versatile enough to allow for various extensions.
with strictly positive mean-squared error (MSE). This One aspect worth mentioning is the possibility to write the
‘‘dynamical’’ transition is similar to the one found in the EM-BP equations in terms of N messages instead of the
cooling of liquids which go into a supercooled glassy state M N parameters as described in Appendix C. This is
instead of crystallizing, and it appears in the decoding of basically the step that goes from the relaxed-BP [24] to the
error-correcting codes [16,17] as well. approximate-message-passing (AMP) [11] algorithm. It
We have applied the same technique to the case of could be particularly useful when the measurement matrix
seeding-measurement matrices (L > 1). The cavity has some special structure, so that the measurements y can
method allows us to analytically locate the dynamical be obtained in many fewer than M N operations (typi-
phase transition of s-BP. In the limit of large N, the cally in N logN operations). We have also checked that the
MSE Ep and the variance messages Vp in each block approach is robust enough for the introduction of a small
p ¼ 1; . . . L, the density , the mean x, and the variance amount of noise in the measurements (see Appendix H).
2 of PðxÞ all evolve according to a dynamical system Finally, let us mention that, in the case where a priori
which can be computed exactly (see Appendix E). One can information on the signal is available, it can be incorpo-
see numerically if this dynamical system converges to the rated in this approach through a better choice of , which
fixed point corresponding to exact reconstruction (Ep ¼ 0 can considerably improve the performance of the algo-
for all p). This study can be used to optimize the design of rithm. For signal with density 0 , the worst case, which
021005-6
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
we addressed here, is when the nonzero components of the We introduce the constrained partition function:
signal are drawn from a continuous distribution. Better ZY N
performance can be obtained with our method if these YðD;Þ ¼ ðdxi ½ð1 Þðxi Þ þ ðxi ÞÞ
nonzero components come from a discrete distribution i¼1
and one uses this distribution in the choice of . Another Y
M X X
N
interesting direction in which our formalism can be ex- Fi ðxi si Þ 1 ðxi si Þ2 > ND ;
tended naturally is the use of nonlinear measurements and ¼1 i i¼1
different types of noises. Altogether, this approach turns (A3)
out to be very efficient for both random and structured data,
as illustrated in Fig. 1, and offers an interesting perspective and the corresponding ‘‘free-entropy density’’: YðD; Þ ¼
for practical compressed-sensing applications. Data and limN!1 EF Es logYðD; Þ=N. The notations EF and Es de-
code are available online at [6]. note, respectively, the expectation value with respect to F
and to s. 1 denotes an indicator function, equal to one if its
ACKNOWLEDGMENTS argument is true, and equal to zero otherwise.
The proof of optimality is obtained by showing that,
We thank Y. Kabashima, R. Urbanke, and especially A. under the conditions above, lim!0 YðD; Þ=½ð 0 Þ
Montanari for useful discussions. This work has been logð1=Þ is finite if D ¼ 0 (statement 1), and it vanishes if
supported in part by the EC Grant ‘‘STAMINA’’, D > 0 (statement 2). This proves that the measure P^ is
No. 265496, and by the Grant DySpaN of ‘‘Triangle de dominated by D ¼ 0, i.e., by the neighborhood of the
la Physique.’’ signal xi ¼ si . The standard ‘‘self-averageness’’ property,
Note added.—During the review process for our paper, which states that the distribution (with respect to the choice
we became aware of the work [38], in which the authors of F and s) of logYðD; Þ=N concentrates around YðD; Þ
give a rigorous proof of our result that the threshold ¼ when N ! 1, completes the proof. We give here the main
0 can be reached asymptotically by the s-BP procedure. lines of the first two steps of the proof.
We first sketch the proof of statement 2. The fact that
APPENDIX A: PROOF OF THE OPTIMALITY OF lim!0 YðD; Þ=½ð 0 Þ logð1=Þ ¼ 0 when D > 0 can
THE PROBABILISTIC APPROACH be derived by a first moment bound:
Here we give the main lines of the proof that our 1
Y ðD; Þ lim Es logYann ðD; Þ; (A4)
probabilistic approach is asymptotically optimal. We con- N!1 N
sider the case where the signal s has iid components where Yann ðD;Þ is the ‘‘annealed partition function’’
Y
N defined as
P0 ðsÞ ¼ ½ð1 0 Þðsi Þ þ 0 0 ðsi Þ; (A1) Yann ðD; Þ ¼ EF YðD; Þ: (A5)
i¼1
In order to evaluate Yann ðD; Þ, one can first compute the
with 0 < 0 < 1. And we study the probability distribution annealed partition function in which the distances between
1Y N x and the signal are fixed in each block. More precisely, we
^
PðxÞ ¼ fdx ½ð1 Þðxi Þ þ ðxi Þg define
Z i¼1 i
X Z Y
N
YM Zðr1 ; ; rL ; Þ ¼ EF fdxi ½ð1 Þðxi Þ þ ðxi Þg
Fi ðxi si Þ ; (A2) i¼1
¼1 i
Y
M X
with a Gaussian ðxÞ of mean zero and unit variance. We Fi ðxi si Þ
¼1 i
stress here that we consider general 0 , i.e., 0 is not
Y
necessarily equal to ðxÞ (and 0 is not necessarily equal L
L X
to ). The measurement matrix F is composed of iid rp ðx si Þ :
2
p¼1 N i2Bp i
elements Fi such that, if belongs to block q and i
By noticing that the M random variables a ¼
belongs to block p, then Fi is a random number generated P
from the Gaussian distribution with zero mean and vari- i Fi ðxi si Þ are independent Gaussian random varia-
ance Jq;p =N. The function ðxÞ is a centered Gaussian bles, one obtains
distribution with variance 2 . YL N =2
1X L
p
We show that, with probability going to 1 in the large N Zðr1 ; ;rL ;Þ ¼ 2 2 þ Jpq rq
L q¼1
limit (at fixed ¼ M=N), the measure P^ (obtained with a p¼1
021005-7
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
Q
where i2V 1 ðsi Þ> 0, the integral over the variables ui in
Z Y n (A11) is strictly positive in the limit ! 0. The divergence
1
c ðrÞ ¼ lim log ðdxi ½ð1 Þðxi Þ of Yð0; Þ in the limit ! 0 is due to the explicit term
n!1 n
i¼1 exp½Nð0 Þ log in Eq. (A11).
1X n The fact that all eigenvalues of M are strictly positive is
þ ðxi ÞÞ r ðxi si Þ2 : (A7) well known in the case of L ¼ 1 where the spectrum has
n i¼1
been obtained by Marčenko and Pastur [39]. In general, the
The behavior of c ðrÞ is easily obtained by standard saddle- fact that all the eigenvalues of M are strictly positive is
point methods. In particular, when r ! 0, one has c ðrÞ ’ equivalent to saying that all the lines of the N 0 N
1
2 0 logr. matrix F (which is the restriction of the measurement
Using (A6), we obtain, in the small limit: matrix to columns with nonzero signal components) are
1 linearly independent. In the case of seeding matrices with
lim Es logYann ðD;Þ general L, this statement is basically obvious by construc-
N!1 N
tion of the matrices, in the regime where, in each block q,
0 1 X
L
1X L
p 1X L
q > 0 and Jqq > 0. A more formal proof can be obtained
¼ max logrp log 2 þ Jpq rq ;
r1 ; ;rL 2 L p¼1 L p¼1 2 L q¼1 as follows. We consider the Gaussian integral
Z Y
(A8)
n
1X vX 2
ZðvÞ ¼ di exp Fi Fj i j þ i :
i¼1 2 i;j; 2 i
where the maximum over r1 ; . . . ; rL is to be taken under the
constraint r1 þ . . . þ rL > LD. Taking the limit of ! 0 (A12)
with a finite D, at least one of the distance rp must remain This quantity is finite if and only if v is smaller than the
finite. It is then easy to show that smallest eigenvalue, min , of M. We now compute the
annealed average Zann ðvÞ ¼ EF ZðvÞ. If Zann ðvÞ is finite,
1 1
lim Es logYann ðD;Þ ¼ logð1=Þ 0 ð20 0 Þ ; then the probability that min v goes to zero in the large
N!1 N L
N limit. Using methods similar to the one above, one can
(A9) show that
where 0 is the fraction of measurements in blocks 2 to L. XL
2L
As 0 > 0 , this is less singular than logð1=Þð 0 Þ, logEF ZðvÞ ¼ max ðlogrp þ vrp Þ
n r1 ;...;rp
p¼1
which proves statement 2.
X
On the contrary, when D ¼ 0, we obtain from the same L
p 1X
analysis log 1 þ Jpq rq : (A13)
p¼1 0 L q
1
lim Es logYann ðD; Þ ¼ logð1=Þð 0 Þ: (A10) The saddle-point equations
N!1 N
where si ¼ 0, jV 0 j ¼ Nð1 0 Þ, and V 1 as the subset have a solution at v ¼ P 0 (and by continuity also at v > 0
of indices i where si 0, jV 1 j ¼ N0 . We obtain a lower small enough), when L1 Lq¼1 0q ¼ 0 > 1. (It can be found,
bound on Yð0; Þ by substituting PðxÞ with the factors for instance, by iteration.) Therefore, 2L
n logEF ZðvÞ is finite
ð1 Þðxi Þ when i 2 V 0 and with ðxi Þ when for some v > 0 small enough, and therefore min > 0.
i 2 V 1 . This gives
Yð0; Þ > expfN½ð1 0 Þ logð1 Þ þ 0 logðÞ APPENDIX B: DERIVATION OF EXPECTATION-
ð=2Þ logð2Þ þ ð0 Þ logÞg MAXIMIZATION BELIEF PROPAGATION
Z Y X In this and the next sections, we present the message-
dui ðsi þ ui Þ exp 2
1
Mij ui uj ; passing algorithm that we used for reconstruction in com-
i2V 1 i;j2V 1 pressed sensing. In this section, we derive its message-
(A11) passing form, where OðNMÞ messages are being sent
PN between each signal component i and each measurement
where Mij ¼ ¼1 Fi Fj . The matrix M, of size 0 N . This algorithm was used in [24], where it was called the
0 N, is a Wishart-like random matrix. For > 0 , generi- relaxed belief propagation, as an approximate algorithm
cally, its eigenvalues are strictly positive, as we show for the case of a sparse measurement matrix F. In the case
below. Using this property, one can show that, if that we use here of a measurement matrix which is not
021005-8
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
sparse (that is, a finite fraction of the elementspofffiffiffiffi F is posterior probability of x after the measurement of y is
nonzero, and all the nonzero elements scale as 1= N ), the given by
algorithm is asymptotically exact. We show here for com-
pleteness how to derive it. In the next section, we then
1Y N
derive asymptotically equivalent equations that depend ^
PðxÞ ¼ ½ð1 Þðxi Þ þ ðxi Þ
only on OðNÞ messages. In statistical physics terms, this Z i¼1
corresponds to the TAP equations [27] with the Onsager Y
M
1 PN
qffiffiffiffiffiffiffiffiffiffiffiffiffi eð1=2 Þðy i¼1 Fi xi Þ ;
2
reaction term, which are asymptotically equivalent to the (B1)
BP on fully connected models. In the context of com- ¼1 2
pressed sensing, this form of equations has been used
previously [11], and it is called approximate message
passing (AMP). In cases when the matrix F can be com- where Z is a normalization constant (the partition function)
puted recursively (e.g., via fast Fourier transform), the and is the variance of the noise in measurement . The
running time of the AMP-type message passing is noiseless case is recovered in the limit ! 0. The
OðN logNÞ (compared to the OðNMÞ for the non-AMP optimal estimate, which minimizes the MSE with respect
form). Apart from this speeding up, both classes of mes- to the original signal s, is obtained from averages of xi with
sage passing give the same performance. ^
respect to the probability measure PðxÞ. Exact computation
We derive here the message-passing algorithm in the of these averages would require exponential time; belief
case where measurements have additive Gaussian noise; propagation provides a standard approximation. The ca-
the noiseless case limit is easily obtained at the end. The nonical BP equations for probability measure PðxÞ ^ read
P
1 Z Y ð1=2 Þð Fj xj þFi xi y Þ2
m!i ðxi Þ ¼ mj! ðxj Þdxj e ji
; (B2)
Z!i jðiÞ
1 Y
mi! ðxi Þ ¼ ½ð1 Þðxi Þ þ ðxi Þ m
!i ðxi Þ; (B3)
Zi!
!i
whereRZ and Zi!Rare normalization factors ensuring with variance of Oð1=NÞ. Using the Hubbard-Stratonovich
that dxi m!i ðxi Þ ¼ dxi mi! ðxi Þ ¼ 1. These are inte- transformation [40]
gral equations for probability distributions that are still
practically intractable in this form. We can, however, 1 Z
eð! =2Þ ¼ pffiffiffiffiffiffiffiffiffiffi d eð =2Þþði !=Þ
2 2
(B4)
take advantage of the fact that, after we properly rescale 2
the linear system y ¼ Fx in such a way that elements of y P
and x are of Oð1Þ, the matrix Fi has random elements for ! ¼ ð ji Fj xj Þ, we can simplify Eq. (B2) as follows:
1 Z YZ
ð1=2 ÞðFi xi y Þ2 ð 2 =2 Þ Fj xj = ðy Fi xi þi Þ
m!i ðxi Þ ¼ qffiffiffiffiffiffiffiffiffiffiffiffiffi e d e dx m ðx
j j! j Þe : (B5)
Z!i 2 ji
Now we expand the last exponential around zero; because the term Fj is small in N, we keep all terms that are of Oð1=NÞ.
Introducing means and variances as new messages
Z
ai!
dxi xi mi! ðxi Þ; (B6)
Z
vi!
dxi x2i mi! ðxi Þ a2i! ; (B7)
we obtain
021005-9
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
X X
Performing the Gaussian integral over , we obtain
ai! ¼ fa A
!i ; B
!i ;
1
m!i ðxi Þ ¼ eðxi =2ÞA!i þB!i xi ;
2
X X (B15)
Z ~!i
sffiffiffiffiffiffiffiffiffiffiffi (B9) a i ¼ fa A
!i ; B
!i ;
2 B2!i =2A!i
Z~!i ¼ e ;
A!i
X X
where we introduce vi! ¼ fc A
!i ; B
!i ;
2
X X (B16)
Fi
A!i ¼ P ; (B10) v i ¼ fc A
!i ; B
!i ;
þ 2
Fj vj!
ji
where the ai and vi are the mean and variance of the
P marginal probabilities of variable xi .
Fi ðy Fj aj! Þ As we discussed in the main text, the parameters , x,
ji
B!i ¼ P 2 ; (B11) and are usually not known in advance. However, their
þ Fj vj!
ji values can be learned within the probabilistic approach. A
standard way to do so is called expectation maximization
and the normalization Z~!i contains all the xi -independent [33]. One realizes that the partition function
factors. The noiseless case corresponds to ¼ 0. N
ZY
N Y
To close the equations on messages ai! and vi! , we
Zð; x;Þ ¼ dxi ð1 Þðxi Þ
notice that i¼1 i¼1
þ pffiffiffiffiffiffiffi e½ðxi xÞ =2
2 2
1
mi! ðxi Þ ¼ ½ð1 Þðxi Þ þ ðxi Þ 2
Z~i!
P P YM PN
ðxi =2Þ
2 A
!i þxi B
!i 1
qffiffiffiffiffiffiffiffiffiffiffiffiffi eð1=2 Þðy i¼1 Fi xi Þ ;
2
e
: (B12) (B17)
¼1 2
ð1 Þe½ðYþx=
2 Þ2 =2ð1=2 þXÞþðx 2 =22 Þ
2
1 (B19)
þ ; (B13)
ð1=2 þ XÞ1=2
P
Z Y Y 1 ½ðy Fi xi Þ2 =2
Z ¼
dxi mi! ðxi Þ qffiffiffiffiffiffiffiffiffiffiffiffiffi e i ;
ðY þ x=
2 Þ2 i i 2
fc ðX; YÞ ¼ 1 þ
ð1=2 þ XÞ3=2 1=2 þ X
(B20)
ð1 Þe½ðYþx=
2 Þ2 =2ð1=2 þXÞþðx 2 =22 Þ
Z
1 Zi ¼ dxi m!i ðxi Þmi! ðxi Þ: (B21)
þ fa2 ðX; YÞ: (B14)
ð1=2 þ XÞ1=2
The stationarity condition of Bethe free entropy (B18)
Then the closed form of the BP update is with respect to leads to
021005-10
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
P 1=2 þUi
a
2 i
i Vi þx=
¼P ½ðVi þx=
2 Þ2 =2ð1=2 þUi Þðx 2 =22 Þ 1
; (B22)
i ½1 þ ð1=2þU Þ1=2 e
i
P P
where Ui ¼
A
!i , and Vi ¼ Stationarity with respect to x and gives
B
!i .
P
i ai
x ¼ P ; (B23)
½ þ ð1 Þð1= þ Ui Þ e½ðVi þx=
2 1=2 2 Þ2 =2ð1=2 þUi Þþðx 2 =22 Þ 1
i
P
i ðvi þ ai Þ
2
2 ¼ P x 2 : (B24)
i ½ þ ð1 Þð1= þ
2
Ui Þ1=2 e½ðVi þx=
2 Þ2 =2ð1=2 þUi Þþðx 2 =22 Þ 1
In statistical physics, conditions (B22) are known as ai! ¼ fa ðUi A!i ; Vi B!i Þ
Nishimori conditions [34,36]. In the expectation maximi-
@fa @f
zation equations (B22)–(B24), these conditions are used ’ ai A!i ðUi ; Vi Þ B!i a ðUi ; Vi Þ:
iteratively for the update of the current guess of parame- @X @Y
ters. A reasonable initial guess is init ¼ . The value of (C5)
0 s can also be obtained with a special line of measure- P
To express ! ¼ i Fi ai! , we see that the first correc-
ment consisting of a unit vector; hence, we assume that, tion term has a contribution in Fi3
and can be safely
given an estimate of , the x ¼ 0 s=. In the case where
neglected. On the contrary, the second term has a contri-
the matrix F is random, with Gaussian elements of 2
bution in Fi , which one should keep. Therefore,
zero mean and variance 1=N, we can also usePthe following
¼1 y =N ¼ X ðy ! Þ X 2 @fa
equation for learning the variance: M 2
0 hs i ¼ ð þ x Þ.
2 2 2 ! ¼ Fi fa ðUi ; Vi Þ F ðU ; V Þ:
i þ
i i @Y i i
(C6)
APPENDIX C: AMP FORM
OF THE MESSAGE PASSING The computation of
is similar; it gives:
In the large N limit, the messages ai! and vi! are X X ðy ! Þ @fc
¼ 2
Fi vi Fi 3
ðU ; V Þ
nearly independent of , but one must be careful to keep i i þ
@Y i i
the correcting Onsager reaction terms. Let us define X
’ Fi2 f ðU ; V Þ:
c i i (C7)
X X i
! ¼ Fi ai! ;
¼ 2 v
Fi i! ; (C1)
i i For a known form of matrix F, these equations can be
slightly simplified further by using the assumptions of the
X X BP approach about independence of Fi and BP messages.
Ui ¼ A!i ; Vi ¼ B!i : (C2) This plus a law of large number implies that, for matrix F
with Gaussian entries of zero mean and unit variance, one
can effectively ‘‘replace’’ every Fi2 by 1=N in Eqs. (C3),
Then we have (C4), (C6), and (C7). This leads, for homogeneous or bloc
matrices, to even simpler equations and a slightly faster
X 2
Fi X 2
Fi algorithm.
Ui ¼ ’ ; (C3)
þ
Fi
2
vi! þ
Equations (C3), (C4), (C6), and (C7), with or without
the later simplification, give a system of closed equations.
They are a special form [PðxÞ and hence functions fa , fc
X Fi ðy ! þ Fi ai! Þ are different in our case] of the approximate message
Vi ¼ passing of [11].
þ
Fi
2
vi!
The final reconstruction algorithm for the general mea-
X ðy ! Þ X 1 surement matrix and with the learning of the PðxÞ parame-
’ Fi þ fa ðUi ; Vi Þ Fi
2
: ters can hence be summarized in a schematic way (see
þ
þ
Algorithm 1).
(C4) Note that in practice we use ‘‘damping.’’ At each update,
the new message is obtained as u times the old value plus
We now compute ! : 1 u times the newly computed value, with a damping
021005-11
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
Algorithm 1. EM-BPðy ; Fi ; criterium; tmax Þ messages. The reported initialization was used to obtain
Initialize randomly messages Ui from interval [0,1] for the results in Fig. 1. However, other initializations are
every component; possible. Note also that, for specific classes of signals or
Initialize randomly messages Vi from interval ½1; 1 for measurement matrices, the initial conditions may be
every component; adjusted to take into account the magnitude of the values
Initialize messages ! y ; Fi and y .
Initialize randomly messages
from interval [0,1] for For a general matrix F, one iteration takes OðNMÞ steps.
every measurement; We have observed the number of iterations needed for
Initialize the parameters , x 0, 2 1. convergence to be basically independent of N; however,
conv criterium þ 1; t 0; the constant depends on the parameters and the signal (see
while conv > criterium and t < tmax : Fig. 3). For matrices that can be computed recursively (i.e.,
do t t þ 1; without storing all their NM elements), a speeding-up is
for each component i: possible, as the message-passing loop takes only OðM þ
do xold
i fa ðUi ; Vi Þ; NÞ steps.
Update Ui according to Eq. (C3).
Update Vi according to Eq. (C4).
for each measurement : APPENDIX D: REPLICA ANALYSIS AND DENSITY
do Update ! according to Eq. (C6). EVOLUTION—FULL MEASUREMENT MATRIX
Update
according to Eq. (C7).
Averaging over disorder leads to replica equations that
Update according to Eq. (B22).
Update x and 2 according to Eq. (B24).
describe the N ! 1 behavior of the partition function as
for each component i: well as the density evolution of the belief-propagation
do xi fa ðUi ; Vi Þ; algorithm. The replica trick evaluates EF;s ðlogZÞ via
conv meanðjxi xoldi jÞ;
return signal components x 1 1 EðZn Þ 1
¼ EðlogZÞ ¼ lim : (D1)
N N n!0 n
0 < u < 1, typically u ¼ 0:5. For both the update of mes- In the case where the matrix F is the full measurement
sages and learning of parameters, empirically this damping with all elements iid from a normal distribution with zero
speeds up the convergence. Note also that the algorithm is mean and variance unity, one finds that is obtained as the
relatively robust with respect to the initialization of saddle-point value of the function:
(D2)
Here, Dz is a Gaussian integration measure with zero mean and R variance equal to one; 0 is the density of the signal; and
0 ðsÞ is the distribution of the signal components and hs2 i ¼ dss2 0 ðsÞ is its second moment. is the variance of the
measurement noise. The noiseless case is recovered by using ¼ 0.
The physical meaning of the order parameters is
1X 2 1X 2 1X
Q¼ hx i; q¼ hx i ; m¼ s hx i; (D3)
N i i N i i N i i i
whereas the other three m, ^ Q^ are auxiliary parameters. Performing the saddle-point derivative with respect to m, q,
^ q,
Q q, m, ^ and Q^ þ q,
^ q, ^ we obtain the following six self-consistent equations [using the Gaussian form of ðxÞ, with mean
x and variance 2 ]:
q 2m þ 0 hs2 i þ
m^ ¼ ¼ Q^ þ q;
^ Q^ ¼ ; (D4)
þQq þQq ð þ Q qÞ2
021005-12
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
Z Z pffiffiffi
0 ms^ þ z q^ þ x= 2
m¼ Dz dss0 ðsÞ q ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi pffiffi ; (D5)
Q^ þ q^ þ 1=2 ð1 Þ Q þ q^ þ 1= e
^ 2 ðx 2 =22 Þ½ðmsþz
^ q^ þx=
2 Þ2 =2ðQþ
^ qþ1=
^ 2 Þ
þ
Z pffiffiffi
ð1 0 Þ z q^ þ x= 2
Qq¼ p ffiffiffi Dzz q ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi pffiffi
ðQ^ þ q^ þ 1=2 Þ q^ ð1 Þ Q^ þ q^ þ 1=2 eðx =2 Þ½ðz q^ þx=
2 2 2 Þ2 =2ðQþ^ qþ1=
^ 2 Þ
þ
Z Z p ffiffiffi
0 ms^ þ z q^ þ x= 2
þ p ffiffiffi Dzz ds0 ðsÞ q ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi pffiffi ;
ðQ^ þ q^ þ 1= Þ q^
2
ð1 Þ Q^ þ q^ þ 1=2 eðx =2 Þ½ðmsþz
2 2 ^ q^ þx= 2 Þ2 =2ðQþ
^ qþ1=
^ 2 Þ
þ
(D6)
Z pffiffiffi
ð1 0 Þ ðz q^ þ x= 2 Þ2 þ Q^ þ q^ þ 1=2
Q¼ Dz q ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi pffiffi
ðQ^ þ q^ þ 1=2 Þ2 ð1 Þ Q^ þ q^ þ 1=2 eðx =2 Þ½ðz q^ þx=
2 2 2 Þ2 =2ðQþ^ qþ1=
^ 2 Þ
þ
Z Z p ffiffiffi
0 ðms ^ þ z q^ þ x= 2 Þ2 þ Q^ þ q^ þ 1=2
þ Dz ds0 ðsÞ q ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi pffiffi : (D7)
ðQ^ þ q^ þ 1=2 Þ2 ð1 Þ Q þ q^ þ 1= e ^ 2 ðx 2 =22 Þ½ðmsþz
^ q^ þx= 2 Þ2 =2ðQþ
^ qþ1=
^ 2 Þ
þ
We now show the connection between this replica com- BP algorithm. Note also that the density-evolution equa-
putation and the evolution of belief-propagation messages, tions are the same for the message-passing and for the
studying first the case where one does not change the AMP equations. It turns out that the above equations close
parameters , x, and . Let us introduce parameters mBP , ðtÞ
in terms of two parameters: the mean-squared error EBP ¼
qBP , QBP , defined via the belief-propagation messages as: ðtÞ ðtÞ ðtÞ ðtÞ ðtÞ
qBP 2mBP þ 0 hs i and the variance VBP ¼ QBP qBP .
2
1 XN
1 XN
From Eqs. (D4)–(D7) one easily obtains a closed mapping
mðtÞ
BP ¼ aðtÞ s ; qðtÞ
BP ¼ ðaðtÞ Þ2 ;
N i¼1 i i N i¼1 i ðEðtþ1Þ ðtþ1Þ ðtÞ
BP ; VBP Þ ¼ fðEBP ; VBP Þ.
ðtÞ
(D8) In the main text, we defined the function ðDÞ, which is
1 XN
QðtÞ qðtÞ ¼ vðtÞ : the free
P entropy restricted to configurations x for which
BP BP
N i¼1 i D¼ N ðx i si Þ2 =N is fixed. This is evaluated as the
i¼1
The density- (state-)evolution equations for these parame- saddle point over Q, q, Q, ^ q,^ m^ of the function
ters can be derived in the same way as in [11,32], and this ðQ; q; ðQ D þ 0 hs2 iÞ=2; Q; ^ q;
^ mÞ.
^ This function is
leads to the result that mBP , qBP , QBP evolve under the plotted in Fig. 3(a).
update of BP in exactly the same way as according to In presence of expectation-maximization learning of the
iterations of Eqs. (D5)–(D7). Hence, the analytical equa- parameters, the density evolution for the conditions (B22)
tions (D5)–(D7) allow us to study the performance of the and (B24) are
Z Z pffiffiffi
ðtþ1Þ ðtÞ gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
¼ Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ pffiffiffi
1 þ gðQ^ þ q; ^ mx^ 0 þ z q^ Þ
Z Z 1 (D9)
1
Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ pffiffiffi ;
1 þ gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
pffiffiffi
1 Z Z
x ðtþ1Þ ¼ Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þfa ðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
Z Z pffiffiffi 1 (D10)
gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ pffiffiffi ;
1 þ gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
pffiffiffi
1 Z Z pffiffiffi
ð2 Þðtþ1Þ ¼ Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ½fa ðQ^ þ q; ^ 0 þ z q^ Þ2 þ fc ðQ^ þ q;
^ mx ^ 0 þ z q^ Þ
^ mx
Z Z pffiffiffi 1
gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ p ffiffiffi ½xðtþ1Þ 2 : (D11)
1 þ gðQ^ þ q; ^ 0 þ z q^ Þ
^ mx
021005-13
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
The density-evolution equations now provide a can be easily obtained. In fact, the form of the matrix
mapping, that we have used is by no means the only one that can
produce the seeding mechanism, and we expect that
ðEðtþ1Þ ðtþ1Þ
EM-BP ; VEM-BP ;
ðtþ1Þ ðtþ1Þ
;x ; ðtþ1Þ Þ better choices, in terms of convergence time, finite-size
¼ fðEðtÞ ðtÞ ðtÞ ðtÞ
EM-BP ; VEM-BP ; ; x ; ðtÞ Þ; (D12) effects, and sensibility to noise, could be unveiled in the
near future.
obtained by complementing the previous equations on With the matrix presented in this work, and in order to
EðtÞ ðtÞ
EM-BP ; VEM-BP with the update equations (D9)–(D11). obtain the best performance (in terms of phase transition
The next section gives explicitly the full set of equations limit and of speed of convergence), one needs to optimize
in the case of seeding matrices; the ones for the full the value of J1 and J2 depending on the type of signal.
matrices are obtained by taking L ¼ 1. These are the Fortunately, this can be analyzed with the replica method.
equations that we study to describe analytically the evolu- The analytic study in the case of seeding-measurement
tion of the EM-BP algorithm and obtain the phase diagram matrices is in fact done using the same techniques as for
for the reconstruction (see Fig. 2). the full matrix. The order parameters are now the MSE
Ep ¼ qp 2mp þ 0 hs2 i and variance Vp ¼ Qp qp in
APPENDIX E: REPLICA ANALYSIS each block p 2 f1; . . . ; Lg. Consequently, we obtain the
AND DENSITY EVOLUTION FOR final dynamical system of 2L þ 3-order parameters de-
SEEDING-MEASUREMENT MATRICES scribing the density evolution of the s-BP algorithm. The
Many choices of J1 and J2 actually work very well, order parameters at iteration t þ 1 of the message-passing
and good performance for seeding-measurement matrices algorithm are given by:
Z Z qffiffiffiffiffi
Eðtþ1Þ
q ¼ dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ Dzðfa ðQ^ q þ q^ q ; m^ q x0 þ z q^ q Þ x0 Þ2 ; (E1)
Z Z qffiffiffiffiffi
Vqðtþ1Þ ¼ dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ Dzfc ðQ^ q þ q^ q ; m^ q x0 þ z q^ q Þ; (E2)
L Z Z pffiffiffiffiffiffi
ðtþ1Þ 1 X
ðtÞ
gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
¼ Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ p ffiffiffiffiffiffi
L p¼1 1 þ gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
X 1
1 L Z Z 1
Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ p ffiffiffiffiffiffi ; (E3)
L p¼1 1 þ gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
L Z Z qffiffiffiffiffiffi
1 1 X
x ðtþ1Þ ¼ Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þfa ðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
L p¼1
X pffiffiffiffiffiffi 1
1 L Z Z gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
Dz dx0 ½ð1 0 Þðx0 Þ þ 0 0 ðx0 Þ p ffiffiffiffiffi
ffi ; (E4)
L p¼1 1 þ gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
L Z Z qffiffiffiffiffiffi qffiffiffiffiffiffi
1 1 X
ð2 Þðtþ1Þ ¼ Dz dx0 ½ð1 0 Þðx0 Þþ0 0 ðx0 Þ½fa ðQ^ p þ q^ p ; m^ p x0 þz q^ p Þ2 þfc ðQ^ p þq^ p ; m^ p x0 þ z q^ p Þ
L p¼1
X pffiffiffiffiffiffi 1
1 L Z Z gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
Dz dx0 ½ð10 Þðx0 Þ þ 0 0 ðx0 Þ p ffiffiffiffiffi
ffi ½xðtþ1Þ 2 ; (E5)
L p¼1 1 þ gðQ^ p þ q^ p ; m^ p x0 þ z q^ p Þ
where:
1X Jpq p
m^ q ¼
L p þ ð1=LÞ PLr¼1 Jpr VrðtÞ
; (E6)
1X Jpq p 1X
q^ q ¼ PL J EðtÞ
s ; (E7)
L p ½ þ ð1=LÞ r¼1 Jpr Vr L s ps
ðtÞ 2
021005-14
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
1X p Jpq Montanari [38] extends our work and proves our claim that
Q^ q ¼ P q^ q ¼ m^ q q^ q :
L p þ ð1=LÞ Lr¼1 Jpr VrðtÞ s-BP can reach the optimal threshold asymptotically.
Practical numerical implementation of s-BP matches
(E8) this theoretical performance only when the size of every
block is large enough (few hundreds of variables). In
The functions fa ðX; YÞ, fc ðX; YÞ were defined in (B13) and practice, for finite size of the signal, if we want to keep
(B14), and the function g is defined as the block-size reasonable, we are hence limited to values of
L of several dozens. Thus, in practice, we do not quite
1 ðY þ x=
2 Þ2 x 2
gðX; YÞ ¼ p ffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffiffi
ffi exp : (E9) saturate the threshold ¼ 0 , but exact reconstruction is
1 þ X2 2ðX þ 1=2 Þ 22 possible very close to it, as illustrated in Fig. 2, for which
the values that we used for the coupling parameters are
This is the dynamical system that we use in the paper in listed in Table I.
the noiseless case ( ¼ 0) in order to optimize the values In Fig. 3, we also presented the result of the s-BP
of 1 , J1 , and J2 . We can estimate the convergence time of reconstruction of the Gaussian signal of density ¼ 0:4
the algorithm as the number of iterations needed in order to for different values of L. We have observed empirically
reach the successful fixed point (where all Ep and Vp that the result is rather robust for choices of J1 , J2 and 1 .
vanish within some given accuracy). Figure 6 shows the In this case, in order to demonstrate that very different
convergence time of the algorithm as a function of J1 and choices give seeding matrices which are efficient, we used
J2 for Gauss-Bernoulli signals. 1 ¼ 0:7; and then J1 ¼ 1043 and J2 ¼ 104 for
The numerical iteration of this dynamical system is fast. L ¼ 2;J1 ¼ 631 and J2 ¼ 0:1 for L ¼ 5; J1 ¼ 158 and
It allows one to obtain the theoretical performance that can J2 ¼ 4 for L ¼ 10; and J1 ¼ 1000 and J2 ¼ 1 for L ¼
be achieved in an infinite-N system. We have used it, in 20. One sees that a common aspect to all these choices is a
particular, to estimate the values of L, 1 , J1 , J2 that have large ratio J1 =J2 . Empirically, this seems to be important in
good performance. For Gauss-Bernoulli signals, using op- order to ensure a short convergence time. A more detailed
timal choices of J1 , J2 , we have found that perfect recon- study of convergence time of the dynamical system will be
struction can be obtained down to the theoretical limit necessary in order to give some more systematic rules for
¼ 0 by taking L ! 1 (with corrections that scale as choosing the couplings. This study is left for future work.
1=L). Recent rigorous work by Donoho, Javanmard, and Even though our theoretical study of the seeded BP was
performed on an example of a specific signal distribution,
the examples presented in Figs. 1 and 2 show that the
performance of the algorithm is robust and also applies
to images which are not drawn from that signal
distribution.
1 0 J1 J2 L
Gaussian signal
0.1 0.130 0.3 0.121 1600 1.44 20
0.2 0.227 0.4 0.218 100 0.64 20
0.3 0.328 0.6 0.314 64 0.16 20
0.4 0.426 0.7 0.412 16 0.16 20
0.6 0.624 0.9 0.609 4 0.04 20
0.8 0.816 0.95 0.809 4 0.04 20
Binary signal
FIG. 6. Convergence time of the s-BP algorithm with L ¼ 2 as 0.1 0.150 0.4 0.137 64 0.16 20
a function of J1 and J2 (in log scale) for Gauss-Bernoulli signal 0.2 0.250 0.6 0.232 64 0.16 20
with 0 ¼ 0:1. The color represents the number of iterations 0.3 0.349 0.7 0.331 16 0.16 20
such that the MSE is smaller than 108 . The white region gives 0.4 0.441 0.8 0.422 16 0.16 20
the fastest convergence. The measurement density is fixed to 0.6 0.630 0.95 0.613 16 0.16 20
¼ 0:25. The parameters in s-BP are chosen as L ¼ 2 and 0.8 0.820 1 0.811 4 0.04 20
1 ¼ 0:3. The axes are in log10 scale.
021005-15
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
APPENDIX F: PHASE DIAGRAM IN THE TABLE II. Parameters used for the s-BP reconstruction of the
VARIABLES USED BY DONOHO AND TANNER Shepp-Logan phantom image and the Lena image of Fig. 1.
FIG. 7. Same data as in Fig. 2, but using the convention of [5]. The phase diagrams are now plotted as a function of the
undersampling ratio DT ¼ K=M ¼ = and of the oversampling ratio DT ¼ M=N ¼ .
021005-16
STATISTICAL-PHYSICS-BASED RECONSTRUCTION IN . . . PHYS. REV. X 2, 021005 (2012)
TABLE III. Mean-squared error obtained after reconstruction small amount of noise. As shown in Appendix B, the
for the Shepp-Logan phantom image (top) and for the Lena probability of x after the measurement of y is given by
image (bottom), with ‘1 minimization, BP, and s-BP.
FIG. 8. Mean-squared error as a function of for different pffiffiffiffivalues of the measurement noise. Left: Gauss-Bernoulli signal with
0 ¼ 0:2, and measurement noise with standard deviation ¼ 104 . The EM-BP (L ¼ 1) and s-BP strategy (L ¼ 4, J1 ¼ 20,
J ¼ 0:1, 1 ¼ 0:4) are able to perform a very good reconstruction up to much lower value of than the ‘1 procedure. Below a critical
value of , these algorithms show a first-order phase
pffiffiffiffi transition to a regime with much larger MSE. Right: Gauss-Bernoulli signal with
0 ¼ 0:4, with a noise with standard deviation ¼ 103 , 104 , 105 . s-BP, with L ¼ 9, J1 ¼ 30, J2 ¼ 8, 1 ¼ 0:8 decodes very
well for all of these values of noises. In this case, ‘1 is unable to reconstruct for all < 0:75, well outside the range of this plot.
021005-17
KRZAKALA et al. PHYS. REV. X 2, 021005 (2012)
[1] E. J. Candès and M. B. Wakin, An Introduction to [22] D. Donoho, A. Maleki, and A. Montanari, in IEEE
Compressive Sampling, IEEE Signal Process. Mag. 25, Information Theory Workshop (ITW) 2010, doi:10.1109/
21 (2008). ITWKSPS.2010.5503228 (IEEE, New York, 2010), pp. 1–5.
[2] M. F. Duarte, M. A. Davenport, D. Takhar, J. N. Laska, [23] S. Rangan, in Proceedings of the 2011 IEEE International
Ting Sun, K. F. Kelly, and R. G. Baraniuk, Single-Pixel Symposium on Information Theory Proceedings (ISIT),
Imaging via Compressive Sampling, IEEE Signal Process. Austin, Texas (IEEE, New York, 2011), pp. 2168–2172.
Mag. 25, 83 (2008). [24] S. Rangan, in 44th Annual Conference on Information
[3] E. J. Candès and T. Tao, Decoding by Linear Sciences and Systems (CISS), 2010 (IEEE, New York,
Programming, IEEE Trans. Inf. Theory 51, 4203 (2005). 2010), pp. 1–6.
[4] D. L. Donoho, Compressed Sensing, IEEE Trans. Inf. [25] For proper normalization, we regularize the linear con-
straint to eðyFxÞ =ð2Þ =ð2ÞM=2 , where can be seen as
2
Theory 52, 1289 (2006).
[5] D. L. Donoho and J. Tanner, Neighborliness of Randomly a variance of noisy measurement, and consider the limit
Projected Simplices in High Dimensions, Proc. Natl. ! 0; see Appendix A.
Acad. Sci. U.S.A. 102, 9452 (2005). [26] B. K. Natarajan, Sparse Approximate Solutions to Linear
[6] Applying Statistical Physics to Inference in Compressed Systems, SIAM J. Comput. 24, 227 (1995).
Sensing, http://aspics.krzakala.org/. [27] D. J. Thouless, P. W. Anderson, and R. G. Palmer, Solution
[7] M. Mézard, G. Parisi, and M. A. Virasoro, Spin-Glass of ‘‘Solvable Model of a Spin-Glass’’, Philos. Mag. 35, 593
Theory and Beyond, Lecture Notes in Physics, Vol. 9 (1977).
(World Scientific, Singapore, 1987). [28] T. A. Tanaka, A Statistical-Mechanics Approach to Large-
[8] Y. Wu and S. Verdu, MMSE Dimension, IEEE Trans. Inf. System Analysis of CDMA Multiuser Detectors, IEEE
Theory 57, 4857 (2011). Trans. Inf. Theory 48, 2888 (2002).
[9] Y. Wu and S. Verdu, Optimal Phase Transitions in [29] D. Guo and C.-C. Wang, in Information Theory Workshop
Compressed Sensing, arXiv:1111.6822v1. (ITW) 2006, Chengdu (IEEE, New York, 2006), pp. 194–198.
[10] D. Guo, D. Baron, and S. Shamai, in 47th Annual Allerton [30] F. R. Kschischang, B. Frey, and H.-A. Loeliger, Factor
Conference on Communication, Control, and Computing, Graphs and the Sum-Product Algorithm, IEEE Trans. Inf.
Monticello, Illinois (IEEE, New York, 2009), pp. 52–59. Theory 47, 498 (2001).
[11] D. L. Donoho, A. Maleki, and A. Montanari, Message- [31] J. Yedidia, W. Freeman, and Y. Weiss, in Exploring
Passing Algorithms for Compressed Sensing, Proc. Natl. Artificial Intelligence in the New Millennium (Morgan
Acad. Sci. U.S.A. 106, 18914 (2009). Kaufmann, San Francisco, 2003), pp. 239–236.
[12] Y. Kabashima, T. Wadayama, and T. Tanaka, A Typical [32] A. Montanari and M. Bayati, IEEE Trans. Inf. Theory 57,
Reconstruction Limit of Compressed Sensing Based on 764 (2011).
Lp -Norm Minimization, J. Stat. Mech. (2009) L09003. [33] A. Dempster, N. Laird, and D. Rubin, Maximum
[13] S. Rangan, A. Fletcher, and V. Goyal, Asymptotic Analysis Likelihood from Incomplete Data via the EM Algorithm,
of MAP Estimation via the Replica Method and J. R. Stat. Soc. Ser. B. Methodol. 39, 1 (1977) [http://
Applications to Compressed Sensing, arXiv:0906.3234v2. www.jstor.org/stable/2984875].
[14] S. Ganguli and H. Sompolinsky, Statistical Mechanics of [34] Y. Iba, The Nishimori Line and Bayesian Statistics, J.
Compressed Sensing, Phys. Rev. Lett. 104, 188701 (2010). Phys. A 32, 3875 (1999).
[15] J. Pearl, in Proceedings of the American Association of [35] A. Decelle, F. Krzakala, C. Moore, and L. Zdeborová,
Artificial Intelligence National Conference on AI Inference and Phase Transition in the Detection of Modules
(Pittsburgh, 1982), ISBN 978-1-57735-192-4, pp. 133– in Sparse Networks. Phys. Rev. Lett. 107, 065701 (2011).
136. [36] H. Nishimori, Statistical Physics of Spin Glasses and
[16] T. Richardson and R. Urbanke, Modern Coding Theory Information Processing (Oxford University Press,
(Cambridge University Press, Cambridge, England, 2008). Oxford, England, 2001).
[17] M. Mézard and A. Montanari, Information, Physics, and [37] D. Guo and S. Verdú, Randomly Spread CDMA:
Computation (Oxford University, Oxford, England, Asymptotics via Statistical Physics, IEEE Trans. Inf.
2009). Theory 51, 1983 (2005).
[18] A. Jimenez Felstrom and K. Zigangirov, Time-Varying [38] D. L. Donoho, A. Javanmard, and A. Montanari,
Periodic Convolutional Codes with Low-Density Parity- Information-Theoretically Optimal Compressed Sensing
Check Matrix, IEEE Trans. Inf. Theory 45, 2181 via Spatial Coupling and Approximate Message Passing,
(1999). arXiv:1112.0708v1.
[19] S. Kudekar, T. Richardson, and R. Urbanke, in [39] V. A. Marčenko and L. A. Pastur, Distribution of
Proceedings of the 2010 IEEE International Symposium Eigenvalues for Some Sets of Random Matrices,
on Information Theory Proceedings (ISIT), Austin, Texas Mathematics of the USSR-Sbornik 1, 457 (1967) [http://
(IEEE, New York, 2010), pp. 684–688. iopscience.iop.org/0025-5734/1/4/A01/].
[20] M. Lentmaier and G. Fettweis, in Proceedings of the 2010 [40] J. Hubbard, Calculation of Partition Functions, Phys. Rev.
IEEE International Symposium on Information Theory Lett. 3, 77 (1959); R. L. Stratonovich, On a Method of
Proceedings (ISIT), Austin, Texas (IEEE, New York, Calculating Quantum Distribution Functions, Sov. Phys.
2010), pp. 709–713. Dokl. 2, 416 (1958).
[21] S. Hassani, N. Macris, and R. Urbanke, in IEEE [41] The ‘1 -MAGIC tool is a free implementation that can
Information Theory Workshop (ITW) 2010, doi:10.1109/ be downloaded from http://users.ece.gatech.edu/~justin/
CIG.2010.5592881 (IEEE, New York, 2010), pp. 1–5. l1magic/.
021005-18