Noisy Tensor Completion via the Sum-of-Squares Hierarchy
Boaz Barak, Ankur Moitra
Introduction
Matrix completion is one of the cornerstone problems in machine learning and has a diverse range of applications. One of the original motivations for it comes from the Netflix Problem where the goal is to predict user-movie ratings based on all the ratings we have observed so far, from across many different users. We can organize this data into a large, partially observed matrix where each row represents a user and each column represents a movie. The goal is to fill in the missing entries. The usual assumptions are that the ratings depend on only a few hidden characteristics of each user and movie and that the underlying matrix is approximately low rank. Another standard assumption is that it is incoherent, which we elaborate on later. How many entries of do we need to observe in order to fill in its missing entries? And are there efficient algorithms for this task?
There have been thousands of papers on this topic and by now we have a relatively complete set of answers. A representative result (building on earlier works by Fazel , Recht, Fazel and Parrilo , Srebro and Shraibman , Candes and Recht , Candes and Tao ) due to Keshavan, Montanari and Oh can be phrased as follows: Suppose is an unknown matrix that has rank but each of its entries has been corrupted by independent Gaussian noise with standard deviation . Then if we observe roughly
of its entries, the locations of which are chosen uniformly at random, there is an algorithm that outputs a matrix that with high probability satisfies
There are extensions to non-uniform sampling models , as well as various efficiency improvements . What is particularly remarkable about these guarantees is that the number of observations needed is within a logarithmic factor of the number of parameters — — that define the model.
In fact, there are benefits to working with even higher-order structure but so far there has been little progress on natural extensions to the tensor setting. To motivate this problem, consider the Groupon Problem (which we introduce here to illustrate this point) where the goal is to predict user-activity ratings. The challenge is that which activities we should recommend (and how much a user liked a given activity) depends on time as well — weekday/weekend, day/night, summer/fall/winter/spring, etc. or even some combination of these. As above, we can cast this problem as a large, partially observed tensor where the first index represents a user, the second index represents an activity and the third index represents the time period. It is again natural to model it as being close to low rank, under the assumption that a much smaller number of (latent) factors about the interests of the user, the type of activity and the time period should contribute to the rating. How many entries of the tensor do we need to observe in order to fill in its missing entries? This problem is emblematic of a larger issue: Can we always solve linear inverse problems when the number of observations is comparable to the number of parameters in the mode, or is computational intractability an obstacle?
In fact, one of the advantages of working with tensors is that their decompositions are unique in important ways that matrix decompositions are not. There has been a groundswell of recent work that uses tensor decompositions for exactly this reason for parameter learning in phylogenetic trees , HMMs , mixture models , topic models and to solve community detection . In these applications, one assumes access to the entire tensor (up to some sampling noise). But given that the underlying tensors are low-rank, can we observe fewer of their entries and still utilize tensor methods?
A wide range of approaches to solving tensor completion have been proposed . However, in terms of provable guarantees noneMost of the existing approaches rely on computing the tensor nuclear norm, which is hard to compute . The only other algorithms we are aware of require that the factors be orthogonal. This is a rather strong assumption. First, orthogonality requires the rank to be at most . Second, even when , most tensors need to be “whitened” to be put in this form and then a random sample from the “whitened” tensor would correspond to a (dense) linear combination of the entries of the original tensor, which would be quite a different sampling model. of them improve upon the following näive algorithm. If the unknown tensor is we can treat it as a collection of matrices each of size . It is easy to see that if has rank at most then each of these slices also has rank at most (and they inherit incoherence properties as well). By treating a third-order tensor as nothing more than an unrelated collection of low-rank matrices, we can complete each slice separately using roughly observations in total. When the rank is constant, this is a quadratic number of observations even though the number of parameters in the model is linear.
Here we show how to solve the (noisy) tensor completion problem with many fewer observations. Let . We give an algorithm based on the sixth level of the sum-of-squares hierarchy that can accurately fill in the missing entries of an unknown, incoherent tensor that is entry-wise close to being rank with roughly
observations. Moreover, our algorithm works even when the observations are corrupted by noise. When , this amounts to about observations per slice which is much smaller than what we would need to apply matrix completion on each slice separately. Our algorithm needs to leverage the structure between the various slices.
We give an algorithm for noisy tensor completion that works for third-order tensors. Let be a third-order tensor that is entry-wise close to being low rank. In particular let
Let represent the locations of the entries that we observe, which (as is standard) are chosen uniformly at random and without replacement. Set . Our goal is to output a hypothesis that has small entry-wise error, defined as:
This measures the error on both the observed and unobserved entries of . Our goal is to give algorithms that achieve vanishing error, as the size of the problem increases. Moreover we will want algorithms that need as few observations as possible. Here and throughout let and . Our main result is:
provided that .
Further, suppose that for a fraction of the entries of , we have and that is a tensor where each entry is a Gaussian with mean zero and variance . Then there is a polynomial time algorithm that outputs a hypothesis that satisfies
We believe that this work is a natural first step in designing practically efficient algorithms for tensor completion. Our algorithms manage to leverage the structure across the slices through the tensor, instead of treating each slice as an independent matrix completion problem. Now that we know this is possible, a natural follow-up question is to get more efficient algorithms. Our algorithms are based on the sixth level of the sum-of-squares hierarchy and run in polynomial time, but are quite far from being practically efficient as stated. Recent work of Hopkins et al. shows how to speed up sum-of-squares and obtain nearly linear time algorithms for a number of problems where the only previously known algorithms ran in a prohibitively large degree polynomial running time. Another approach would be to obtain similar guarantees for alternating minimization. Currently, the only known approaches require that the factors are orthonormal and only work in the undercomplete case. Finally, it would be interesting to get algorithms that recover a low rank tensor exactly when there is no noise.
2 Our approach
All of our algorithms are based on solving the following optimization problem:
and outputting the minimizer , where is some norm that can be computed in polynomial time. It will be clear from the way we define the norm that the low rank part of will itself be a good candidate solution. But this is not necessarily the solution that the convex program finds. How do we know that whatever it finds not only has low entry-wise error on the observed entries of , but also on the unobserved entries too?
This is a well-studied topic in statistical learning theory, and as is standard we can use the notion of Rademacher complexity as a tool to bound the error. The Rademacher complexity is a property of the norm we choose, and our main innovation is to use the sum-of-squares hierarchy to suggest a suitable norm. Our results are based on establishing a connection between noisy tensor completion and refuting random constraint satisfaction problems. Moreover, our analysis follows by embedding algorithms for refutation within the sum-of-squares hierarchy as a method to bound the Rademacher complexity.
A natural question to ask is: Are there other norms that have even better Rademacher complexity than the ones we use here, and that are still computable in polynomial time? It turns out that any such norm would immediately lead to much better algorithms for refuting random constraint satisfaction problems than we currently know. We have not yet introduced Rademacher complexity yet, so we state our lower bounds informally:
For any , if there is a polynomial time algorithm that achieves error
through the framework of Rademacher complexity then there is an efficient algorithm for refuting a random -SAT formula on variables with clauses. Moreover the natural sum-of-squares relaxation requires at least -levels in order to achieve the above error (again through the framework of Rademacher complexity).
These results follow directly from the works of Grigoriev , Schoenebeck and Feige . There are similar connections between our upper bounds and the work of Coja-Oghlan, Goerdt and Lanka who give an algorithm for strongly refuting random -SAT. In Section 2 we explain some preliminary connections between these fields, at which point we will be in a better position to explain how we can borrow tools from one area to address open questions in another. We state this theorem more precisely in Corollary 2.13 and Corollary 5.6, which provide both conditional and unconditional lower bounds that match our upper bounds.
3 Computational vs. Sample Complexity Tradeoffs
It is interesting to compare the story of matrix completion and tensor completion. In matrix completion, we have the best of both worlds: There are efficient algorithms which work when the number of observations is close to the information theoretic minimum. In tensor completion, we gave algorithms that improve upon the number of observations needed by a polynomial factor but still require a polynomial factor more observations than can be achieved if we ignore computational considerations. We believe that for many other linear inverse problems (e.g. sparse phase retrieval), there may well be gaps between what can be achieved information theoretically and what can be achieved with computationally efficient estimators. Moreover, proving lower bounds against the sum-of-squares hierarchy offers a new type of evidence that problems are hard, that does not rely on reductions from other average-case hard problems which seem (in general) to be brittle and difficult to execute while preserving the naturalness of the input distribution. In fact, even when there are such reductions , the sum-of-squares hierarchy offers a methodology to make sharper predictions for questions like: Is there a quasi-polynomial time algorithm for sparse PCA, or does it require exponential time?
Organization
In Section 2 we introduce Rademacher complexity, the tensor nuclear norm and strong refutation. We connect these concepts by showing that any norm that can be computed in polynomial time and has good Rademacher complexity yields an algorithm for strongly refuting random -SAT. In Section 3 we show how a particular algorithm for strong refutation can be embedded into the sum-of-squares hierarchy and directly leads to a norm that can be computed in polynomial time and has good Rademacher complexity. In Section 4 we establish certain spectral bounds that we need, and prove our main upper bounds. In Section 5 we prove lower bounds on the Rademacher complexity of the sequence of norms arising from the sum-of-squares hierarchy by a direct reduction to lower bounds for refuting random -XOR. In Appendix A we give a reduction from noisy tensor completion on asymmetric tensors to symmetric tensors. This is what allows us to extend our analysis to arbitrary order tensors, but the proofs are essentially identical to those in the case but more notationally involved so we omit them.
Noisy Tensor Completion and Refutation
Here we make the connection between noisy tensor completion and strong refutation explicit. Our first step is to formulate a problem that is a special case of both, and studying it will help us clarify how notions from one problem translate to the other.
Here we introduce a problem that we call the distinguishing problem. We are given random observations from a tensor and promised that the underlying tensor fits into one of the two following categories. We want an algorithm that can tell which case the samples came from, and succeeds using as few observations as possible. The two cases are:
Each observation is chosen uniformly at random (and without replacement) from a tensor where independently for each entry we set
where is a vector whose entries are .
Alternatively, each observation is chosen uniformly at random (and without replacement) from a tensor each of whose entries is independently set to either or and with equal probability.
In the first case, the entries of the underlying tensor are predictable. It is possible to guess a fraction of them correctly, once we have observed enough of its entries to be able to deduce . And in the second case, the entries of are completely unpredictable because no matter how many entries we have observed, the remaining entries are still random. Thus we cannot predict any of the unobserved entries better than random guessing.
Now we will explain how the distinguishing problem can be equivalently reformulated in the language of refutation. We give a formal definition for strong refutation later (Definition 2.10), but for the time being we can think of it as the task of (given an instance of a constraint satisfaction problem) certifying that there is no assignment that satisfies many of the clauses. We will be interested in -XOR formulas, where there are variables that are constrained to take on values or . Each clause takes the form
Each clause in the formula is generated by choosing an ordered triple of variables uniformly at random (and without replacement) and we set
where is a vector whose entries are . Now represents a planted solution and by design our sampling procedure guarantees that many of the clauses that are generated are consistent with it.
Alternatively, each clause in the formula is generated by choosing an ordered triple of variables uniformly at random (and without replacement) and we set where is a random variable that takes on values and .
In the first case, the -XOR formula has an assignment that satisfies a fraction of the clauses in expectation by setting . In the second case, any fixed assignment satisfies at most half of the clauses in expectation. Moreover if we are given clauses, it is easy to see by applying the Chernoff bound and taking a union bound over all possible assignments that with high probability there is no assignment that satisfies more than a fraction of the clauses.
This will be the starting point for the connections we establish between noisy tensor completion and refutation. Even in the matrix case these connections seem to have gone unnoticed, and the same spectral bounds that are used to analyze the Rademacher complexity of the nuclear norm are also used to refute random -SAT formulas , but this is no accident.
2 Rademacher Complexity
Ultimately our goal is to show that the hypothesis that our convex program finds is entry-wise close to the unknown tensor . By virtue of the fact that is a feasible solution to (2) we know that it is entry-wise close to on the observed entries. This is often called the empirical error:
For a hypothesis , the empirical error is
Recall that is the average entry-wise error between and , over all (observed and unobserved) entries. Also recall that among the candidate ’s that have low empirical error, the convex program finds the one that minimizes for some polynomial time computable norm. The way we will choose the norm and our bound on the maximum magnitude of an entry of will guarantee that the low rank part of will with high probability be a feasible solution. This ensures that for the we find is not too large either. One way to bound is to show that no hypothesis in the unit norm ball can have too large a gap between its error and its empirical error (and then dilate the unit norm ball so that it contains ). With this in mind, we define:
For a norm and a set of observations, the generalization error is
It turns out that one can bound the generalization error via the Rademacher complexity.
It follows from a standard symmetrization argument from empirical process theory that the Rademacher complexity does indeed bound the generalization error.
Let and suppose each with has bounded loss — i.e. and that locations are chosen uniformly at random and without replacement. Then with probability at least , for every with , we have
We repeat the proof here following for the sake of completeness but readers familiar with Rademacher complexity can feel free to skip ahead to Definition 2.5. The main idea is to let be an independent set of samples from the same distribution, again without replacement. The expected generalization error is:
Let be an tensor such that
This definition greatly simplifies our notation. In particular we have
where we have introduced the notation to denote the natural inner-product between tensors. Our main technical goal in this paper will be to analyze the Rademacher complexity of a sequence of successively tighter norms that we get from the sum-of-squares hierarchy, and to derive implications for noisy tensor completion and for refutation from these bounds.
3 The Tensor Nuclear Norm
A length vector is -incoherent if and .
Recall that we chose to work with vectors whose typical entry is a constant so that the entries in do not become vanishingly small as the dimensions of the tensor increase. We can now define the tensor nuclear normThe usual definition of the tensor nuclear norm has no constraints that the vectors , and be -incoherent. However, adding this additional requirement only serves to further restrict the unit norm ball, while ensuring that the low rank part of (when scaled down) is still in it, since the factors of are anyways assumed to be -incoherent. This makes it easier to prove recovery guarantees because we do not need to worry about sparse vectors behaving very differently than incoherent ones, and since we are not going to compute this norm anyways this modification will make our analysis easier.:
The tensor nuclear norm of which is denoted by is the infimum over such that .
In particular . Finally we give an elementary bound on the Rademacher complexity of the tensor nuclear norm. Recall that .
Recall the definition of given in Definition 2.5. With this we can write
We can now adapt the discretization approach in , although our task is considerably simpler because we are constrained to -incoherent ’s. In particular, let
By standard bounds on the size of an -net , we get that . Suppose that for all . Then for an arbitrary, but -incoherent we can expand it as where each and similarly for and . And now
Moreover since each entry in has magnitude at most we can apply a Chernoff bound to conclude that for any particular we have
with probability at least . Finally, if we set and we set we get that
The important point is that the Rademacher complexity of the tensor nuclear norm is whenever . In the next subsection we will connect this to refutation in a way that allows us to strengthen known hardness results for computing the tensor nuclear norm and show that it is even hard to compute in an average-case sense based on some standard conjectures about the difficulty of refuting random -SAT.
4 From Rademacher Complexity to Refutation
Here we show the first implication of the connection we have established. Any norm that can be computed in polynomial time and has good Rademacher complexity immediately yields an algorithm for strongly refuting random -SAT and -XOR formulas. Next let us finally define strong refutation.
For a formula , let be the largest fraction of clauses that can be satisfied by any assignment.
In what follows, we will use the term random -XOR formula to refer to a formula where each clause is generated by choosing an ordered triple of variables uniformly at random (and without replacement) and setting where is a random variable that takes on values and .
An algorithm for strongly refuting random -XOR takes as input a -XOR formula and outputs a quantity that satisfies
For any -XOR formula ,
If is a random -XOR formula with clauses, then with high probability
This definition only makes sense when is large enough so that holds with high probability, which happens when . The goal is to design algorithms that use as few clauses as possible, and are able to certify that a random formula is indeed far from satisfiable (without underestimating the fraction of clauses that can be satisfied) and to do so as close as possible to the information theoretic threshold.
Now let us use a polynomial time computable norm that has good Rademacher complexity to give an algorithm for strongly refuting random -XOR. As in Section 2.1, given a formula we map its clauses to a collection of observations according to the usual rule: If there are variables, we construct an tensor where for each clause of the form we put the entry at location . All the rest of the entries in are set to zero. We solve the following optimization problem:
Let be the optimum value. We set . What remains is to prove that the output of this algorithm solves the strong refutation problem for -XOR.
Suppose that is computable in polynomial time and satisfies whenever and is a vector with entries. Further suppose that for any with its entries are bounded by in absolute value. Then (3) can be solved in polynomial time and if then setting solves strong refutation for -XOR with clauses.
The key observation is the following inequality which relates (3) to .
To establish this inequality, let be the assignment that maximizes the fraction of clauses satisfied. If we set and we have that by assumption. Thus is a feasible solution. Now with this choice of for the right hand side, every term in the sum that corresponds to a satisfied clause contributes and every term that corresponds to an unsatisfied clause contributes . We get for this choice of , and this completes the proof of the inequality above.
The crucial point is that the expectation of the right hand side over and is exactly the Rademacher complexity. However we want a bound that holds with high probability instead of just in expectation. It follows from McDiarmid’s inequality and the fact that the entries of and of are bounded by and by in absolute value respectively that if we take observations the right hand side will be with high probability. In this case, rearranging the inequality we have
The right hand side is exactly and is with high probability, which implies that both conditions in the definition for strong refutation hold and this completes the proof. ∎
We can now combine Theorem 2.11 with the bound on the Rademacher complexity of the tensor nuclear norm given in Lemma 2.8 to conclude that if we could compute the tensor nuclear norm we would also obtain an algorithm for strongly refuting random -XOR with only clauses. It is not obvious but it turns out that any algorithm for strongly refuting random -XOR implies one for -SAT. Let us define strong refutation for -SAT. We will refer to any variable or its negation as a literal. We will use the term random -SAT formula to refer to a formula where each clause is generated by choosing an ordered triple of literals uniformly at random (and without replacement) and setting .
An algorithm for strongly refuting random -SAT takes as input a -SAT formula and outputs a quantity that satisfies
For any -SAT formula ,
If is a random -SAT formula with clauses, then with high probability
The only change from Definition 2.10 comes from the fact that for -SAT a random assignment satisfies a fraction of the clauses in expectation. Our goal here is to certify that the largest fraction of clauses that can be satisfied is . The connection between refuting random -XOR and -SAT is often called “Feige’s XOR Trick” . The first version of it was used to show that an algorithm for -refuting -XOR can be turned into an algorithm for -refuting -SAT. However we will not use this notion of refutation so for further details we refer the reader to . The reduction was extended later by Coja-Oghlan, Goerdt and Lanka to strong refutation, which for us yields the following corollary:
Suppose that is computable in polynomial time and satisfies whenever and is a vector with entries. Suppose further that for any with its entries are bounded by in absolute value and that . Then there is a polynomial time algorithm for strongly refuting a random -SAT formula with clauses.
Now we can get a better understanding of the obstacles to noisy tensor completion by connecting it to the literature on refuting random -SAT. Despite a long line of work on refuting random -SAT , there is no known polynomial time algorithm that works with clauses for any . Feige conjectured that for any constant , there is no polynomial time algorithm for refuting random -SAT with clausesIn Feige’s paper there was no need to make the conjecture any stronger because it was already strong enough for all of the applications in inapproximability.. Daniely et al. conjectured that there is no polynomial time algorithm for for any . What we have shown above is that any norm that is a relaxation to the tensor nuclear norm and can be computed in polynomial time but has Rademacher complexity is for would disprove the conjecture of Daniely et al. and would yield much better algorithms for refuting random -SAT than we currently know, despite fifteen years of work on the subject.
This leaves open an important question. While there are no known algorithms for strongly refuting random -SAT with clauses, there are algorithms that work with roughly clauses . Do these algorithms have any implications for noisy tensor completion? We will adapt the algorithm of Coja-Oghlan, Goerdt and Lanka and embed it within the sum-of-squares hierarchy. In turn, this will give us a norm that we can use to solve noisy tensor completion which uses a polynomial factor fewer observations than known algorithms.
Using Resolution to Bound the Rademacher Complexity
Here we introduce the sum-of-squares hierarchy and will use it (at level six) to give a relaxation to the tensor nuclear norm. This will be the norm that we will use in proving our main upper bounds. First we introduce the notion of a pseudo-expectation operator from :
(1) (normalization)
(2) , for any degree at most polynomial (nonnegativity)
Moreover suppose that with . We say that satisfies the constraint if for every . And we say that satisfies the constraint if for every .
, and
, and for all and
for all and .
The constraints in Definition 3.1 can be expressed as an -sized semidefinite program. This implies that given any set of polynomial constraints of the form , , one can efficiently find a degree pseudo-distribution satisfying those constraints if one exists. This is often called the degree Sum-of-Squares algorithm . Hence we can compute the norm of any tensor to within arbitrary accuracy in polynomial time. And because it is a relaxation to the tensor nuclear norm which is defined analogously but over a distribution on -incoherent vectors instead of a pseudo-distribution over them, we have that for every tensor . Throughout most of this paper, we will be interested in the case .
Recall that any polynomial time computable norm with good Rademacher complexity with observations yields an algorithm for strong refutation with roughly clauses too. Here we will use an algorithm for strongly refuting random -SAT to guide our search for an appropriate norm. We will adapt an algorithm due to Coja-Oghlan, Goerdt and Lanka that strongly refutes random -SAT, and will instead give an algorithm that strongly refutes random -XOR. Moreover each of the steps in the algorithm embeds into the sixth level of the sum-of-squares hierarchy by mapping resolution operations to applications of Cauchy-Schwartz, that ultimately show how the inequalities that define the norm (Definition 3.2) can be manipulated to give bounds on its own Rademacher complexity.
Let’s return to the task of bounding the Rademacher complexity of . Let be arbitrary but satisfy . Then there is a degree six pseudo-expectation meeting the conditions of Definition 3.2. Using Cauchy-Schwartz we have:
To simplify our notation, we will define the following polynomial
which we will use repeatedly. If is even then any degree pseudo-expectation operator satisfies the constraint for every polynomial of degree at most (e.g., see Lemma in ). Hence the right hand side of (4) can be bounded as:
It turns out that bounding the right-hand side of (5) boils down to bounding the spectral norm of the following matrix.
Let be the matrix whose rows and columns are indexed over ordered pairs and respectively, defined as
We can now make the connection to resolution more explicit: We can think of a pair of observations as a pair of -XOR constraints, as usual. Resolving them (i.e. multiplying them) we obtain a -XOR constraint
captures the effect of resolving certain pairs of -XOR constraints into -XOR constraints. The challenge is that the entries in are not independent, so bounding its maximum singular value will require some care. It is important that the rows of are indexed by and the columns are indexed by , so that and come from different -XOR clauses, as do and , and otherwise the spectral bounds that we will want to prove about would simply not be true! This is perhaps the key insight in .
It will be more convenient to decompose and reason about its two types of contributions separately. To that end, we let be the matrix whose non-zero entries are of the form
and all of its other entries are set to zero. Then let be the matrix whose entries are of the form
By construction we have . Finally:
The pseudo-expectation operator satisfies for all , and hence we have
We will now bound the contribution of and separately.
is positive semidefinite and has trace at most
It is easy to see that a quadratic form on corresponds to for some and this implies the first part of the claim. Finally
where the last equality follows because the pseudo-expectation operator satisfies the constraints and . ∎
Hence we can bound the contribution of the first term as . Now we proceed to bound the contribution of the second term:
It is easy to verify by direct computation that the following equality holds:
Moreover the pseudo-expectation of each of the three terms above is nonnegative, by construction. This implies the claim. ∎
Moreover each entry in is in the set and there are precisely non-zeros. Thus the sum of the absolute values of all entries in is at most . Now we have:
And this completes the proof of the lemma. ∎
Spectral Bounds
Recall the definition of given in the previous section. In fact, for our spectral bounds it will be more convenient to relabel the variables (but keeping the definition intact):
Let us consider the following random process: For partition the set of all ordered triples into two sets and . We will use this ensemble of partitions to define an ensemble of matrices : Set as equal to if and zero otherwise. Similarly set equal to if and zero otherwise. Also let be the event that there is no where and or vice-versa. Now let
As is standard, we are interested in bounding in order to bound . But note that is not symmetric. Also note that the random variables and are not independent, however whether or not they are non-zero is non-positively correlated and their signs are mutually independent. Expanding the trace above we have
Fix a particular term in the above sum where each random variable appears an even number of times. Let be the number of distinct values for . Moreover let be the order that these indices first appear. Now let denote the number of distinct values for that appear with in terms — i.e. is the number of distinct ’s that appear as . Let denote the number of distinct values for that appear with in terms — i.e. is the number of distinct ’s that appear as or . Similarly let denote the number of distinct values for that appear with in terms — i.e. is the number of distinct ’s that appear as . And finally let denote the number of distinct values for that appear with in terms — i.e. is the number of distinct ’s that appear as .
We give our encoding below. It is more convenient to think of the encoding as any way to answer the following questions about the term.
What is the order of the first appearance of each distinct value of ?
For each that appears, what is the order of each of the distinct values of ’s and ’s that appear along with it in ? Similarly, what is the order of each of the distinct values of ’s and ’s that appear along with it in ?
For each step (i.e. a new variable in the term when reading from left to right), has the value of been visited already? Also, has the value for or that appears along with been visited? Has the value for or that appears along with been visited? Note that whether or not or has been visited (together in ) depends on what the value of is, and if is a new value then the or value must be new too, by definition. Finally, if any value has already been visited, which earlier value is it?
It is easy to see that this encoding is injective, since given the answers to the above questions one can simulate each step and recover the sequence of random variables. Next we establish some easy facts that allow us to bound .
For any valid encoding, and .
This holds because in each step where the variable is new and has not been visited before, by definition the variable is new too (for the current ) and similarly for the variable. ∎
Finally, if and are defined as above then for any contributing term
its expectation is at most where because there are exactly distinct variables and distinct variables whose values are in the set and whether or not a variable is non-zero is non-positively correlated and the signs are mutually independent.
Note that the indicator variables only have the effect of zeroing out some terms that could otherwise contribute to . Returning to the task at hand, we have
Now if then using Claim 4.2 followed by the first half of Claim 4.1 we have:
where the last inequality follows because . Alternatively if then we can directly invoke the second half of Claim 4.1 and get:
As before, let . Then the last piece we need to bound the Rademacher complexity is the following spectral bound:
With high probability, \|B\|\leq O\Big{(}\frac{m\log^{4}n}{n_{1}^{1/2}\min(n_{2},n_{3})}\Big{)}
where we have used the fact that . This completes the proof of the theorem. ∎
We can now bound the Rademacher complexity of the norm that we get from the six level sum-of-squares relaxation to the tensor nuclear norm:
R^{m}(\|\cdot\|_{\mathcal{K}_{6}})\leq O\Big{(}\sqrt{\frac{(n_{1})^{1/2}(n_{2}+n_{3})\log^{4}n}{m}}\Big{)}
Consider any with . Then using Lemma 3.4 and Theorem 4.4 we have
Recall that was defined in Definition 2.5. The Rademacher complexity can now be bounded as
which completes the proof of the theorem. ∎
Recall that bounds on the Rademacher complexity readily imply bounds on the generalization error (see Theorem 2.4). We can now prove Theorem 1.1:
We solve (2) using the norm . Since this norm comes from the sixth level of the sum-of-squares hierarchy, it follows that (2) is an -sized semidefinite program and there is an efficient algorithm to solve it to arbitrary accuracy. Moreover we can always plug in and the bounds on the maximum magnitude of an entry in together with the Chernoff bound imply that with high probability is a feasible solution. Moreover . Hence with high probability, the minimizer satisfies . Now if we take any such returned by the convex program, because it is feasible its empirical error is at most . And since the bounds on the Rademacher complexity (Theorem 4.5) together with Theorem 2.4 give the desired bounds on and complete the proof of our main theorem. ∎
Our goal is to lower bound the absolute value of a typical entry in . To be concrete, suppose that for a fraction of the entries where . Consider , which we will view as a degree three polynomial in Gaussian random variables. Then the anti-concentration bounds of Carbery and Wright now imply that with probability . With this in mind, we define
and it follows form Markov’s bound that that . Now consider just those entries in which we get substantially wrong:
We can now invoke Theorem 1.1 which guarantees that the hypothesis that results from solving (2) satisfies with probability provided that . This bound on the error immediately implies that and so . This completes the proof of the corollary. ∎
Sum-of-Squares Lower Bounds
Here we will show strong lower bounds on the Rademacher complexity of the sequence of relaxations to the tensor nuclear norm that we get from the sum-of-squares hierarchy. Our lower bounds follow as a corollary from known lower bounds for refuting random instances of -XOR . First we need to introduce the formulation of the sum-of-squares hierarchy used in : We will call a Boolean function a -junta if there is set of at most variables so that is determined by the values in .
The -round Lasserre hierarchy is the following relaxation:
, for all
for all that are -juntas and
for all that are -juntas and satisfy
Here we define a vector for each -junta, and is a class of constraints that must be satisfied by any Boolean solution (and are necessarily -juntas themselves). See for more background, but it is easy to construct a feasible solution to the above convex program given a distribution on feasible solutions for some constraint satisfaction problem. In the above relaxation, we think of functions as being -valued. It will be more convenient to work with an intermediate relaxation where functions are -valued and the intuition is that for some set should correspond to the vector for the character .
Alternatively, the -round Lasserre hierarchy is the following relaxation:
, for all
for sets that are size at most and satisfy , where is the symmetric difference.
Here we have explicitly made the switch to XOR-constraints — namely has and correspond to the constraint that the parity on the set is equal to . Now if we have a feasible solution to the constraints in Definition 5.1 where all the clauses are XOR-constraints, we can construct a feasible solution to the constraints in Definition 5.2 as follows. If is a set of size at most , we define
where is the parity function on and is its complement. Moreover let .
is a feasible solution to the constraints in Definition 5.2
Consider Constraint in Definition 5.2, and let be sets of size at most that satisfy . Then our goal is to show that
where is the parity function on , and similarly for the other functions. Then we have because , and this implies that . An identical argument holds for the other terms. This implies that all the Constraints hold. Similarly suppose . Since and it is well-known that and are orthogonal and since in Definition 5.1, we have (see ). Thus
Now following Barak et al. we can use the constraints in Definition 5.2 to define the operator . In particular, given where and is multilinear, we set
Here we will also need to define when is not multilinear, and in that case if appears an even number of times we replace it with and if it appears an odd number of times we replace it by to get a multilinear polynomial and then set .
is a feasible solution to the constraints in Definition 3.2, and for any we have .
Then by construction , and the proof that is given in , but we repeat it here for completeness. Let be multilinear where we follow the above recipe and replace terms of the form with as needed. Then and moreover
as desired. Next we must verify that satisfies the constraints and for all , in accordance with Definition 3.1. To that end, observe that
which holds for any polynomial . Finally consider
which follows because and holds for any polynomial . This completes the proof. ∎
Let be a random -XOR formula on variables with clauses. Then for any and any , the round Lasserre hierarchy given in Definition 5.1 permits a feasible solution, with probability .
Note that the constant in the depends on and . Then using the above reductions, we have the following as an immediate corollary:
For any and any and , if the Rademacher complexity .
Thus there is a sharp phase transition (as a function of the number of observations) in the Rademacher complexity of the norms derived from the sum-of-squares hierarchy. At level six, whenever . In contrast, when even for very strong relaxations derived from rounds of the sum-of-squares hierarchy. These norms require time to compute but still achieve essentially no better bounds on their Rademacher complexity.
We would like to thank Aram Harrow for many helpful discussions.
References
Appendix A Reduction from Asymmetric to Symmetric Tensors
Here we give a general reduction, and show that any algorithm for tensor prediction that works for symmetric tensors can be used to predict the entries of an asymmetric tensor too. Hardt gave a related reduction for the cases of matrices and it is instructive to first understand this reduction, before proceeding to the tensor case. Suppose we are given a matrix that is not necessarily symmetric. Then the approach of is to construct the following symmetric matrix:
We have not precisely defined the notion of incoherence that is used in the matrix completion literature, but it turns out to be easy to see that is low rank and incoherent as well.
Now let us proceed to the tensor case. Let us introduce the following definition, for ease of notation:
Let be such that, there is an algorithm that on a rank , order , size symmetric tensor where each factor has norm at most , the algorithm returns an estimate with with probability when it is given samples chosen uniformly at random (and without replacement).
For any odd , suppose we are given samples chosen uniformly at random (and without replacement) from an tensor
where each factor is unit norm. There is an algorithm that with probability at least returns an estimate with
Our goal is to symmetrize an asymmetric tensor, and in such a way that each entry in the symmetrized tensor is either zero or else corresponds to an entry in the original tensor. Our reduction will work for any odd order tensor. In particular let
be an order tensor where the dimension of is . Also let . Then we will construct a symmetric, order tensor as follows. Let be a collection of random variables that are chosen uniformly at random from the configurations where . Then we consider the following random vector
Here is an -dimensional vector that results from concatenating the vectors but after flipping some of their signs according to . Then we set
It is immediate that is symmetric and has rank at most by expanding out the expectation into a sum over the valid sign configurations. Moreover each rank one term in the decomposition is of the form where because it is the concatenation of unit vectors.
If is fixed, then each entry in is itself a degree polynomial in the variables. By our construction of the variables, and because is odd so there are no terms where every variable appears to an even power, it follows that all the terms vanish in expectation except for the terms which have a factor of , and these are exactly terms that correspond to some permutation , and a term of the form
Hence all of the entries in are either zero or are times an entry in . As before, we can generate uniformly random samples from given uniformly random samples from , by simply choosing to sample an entry from one of the blocks of zeros with the appropriate probability, or else revealing an entry of and choosing where in to reveal this entry uniformly at random. Hence:
where represents the locations in where an entry of appears. The right hand side above is at most with probability . Moreover each entry in appears in exactly locations in . And when it does appear, it is scaled by . And hence if we multiply the left hand side by
we obtain . This completes the reduction. ∎
Note that in the case where , the error and the rank in this reduction increase only by at most an and factor respectively.