Working Locally Thinking Globally - Part I: Theoretical Guarantees for Convolutional Sparse Coding
Vardan Papyan, Jeremias Sulam, Michael Elad
I Introduction
Generally, solving a pursuit problem is a computationally challenging task. As a consequence, most such recent successful methods have been deployed on relatively small dimensional signals, commonly referred to as patches. Under this local paradigm, the signal is broken into overlapped blocks and the above defined sparse coding problem is reformulated as
The above discussion suggests that in order to find a consistent global representation for the signal, one should propose a global sparse model. However, employing a general global dictionary is infeasible due to the curse of dimensionality and the complexity involved. An alternative is a global model in which the signal is composed as a superposition of local atoms. The family of dictionaries giving rise to such signals is a concatenation of banded Circulant matrices. This global model benefits from having a local shift invariant structure – a popular assumption in signal and image processing – suggesting an interesting connection to the above-mentioned local modeling.
When the dictionary has this structure of a concatenation of banded Circulant matrices, the pursuit problem in (1) is usually known as convolutional sparse coding . Recently, several works have addressed the problem of using and training such a model in the context of image inpainting, super-resolution, and general image representation . These methods exploit an ADMM formulation in the Fourier domain in order to search for the sparse codes and train the dictionary involved. Several variations have been proposed for solving the pursuit problem, yet there has been no theoretical analysis of their success.
In this two-part work, we consider the following set of questions: Assume a signal is created by multiplying a sparse vector by a global structured dictionary that consists of a union of banded and Circulant matrices. Then,
Can we guarantee the uniqueness of such a global (convolutional) sparse vector?
Can global pursuit algorithms, such as the ones suggested in recent works, be guaranteed to find the true underlying sparse code, and if so, under which conditions?
Can we guarantee a stability of the sparse approximation problem, and a stability of corresponding pursuit methods in a noisy regime?; And
Can we solve the global pursuit by restricting the process to local pursuit operations?
In part I of our work, we focus on answering questions 1 and 2, while questions 3 and 4 are analyzed and answered in the sequel paper.
A naïve approach to address such theoretical questions is to apply the fairly extensive results for sparse representation and compressed sensing to the above defined model . However, as we will show throughout this paper, this strategy provides nearly useless results and bounds from a global perspective. Therefore, there exists a true need for a deeper and alternative analysis of the sparse coding problem in the convolutional case which would yield meaningful bounds.
This paper is organized as follows. We begin by reviewing the unconstrained global (traditional) sparse representation model in Section II, followed by a detailed description of the convolutional structure in Section III. Section IV briefly motivates the need of a thorough analysis of this model, which is then provided in Section V. We introduce additional mathematical tools in Section VI, which provide further insight into the convolutional model. Finally, we conclude and motivate the next part of this work in Section VII.
II The Global Sparse Model – Preliminaries
Several results have shed light on the theoretical aspects of this problem, claiming a unique solution under certain circumstances. These guarantees are given in terms of properties of the dictionary , such as the Spark, defined as the minimum number of linearly dependent columns (atoms) in . Formally,
Based on this property, a solution obeying is necessarily the sparsest one . Unfortunately, this bound is of little practical use, as computing the Spark of a matrix is a combinatorial problem – and infeasible in practice.
Another guarantee is given in terms of the mutual coherence of the dictionary, . This measure quantifies the similarity of atoms in the dictionary, defined in as:
Solving the problem is NP-hard in general. Nevertheless, its solution can be approximated by either greedy pursuit algorithms, such as the Orthogonal Matching Pursuit (OMP) , or convex relaxation approaches like Basis Pursuit (BP) . Despite the difficulty of this problem, these methods (and other similar ones) have been proven to recover the true solution if .
In real world applications, due to noisy measurements and model imperfections, the idealistic setting portrayed above is not directly applicable. Nevertheless, one can extend the model to include signal perturbations, obtaining the following problem:
We will deffer studying this case here, as it will be analyzed in detail in part two of this report.
III The Convolutional Sparse Model
When handling large dimensional signals, using an unstructured dictionary becomes unfeasible. In this section we will enforce a constraint on the global dictionary, resulting in both theoretical and practical benefits.
Consider the global dictionary to be a concatenation of banded Circulant matricesThe choice of Circulant matrices comes to alleviate boundary problems., where each such matrix has a band of width . As such, by simple permutation of its columns, such a dictionary consists of all shifted versions of a local dictionary of size . This model is commonly known as Convolutional Sparse Representation . Hereafter, whenever we refer to the global dictionary , we assume it has this structure. Assume a signal to be generated as . In Figure 1 we describe such a global signal, its corresponding dictionary that is of size and its sparse representation, of length . We note that is built of distinct and independent sparse parts, each of length , which we will refer to as the local sparse vectors . In this section we shall propose several different interpretations of signals emerging from this model, and as we shall see, these will serve us well in the later analysis.
Consider a global sparse vector . Define as its stripe representation.
Note that a stripe can be also seen as a group of adjacent local sparse vectors of length from , centered at location .
Consider a convolutional dictionary defined by a local dictionary of size . Define the stripe dictionary of size , as the one obtained by extracting consecutive rows from , followed by the removal of its zero columns, namely .
Observe that , depicted in Figure 2, is independent of , being the same for all locations due to the union-of-Circulant-matrices structure of . In other words, the shift invariant property is satisfied for this model – all patches share the same stripe dictionary in their construction. Armed with the above two definitions, Equation (7) reads .
From a different perspective, one can synthesize the signal by a different interpretation of the relation , shown in Figure 1. The matrix is a concatenation of vertical stripes of size , where each can be represented as . In other words, the vertical stripe is constructed by taking the small and local dictionary and positioning it in the row. As we have already said, the same partitioning applies to , leading to the ingredients. Thus,
Since play the role of local sparse vectors, are reconstructed patches (which are not the same as ), and the sum above proposes a patch averaging approach as practiced in several papers . This formulation provides another local interpretation of the convolutional model.
Yet a third interpretation of the very same signal construction can be suggested, in which the signal is seen as resulting from a sum of local/small atoms which appear in a small number of locations throughout the signal. This can be formally expressed as
This model (adopting the last, convolutional, interpretation) has received growing attention in recent years in various applications. In a convolutional sparse coding framework was used for pattern detection in images and the analysis of instruments in music signals, while in it was used for the reconstruction of 3D trajectories. The problem of learning the local dictionary was also studied in several works .
IV From Global to Local Analysis
Let us now introduce a measure that will provide a local notion of sparsity within a global sparse vector.
Armed with the above definition, we move now to define the problem:
When dealing with a global signal, instead of solving the problem (defined in Equation (3)) as is commonly done, we aim to solve the above defined objective instead. The key difference is that we are not limiting the overall number of zeros in , but rather putting a restriction on its local density.
IV-B Global versus Local Bounds
As mentioned previously, theoretical bounds are often given in terms of the mutual coherence of the dictionary. In this respect, a lower bound on this value is much desired. In the case of our convolution sparse model, this value quantifies not only the correlation between the atoms in , but also the correlation between their shifts. Though in a different context, a bound for this value was derived in , and it is given by
For example, if (one local atom with all its shifts), this suggests that might be an orthogonal matrix, and thus . Going to the other extreme, for a large value of one obtains that the best possible coherence is – this is a very high value (e.g., if , this coherence bound is ), considering the fact that it characterizes the whole global dictionary. This implies that if we are to apply BP or OMP to recover the sparsest that represents , the classical sparse approximation results would allow merely non-zeros in all , for any , no matter how long is!
As we shall see next, the situation is not as grave as may seem, due to our migration from to . Leveraging on the definitions from the previous subsection, we will provide recovery guarantees that will have a local flavor, and the bounds will be given in terms of the number of non-zeros in the densest stripe. This way, we will show that the guarantee conditions can be significantly enhanced to non-zeros locally rather than globally.
V Theoretical Study
As motivated in the previous section, the concerns of uniqueness, recovery guarantees and stability of sparse solutions in the convolutional case require special attention. We now formally address these questions by closely following the path taken in , carefully generalizing each and every statement to the global-local model discussed here.
Before proceeding onto theoretical grounds, we briefly summarize, for the convenience of the reader, all notations used throughout this work in Table V. Note the unorthodox choice of capital letters for global vectors and lowercase for local ones.
As can be seen from these results, the theoretical bound is far from being tight. However, in the traditional sparse representation model the corresponding bounds have the same loose flavor . This kind of results is in fact expected when using such a worst-case analysis. Tighter bounds could likely be obtained by a probabilistic study, which we leave for future work.
When considering the mutual coherence , one needs to look at the maximal correlation between every pair of atoms in the global dictionary. One should note, however, that atoms having a non-zero correlation must have overlapping supports. As we see, provides a bound for these values independently of the amount of overlap. One could go beyond this characterization of the convolutional dictionary by a single value and propose to bound all the inner products between atoms for a given shift. In this section we briefly explore this direction of analysis, introducing new tools for this model and addressing the theoretical consequences that they convey. We will only present the main points of these results here for the sake of brevity; the interested reader can find a more detailed discussion on this matter in Appendix C.
Recall that is defined as a stripe extracted from the global dictionary , as explained in Section III. Consider the sub-system given by , corresponding to the patch in . Note that can be split into a set of blocks of size , where each block is denoted by , i.e.,
as shown previously in Figure 2. Similarly, can be split into a set of vectors of length , each denoted by and corresponding to . In other words, . Note that previously we denoted local sparse vectors of length by . Yet, we will also denote them by in order to emphasize the fact that they correspond to the shift inside . Denote the number of non-zeros in as . We can also write , where is the number of non-zeros in each .
Define the shifted mutual coherence by
where is a column extracted from , is extracted from , and we requireThe condition if is necessary so as to avoid the inner product of an atom by itself. that if .
The above definition can be seen as a generalization of the mutual coherence for the shift-invariant local model presented in Section III. Indeed, characterizes just as characterizes the coherence of a general dictionary. Note that if the above definition boils down to the traditional mutual coherence of , i.e., . It is important to stress that the atoms used in the above definition are normalized globally according to and not . In Appendix C we comment on several interesting properties of this measure.
Following the definition of the shifted mutual coherence, for a given stripe from the linear system of equations , we can formulate a new measure:
According to this definition, each stripe has a coherence given by the sum of its non-zeros weighted by the shifted mutual coherence. As a particular case, if all non-zeros correspond to atoms in the center sub-dictionary, , this becomes . Note that unlike the traditional mutual coherence, this new measure depends on the location of the non-zeros in – it is a function of the support of the sparse vector, and not just of the dictionary. As such, it characterizes the correlation between the atoms participating in a given stripe. In what follows, we will use the notation for .
Next, we will present results based on these measures. Although these theorems are generally sharper, they are harder to grasp. We begin with a recovery guarantee for the OMP and BP algorithms, followed by a discussion on their implications.
(Global OMP recovery guarantee using the stripe coherence): Given the system of linear equations , if a solution exists satisfying
(Global BP recovery guarantee using the stripe coherence): Given the system of linear equations , if a solution exists satisfying
then Basis Pursuit is guaranteed to recover it.
The corresponding proofs are similar to their counterparts presented in the preceding section but require a more delicate analysis; one of them is thoroughly discussed in Appendix D.
In order to provide an intuitive interpretation for these results, the above bounds can be tied to a concrete number of non-zeros per stripe. First, notice that requiring the maximal stripe coherence to be less than a certain threshold is equal to requiring the same for every stripe:
Multiplying and dividing the left-hand side of the above inequality by and rearranging the resulting expression, we obtain
Define . Recall that and as such is simply the (weighted) average shifted mutual coherence in the stripe. Putting this definition into the above condition, the inequality becomes
Thus, the condition in (27) boils down to requiring the sparsity of all stripes to be less than a certain number. Naturally, this inequality resembles the one presented in the previous section for the OMP and BP guarantees. The reader might wonder about how they are related. In Appendix C we prove that under the assumption that , the shifted mutual coherence condition is at least as strong as the original one.
As a final note, the shifted mutual coherence, , is a considerably more informative measure than the standard mutual coherence. In some applications, the signals created by the convolutional dictionary are built of atoms which are known a priori to be separated by some minimal lag, or shift. In radio communications, for example, such a situation appears when there exists a minimal time between consecutive transmissions . In these cases, knowing how the correlation between the atoms depends on their shifts is fundamental for the design of the dictionary and its utilization.
One of the cardinal motivations for this work was a series of recent practical methods addressing the convolutional sparse coding problem; and in particular, the need for their theoretical foundation. However, our results are as of yet not directly applicable to these, as we have restricted our analysis to the ideal case of noiseless signals. The natural extension to this work is therefore the study of signals under noise contamination and model imperfections. This is indeed the path we undertake in part II of our work, exploring the question of whether the convolutional model remains stable in the presence of noise. Moreover, we show how to decompose and solve the global pursuit by performing merely local operations. This will tie the algorithmic solutions for the convolutional model to patch-based methods, which are the current practice in state-of-the-art signal and image restoration.
The research leading to these results has received funding from the European Research Council under European Union’s Seventh Framework Programme, ERC Grant agreement no. 320649. The authors would like to thank Dmitry Batenkov, Yaniv Romano and Raja Giryes for the prolific conversations and most useful advice which helped shape this work.
Let and be two global sparse vectors. Denote the stripe extracted from each as and , respectively. Notice that
In this section we prove both theorems presented in Section V, which guarantee the success of OMP and BP in solving the problem. We begin by presenting the OMP proof.
Denoting by the support of the solution , we can write
Suppose, without loss of generality, that the sparsest solution has its largest coefficient (in absolute value) in . For the first step of the OMP to choose one of the atoms in the support, we require
Substituting Equation (B-1) in this requirement we obtain
Using the reverse triangle inequality, the assumption that the atoms are normalized, and that , we construct a lower bound for the left hand side:
Consider the stripe which completely contains the atom as shown in Figure 4. Notice that is zero for every atom too far from because the atoms do not overlap. Denoting the stripe which fully contains the atom as and its support as , we can restrict the summation as:
We can bound the right side by using the number of non-zeros in the support , denoted by , together with the definition of the mutual coherence, obtaining:
Now, we construct an upper bound for the right hand side of Equation (B-3), using the triangle inequality and the fact that is the maximal value in the sparse vector:
Relying on the same rational as above, we obtain:
From this we obtain the requirement stated in the theorem. Thus, this condition guarantees the success of the first OMP step, implying it will choose an atom inside the true support.
B-B BP Success Guarantee (Proof of Theorem 9)
By defining , we obtain a shifted version of the set,
In what follows, we will enlarge the set and prove that it remains empty even after this expansion. Since , then . By subtracting from both sides, we obtain
Taking an entry-wise absolute value on both sides, we obtain
where we have applied the triangle inequality to the multiplication of the row of by the vector . Note that in the convolutional case is zero for inner products of atoms which do not overlap. Furthermore, the row of is non-zero only in the indices which correspond to the stripe that fully contains the atom, and these non-zero entries can be bounded by . Thus, extracting the row from the above equation gives
where is the stripe centered around the atom and is the corresponding sparse vector of length extracted from , as can be seen in Figure 5.
The above expression is a relaxation of the equality in Equation (B-16), since each entry is no longer constrained to a specific value, but rather bounded from below and above. Therefore, by putting the above into , we obtain a larger set :
Next, let us examine the second requirement
where, as before, denotes the support of . Using the reverse triangle inequality, , we obtain
where the vector contains ones in the entries corresponding to the support of and zeros elsewhere. Note that every vector satisfying Equation (B-21) will necessarily satisfy Equation (B-22). Therefore, by relaxing this constraint in , we obtain a larger set
Next, we will show the above defined set is empty for a small-enough support. We begin by summing the inequalities over the support of . Recall that is defined to be a stripe of length extracted from the global representation vector and corresponds to the central coefficients in the stripe. Also, note that is equal for all the entries inside the support of . Since all the atoms inside the support of are fully overlapping, does not change, as explained in Figure 5. Thus, we obtain
Summing over all different we obtain
Define a banded matrix (with a band of width ) such that , where . Notice that the summation in (B-28) is equal to the sum of all entries in this matrix, where the first sum considers all its rows while the second sum considers all its columns (the second sum is restricted to the non-zero band). Instead, this interpretation suggests that we could first sum over all the columns , and only then sum over all the rows which are inside the band. As a result, we obtain that
Using the definition of
For the set to be non-empty, there must exist a which satisfies
where the first and second inequalities are given in (B-22) and (B-35), respectively. Rearranging the above we obtain . However, we have assumed that and thus the previous inequality is not satisfied. As a result, the set we have defined is empty, implying that BP leads to the desired solution. ∎
Appendix C Properties of the Shifted Mutual Coherence and Stripe Coherence
The shifted mutual coherence exhibits some interesting properties:
is symmetric with respect to the shift , i.e. .
Its maximum over all shifts equals the global mutual coherence of the convolutional dictionary: .
The mutual coherence of the local dictionary is bounded by that of the global one: .
We now briefly remind the definition of the maximal stripe coherence, as we will make use of it throughout the rest of the appendix. Given a vector , recall that the stripe coherence is defined as , where is the number of non-zeros in the shift of , taken from . The reader might ponder how the maximal stripe coherence might be computed. Let us now define the vector which contains in its entry the number . Using this definition, the coherence of every stripe can be calculated efficiently by convolving the vector with the vector of the shifted mutual coherences .
Next, we provide an experiment in order to illustrate the shifted mutual coherence. To this end, we generate a random local dictionary with atoms of length and afterwards normalize its columns. We then construct a convolutional dictionary which contains global atoms of length . We exhibit the shifted mutual coherences for this dictionary in Figure 6(a).
We now present a theorem relating the stripe coherences of related sparse vectors.
Let and be two global sparse vectors such that the support of is contained in the support of . Then the maximal stripe coherence of is less or equal than the maximal stripe coherence of .
Denote by and the stripe extracted from and , respectively. Also, denote by and the number of non-zeros in the shift of and , respectively. Since the support of is contained in the support of , we have that . As a result, we have that
The left-hand side of the above inequality is the maximal stripe coherence of , while the right-hand side is the corresponding one for . Thus, we conclude that the maximal stripe coherence of is less or equal than the maximal stripe coherence of . ∎
Appendix D OMP Success Guarantee via Stripe Coherence (Proof of Theorem 12)
The first steps of this proof are exactly those derived in proving Theorem 8, and thus we omit them for the sake brevity. Recall that in order for the first step of OMP to succeed, we require
Lower bounding the left hand side of the above inequality, we can write
as stated previously in Equation (B-6). Instead of summing over the support , we can sum over all the supports , which correspond to all possible shifts. We can then write
We can bound the right term by using the number of non-zeros in each sub-support , denoted by , together with the corresponding shifted mutual coherence . Also, we can disregard the constraint in the above summation by subtracting an extra term, obtaining:
Bounding the above by the maximal stripe coherence, we obtain
In order to upper bound the right hand side of Equation (D-1) we follow the steps leading to Equation (B-9), resulting in
Using a similar decomposition of the support and the definition of the shifted mutual coherence, we have
Once again bounding this expression by the maximal stripe coherence, we obtain
which is the requirement stated in the theorem. Thus, this condition guarantees the success of the first OMP step, implying it will choose an atom inside the true support .
The next step in the OMP algorithm is an update of the residual. This is done by decreasing a term proportional to the chosen atom (or atoms within the correct support in subsequent iterations) from the signal. Thus, the support of this residual is contained within the support of the true signal. As a result, according to Theorem 15, the maximal stripe coherence corresponding to the residual is less or equal to the one of the true sparse code . Using the same set of steps we obtain that the condition on the maximal stripe coherence (27) guarantees that the algorithm chooses again an atom from the true support of the solution. Furthermore, the orthogonality enforced by the least-squares step guarantees that the same atom is never chosen twice. As a result, after iterations the OMP will find all the atoms in the correct support, reaching a residual equal to zero. ∎
As a final note, we mention that assuming is in fact a reasonable assumption. Recall that in order to compute we evaluate inner products between atoms which are indexes shifted from each other. As a result, the higher the shift is, the less overlap the atoms have, and the less is expected to be. Thus, we expect the value to be the largest or close to it in most cases.