Convex relaxations of structured matrix factorizations

Francis Bach

Introduction

Structured matrix factorization has many applications in various areas of science engineering, i.e., clustering and principal component analysis , source separation , signal processing , machine learning , and all domains where reduced representations are desired.

Without any structure, traditional principal component analysis may be solved exactly in polynomial time through a singular value decomposition. However, adding additional structure on the components UU and VV of the factorization of X=UV⊤X=UV^{\top} (e.g., sparsity, non-negativity or discreteness) is most often done through alternating minimization (with respect to UU and VV). While all steps are usually done through convex optimization, the problem is not jointly convex, and there are typically multiple local minima, and algorithms usually come with no convergence guarantees. For example, in presence of positivity constraints, the problem of non-negative matrix factorization (NMF) may not be solved in polynomial-time in general, and most algorithms for NMF (e.g., ) perform a form a block coordinate descent with no guarantees (see hardness results and particular situations of actual solvability in ). In this paper, we follow a convex relaxation approach.

We impose some structure on each column of UU and VV and consider a general convex framework which amounts to computing a certain gauge function (such as a norm) at XX, from which the decomposition may be obtained (for example, the nuclear norm leads to the usual singular value decomposition). This convex framework corresponds to removing any rank constraint on the decomposition and has appeared under various forms in the literature, as summing norms , decomposition norms or (a special case of) atomic norms . This is presented in details (equivalent representations, rotation-invariant cases, weighted nuclear norm formulations) in Section 4. An interesting aspect is that the gauge functions we consider are polar to generalizations of matrix norms, which are commonly used in many areas of applied mathematics, in particular in robust optimization and control .

The statistical and recovery properties of these norms and their relaxations have been studied in several contexts ; in this paper, we focus on optimization aspects. The convex framework we introduce in Section 4 only lead to polynomial-time algorithms in few situations (e.g., the nuclear norm based on the singular value decomposition). In Section 5, we consider computable additional relaxations based on semi-definite programming. These may be used to compute the related gauge functions as well as their polar, with constant-factor approximation guarantees in some cases (see Section 5.1). The first setting where one can get dimension-independent guarantees has already been studied by and corresponds to gauge functions that have variationl diagonal representations. We also consider a more general setting with dimension-dependent bounds.

A key practical problem is to obtain not only a lower-bound on the value of the gauge function, but also an explicit decomposition which preserves the approximation guarantees. We present in Section 6 iterative conditional gradient algorithms and their analysis, which extend existing results in several ways: (a) we obtain convergence guarantees even when the polar gauge function may be approximately computed with a multiplicative approximatio ratio—earlier work considers only additive approximations, (b) following , they may be applied to penalized versions of the problem, i.e., to solve a generalized basis pursuit problem, (c) under some additional assumptions, they may find approximate decompositions of XX that converge linearly.

Finally, our framework relies on variational representations of gauge functions and in particular norms as maxima and minima of quadratic functions. These representations have been already used in several contexts (machine learning, signal processing, optimization, see, e.g., and references therein). In this paper, we provide in Section 3 a thorough analysis of these decompositions (minimal and maximal representations, duality between lower and upper bounds, sufficient and necessary conditions for diagonal or rotation-invariant representations).

Review of gauge function theory

In this section, we present relevant concepts and results from convex analysis. These tools are needed because the type of structure we want to impose go beyond what can be characterized by norms (such as positivity). See for more details on gauge functions and their properties.

Polar sets and functions.

Given any set C{\mathcal{C}} (not necessarily convex), the polar of C{\mathcal{C}} is the set C∘{\mathcal{C}}^{\circ} defined as

It is always closed and convex. Moreover, the polar of C{\mathcal{C}} is equal to the polar of the closure of hull(C∪{0}){\rm hull}({\mathcal{C}}\cup\{0\}).

When C{\mathcal{C}} is the unit ball of a norm Ω\Omega, C∘{\mathcal{C}}^{\circ} is the unit ball of the dual norm which we denote Ω∘\Omega^{\circ} (instead of the usual definition Ω∗\Omega^{\ast}, because the Fenchel conjugate of Ω\Omega is not the dual norm, but the indicator function of the dual unit ball).

If C{\mathcal{C}} is a closed convex set containing the origin, then C∘∘=C{\mathcal{C}}^{\circ\circ}={\mathcal{C}}—more generally, for any set C{\mathcal{C}}, C∘∘{\mathcal{C}}^{\circ\circ} is the closure of hull(C∪{0}){\rm hull}({\mathcal{C}}\cup\{0\}). The polarity is a one-to-one mapping from closed convex sets containing the origin to themselves. In this paper, we will also consider gauge functions associated with closed potentially non convex sets C{\mathcal{C}}, in which case, we mean the gauge function associated to C∘∘=hull(C∪{0}){\mathcal{C}}^{\circ\circ}={\rm hull}({\mathcal{C}}\cup\{0\}), i.e., γC=γC∘∘\gamma_{\mathcal{C}}=\gamma_{{\mathcal{C}}^{\circ\circ}}.

Operations on gauge functions.

Links with convex hulls.

Given a compact set P{\mathcal{P}} and its compact convex hull C{\mathcal{C}} (for example, P{\mathcal{P}} might be the set of extreme points of C{\mathcal{C}}), we have P∘=C∘,{\mathcal{P}}^{\circ}={\mathcal{C}}^{\circ}, since maxima of linear functions on C{\mathcal{C}} or P{\mathcal{P}} are equal. An alternative definition of γC\gamma_{\mathcal{C}} is then

Moreover, in the definition above, by Caratheodory’s theorem for cones, we may restrict the cardinality of II to be less than or equal to dd.

Representations of gauge functions through quadratic functions

We first consider closed convex sets K⊂Sd+{\mathcal{K}}\subset{\mathcal{S}}_{d}^{+} of symmetric positive definite matrices such that

When U{\mathcal{U}} is symmetric, the variational formulation in Eq. (1) leads to a representation of γU(x)2\gamma_{\mathcal{U}}(x)^{2} as a convex function IK∗(xx⊤)I_{\mathcal{K}}^{\ast}(xx^{\top}) of xx⊤xx^{\top}. Note that in general, K{\mathcal{K}} is not unique. We now show that there always exists a set K{\mathcal{K}} satisfying Eq. (1), and provide a description of the largest such set.

Let PU={xx⊤, x∈U}{\mathcal{P}}_{\mathcal{U}}=\{xx^{\top},\ x\in{\mathcal{U}}\}. Then PU∘∩Sd+{\mathcal{P}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+} is the larget closed convex set K⊂Sd+{\mathcal{K}}\subset{\mathcal{S}}_{d}^{+} of positive semidefinite matrices such that Eq. (1) is satisfied.

We now have sup⁡M∈PU∘∩Sd+x⊤Mx⩽sup⁡M∈PU∘x⊤Mx=min⁡{γU(x),γU(−x)}\sup_{M\in{\mathcal{P}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+}}\sqrt{x^{\top}Mx}\leqslant\sup_{M\in{\mathcal{P}}_{\mathcal{U}}^{\circ}}\sqrt{x^{\top}Mx}=\min\{\gamma_{\mathcal{U}}(x),\gamma_{\mathcal{U}}(-x)\}. Since this is a convex function of xx, it must be less than its convex envelope, which is γU∪(−U)(x)\gamma_{{\mathcal{U}}\cup(-{\mathcal{U}})}(x) (since they have the same Fenchel conjugates).

Moreover, if v∈U∘∩(−U)∘v\in{\mathcal{U}}^{\circ}\cap(-{\mathcal{U}})^{\circ}, then vv⊤∈PU∘vv^{\top}\in{\mathcal{P}}_{\mathcal{U}}^{\circ}. This implies that sup⁡M∈PU∘x⊤Mx⩾sup⁡v∈U∘∩(−U)∘(v⊤x)2=IU∘∩(−U)∘∗(x)2=γU∘∩(−U∘)∘(x)2=γU∪(−U)(x)2\sup_{M\in{\mathcal{P}}_{\mathcal{U}}^{\circ}}x^{\top}Mx\geqslant\sup_{v\in{\mathcal{U}}^{\circ}\cap(-{\mathcal{U}})^{\circ}}(v^{\top}x)^{2}=I^{\ast}_{{\mathcal{U}}^{\circ}\cap(-{\mathcal{U}})^{\circ}}(x)^{2}=\gamma_{{\mathcal{U}}^{\circ}\cap(-{\mathcal{U}}^{\circ})}^{\circ}(x)^{2}=\gamma_{{\mathcal{U}}\cup(-{\mathcal{U}})}(x)^{2}. Thus Eq. (1) is indeed satisfied by PU∘∩Sd+{\mathcal{P}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+}. Finally, if K{\mathcal{K}} satisfies Eq. (1), then we must have K⊂PU∘{\mathcal{K}}\subset{\mathcal{P}}_{\mathcal{U}}^{\circ} by definition of polar sets, hence PU∘∩Sd+{\mathcal{P}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+} is the largest.

Note that the set PU∘∩Sd+{\mathcal{P}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+} is equal to {M∈Sd+, ∀u∈U,u⊤Mu⩽1}\{M\in{\mathcal{S}}_{d}^{+},\ \forall u\in{\mathcal{U}},u^{\top}Mu\leqslant 1\}—this representation was already considered in for norms. In certain situations, the largest possible set is desirable (for example when deriving convex relaxations). In other situations (for example when using these representations for optimization), smallest sets are desirable. However, such a notion is not possible. Indeed, for γU=∥⋅∥2\gamma_{\mathcal{U}}=\|\cdot\|_{2}, the sets K={I}{\mathcal{K}}=\{I\} and K={M∈Sd+, ∥M∥F⩽1}{\mathcal{K}}=\{M\in{\mathcal{S}}_{d}^{+},\ \|M\|_{F}\leqslant 1\}, are two possible sets, and thus there is no single smallest set. One possible small set is the convex hull of the maximal elements of PU∘∩Sd+{{\mathcal{P}}}_{\mathcal{U}}^{\circ}\cap{\mathcal{S}}_{d}^{+} (for the positive semi-definite order).

In the proof of Prop. 1, we have introduced the gauge function γPU\gamma_{{\mathcal{P}}_{\mathcal{U}}}. We now provide a representation of γPU\gamma_{{\mathcal{P}}_{\mathcal{U}}} related to factorizations of positive semidefinite matrices (note that in the following proposition, we only assume that U{\mathcal{U}} is closed and contains 0 in its hull).

Let PU={xx⊤, x∈U}⊂Sd+{\mathcal{P}}_{\mathcal{U}}=\{xx^{\top},\ x\in{\mathcal{U}}\}\subset{\mathcal{S}}_{d}^{+}. We have, for all positive semidefinite matrix M∈Sd+M\in{\mathcal{S}}_{d}^{+}:

Moreover, we may choose r⩽d(d+1)/2r\leqslant d(d+1)/2.

Proof This is a direct application of the representation of gauge functions and and the property γPU(xx⊤)=min⁡{γU(x),γU(−x)}2\gamma_{{\mathcal{P}}_{\mathcal{U}}}(xx^{\top})=\min\{\gamma_{\mathcal{U}}(x),\gamma_{\mathcal{U}}(-x)\}^{2}, that was shown in the proof of Prop. 1.

The last proposition provides a structured decomposition framework for positive semi-definite matrices, that will be considered for rectangular matrices in Section 4. Obtaining explicitly the decomposition M=UU⊤M=UU^{\top} from MM may be done with the iterative algorithms presented in Section 6. Note that by considering a representation of UU as U=M1/2SU=M^{1/2}S where SS⊤=ISS^{\top}=I, computing γPU\gamma_{{\mathcal{P}}_{\mathcal{U}}} may be seen as a factorization problem with two factors (which can then be used in alternating minimization procedures).

2 Minima of quadratic functions

We now consider closed convex sets L⊂Sd+{\mathcal{L}}\subset{\mathcal{S}}_{d}^{+} such that

Here, we define x⊤M−1xx^{\top}M^{-1}x as x⊤M−1x=inf⁡tx^{\top}M^{-1}x=\inf t such that \Big{(}\!\begin{array}[]{cc}M&x\\ x^{\top}&t\end{array}\!\Big{)}\succcurlyeq 0\Leftrightarrow tM\succcurlyeq xx^{\top}. This implies that the value may be finite even when MM is not invertible.

When U{\mathcal{U}} is symmetric, the variational formulation in Eq. (2) leads to a representation of γU(x)2\gamma_{\mathcal{U}}(x)^{2} as a concave inf⁡M∈LtrM−1xx⊤\inf_{M\in{\mathcal{L}}}\mathop{\rm tr}M^{-1}xx^{\top} of xx⊤xx^{\top}. This is to be contrasted with the fact that it is also a convex function of xx⊤xx^{\top} because of the representation discussed in Section 3.1. The two properties are in fact related through a duality argument:

Let L⊂Sd+{\mathcal{L}}\subset{\mathcal{S}}_{d}^{+} be a closed convex set. Then the following two properties are equivalent:

This implies that the largest set L{\mathcal{L}} such that (a) is valid is PU∘∘∩Sd+{\mathcal{P}}^{\circ}_{{\mathcal{U}}^{\circ}}\cap{\mathcal{S}}_{d}^{+} defined in Prop. 1.

Similarly, if we have a concave upper-bound of max⁡{γU∘(y),γU∘(−y)}2\max\{\gamma_{\mathcal{U}}^{\circ}(y),\gamma_{\mathcal{U}}^{\circ}(-y)\}^{2} based on K{\mathcal{K}}, we also have a convex upper-bound based on K∘∩Sd+{\mathcal{K}}^{\circ}\cap{\mathcal{S}}_{d}^{+}. Note that it is not true in general that exact representations of one kind transfer to exact representation of the other kind.

3 Examples

4 Diagonal representations

We now consider cases where the set of matrices K{\mathcal{K}} and L{\mathcal{L}} are diagonal, i.e., all principal axes of ellipsoids are aligned with the canonical basis. For simplicity, we consider only sets U{\mathcal{U}} which are compact, have zero in their interior, and are invariant by sign flips of any components. The corresponding gauge functions are then absolute norms, which are functions of the absolute values of each component . The following proposition provides several characterizations (see related work in ).

Proof It is straightforward to see that (a) implies both (b) and (c).

5 Matrix norms invariant by rotation

W↦Ω(W)2W\mapsto\Omega(W)^{2} is a concave function of WW⊤WW^{\top}.

V↦Ω∘(V)2V\mapsto\Omega^{\circ}(V)^{2} is a concave function of VV⊤VV^{\top}.

Gauge functions and structured matrix factorizations

This function is exactly the gauge function of the set {uv⊤, u∈U,v∈V}\{uv^{\top},\ u\in{\mathcal{U}},v\in V\}. It is therefore convex, non-negative and positively homogeneous. Moreover, the minimization problem defining it is attained for r⩽ndr\leqslant nd. There are several equivalent formulations which we are going to use, which relies on the fact that umvm⊤=(umλm)(vmλm−1)⊤u_{m}v_{m}^{\top}=(u_{m}\lambda_{m})(v_{m}\lambda_{m}^{-1})^{\top} for any λm>0\lambda_{m}>0 (in the following expressions, we always minimize with respect to the rank and thus omit the notation inf⁡r⩾0\inf_{r\geqslant 0}).

Moreover, Θ\Theta is a norm as soon as U{\mathcal{U}} and V{\mathcal{V}} are norm balls—in this case, these norms were studied in several settings . The next proposition shows that the polar of Θ\Theta has a simple form (the proof is straightforward from the gauge function interpretation). It is a matrix norm when U{\mathcal{U}} and V{\mathcal{V}} are norm balls; moreover, in all cases, since U{\mathcal{U}} and V{\mathcal{V}} are assumed compact, Θ∘\Theta^{\circ} has full domain.

2 Examples

γV=∥⋅∥2\gamma_{\mathcal{V}}=\|\cdot\|_{2} and γU=ν∥⋅∥22+(1−ν)∥⋅∥12\gamma_{\mathcal{U}}=\sqrt{\nu\|\cdot\|_{2}^{2}+(1-\nu)\|\cdot\|_{1}^{2}}: this is a convex relaxation of sparse coding , where a decomposition with a sparse factor UU and low rank (small number of columns for UU and VV) is looked for.

γV=∥⋅∥2\gamma_{\mathcal{V}}=\|\cdot\|_{2} and γU=∥⋅∥4/3\gamma_{\mathcal{U}}=\|\cdot\|_{4/3}: we have γU∘=∥⋅∥4\gamma_{\mathcal{U}}^{\circ}=\|\cdot\|_{4}, and the dual norm is such that Θ∘(Y)4=max⁡∥v∥2⩽1∑i(Yv)i4\Theta^{\circ}(Y)^{4}=\max_{\|v\|_{2}\leqslant 1}\sum_{i}(Yv)_{i}^{4}, and corresponds to maximum kurtosis projections .

3 Special case: γ𝒱=∥⋅∥2\gamma_{\mathcal{V}}=\|\cdot\|_{2}

Computing the polar of Θ\Theta then corresponds to a quadratic maximization problem:

We also have a variational quadratic representation corresponding to a rotation-invariant gauge function (Section 3.5); following Section 3.1, we denote by PU{\mathcal{P}}_{\mathcal{U}} the set {uu⊤, u∈U}\{uu^{\top},\ u\in{\mathcal{U}}\}. There is an explicit representation of Θ\Theta and Θ∘\Theta^{\circ} in terms of PU∘∘{\mathcal{P}}_{\mathcal{U}}^{\circ\circ} (the convex hull of PU∪{0}{\mathcal{P}}_{\mathcal{U}}\cup\{0\}).

Note that Θ\Theta may not have full domain.

From the earlier representation, we get \displaystyle\Theta(X)=\inf_{M\succcurlyeq 0}\big{\{}\frac{1}{2}\gamma_{{\mathcal{P}}_{\mathcal{U}}}(M)+\frac{1}{2}\mathop{\rm tr}X^{\top}M^{-1}X\big{\}}, by simply optimizing over the half line generated by MM. We also have a representation involving all decompositions of XX as UV⊤UV^{\top}:

which is straightforward from the definitions and interpretations of Θ\Theta and γPU\gamma_{{\mathcal{P}}_{\mathcal{U}}} as a gauge function (Prop. 2). In the next section, we show how to compute UU and VV from the solution of a certain convex problem—note however that this does not lead to the optimal UU and VV in Eq. (3). See more details in Section 6.

4 General case

We now consider all possible cases, beyond γV=∥⋅∥2\gamma_{\mathcal{V}}=\|\cdot\|_{2} (where γPV(M)=trM\gamma_{{\mathcal{P}}_{\mathcal{V}}}(M)=\mathop{\rm tr}M). We only assume that U{\mathcal{U}} and V{\mathcal{V}} are compact and contain zero in their hulls. It is tempting to consider an extension of Eq. (5), by considering

Let Θ~\widetilde{\Theta} be defined in Eq. (6). Then, Θ~\widetilde{\Theta} is a gauge function (convex, positively homogeneous, and non-negative), and its polar has the expression:

because sup⁡UU⊤=I, VV⊤=ItrZ⊤UV⊤=∥Z∥∗\sup_{UU^{\top}=I,\ VV^{\top}=I}\mathop{\rm tr}Z^{\top}UV^{\top}=\|Z\|_{\ast}. Finally, we have for some um,vmu_{m},v_{m}, \Theta(X)=\frac{1}{2}\sum_{m}\big{\{}\gamma_{\mathcal{U}}(u_{m})^{2}+\gamma_{\mathcal{V}}(v_{m})^{2}\big{\}}. Thus Θ(X)⩾Θ~(X)\Theta(X)\geqslant\widetilde{\Theta}(X) because of Prop. 2, which implies the reverse inequality for polars.

We may now give several equivalent expressions for Θ~\widetilde{\Theta} and Θ~∘\widetilde{\Theta}^{\circ} which are obtained using usual convex duality.

Let Θ~\widetilde{\Theta} be defined in Eq. (6). We have:

Proof The four expressions are equivalent by taking polars or using Lagrangian duality for the semi-definite cone. The only element left to prove is that any of these four statements are correct. We have:

which leads to desired result from Prop. 9. Note that we may obtain directly a relationship between Θ~∘\widetilde{\Theta}^{\circ} and Θ∘\Theta^{\circ} as follows:

that is a variational formulation of Θ~\widetilde{\Theta} through nuclear norms (note again that PU∘{\mathcal{P}}_{\mathcal{U}}^{\circ} and PV∘{\mathcal{P}}_{\mathcal{V}}^{\circ} may not be compact and Θ~(X)\widetilde{\Theta}(X) potentially infinite). This is to be contrasted with the following representation:

which is a variational upper-bound as a weighted Frobenius norm, which is not computable as a convex program. Instead, alternate minimization could be used, each of the steps being doable as a convex program.

Given the primal/dual solutions (M,N)(M,N), (Q,S)(Q,S) of the dual convex optimization problems defining Θ~\widetilde{\Theta} in Prop. 10, we have the following optimality conditions: Assume Q=AA⊤Q=AA^{\top} with AA full rank (i.e., A⊤AA^{\top}A invertible), and S=BB⊤S=BB^{\top} with BB full rank. Then an optimal ZZ is Z=AGH⊤B⊤Z=AGH^{\top}B^{\top}, with A⊤XB=GDiag(s)H⊤A^{\top}XB=G\mathop{\rm Diag}(s)H^{\top} is the singular value decomposition of A⊤XBA^{\top}XB. We have

which implies (since AA and BB have full rank):

This leads to A⊤MA=A⊤XBHG⊤=GDiag(s)G⊤A^{\top}MA=A^{\top}XBHG^{\top}=G\mathop{\rm Diag}(s)G^{\top}. We then take as candidates U=(AA⊤)†AGDiag(s)1/2U=(AA^{\top})^{\dagger}AG\mathop{\rm Diag}(s)^{1/2} and V=(BB⊤)†BHDiag(s)1/2V=(BB^{\top})^{\dagger}BH\mathop{\rm Diag}(s)^{1/2}, leading to UU⊤=MUU^{\top}=M and VV⊤=NVV^{\top}=N. Moreover, X=UV⊤X=UV^{\top} and Θ~(X)=∥A⊤XB∥∗=trDiag(s)\widetilde{\Theta}(X)=\|A^{\top}XB\|_{\ast}=\mathop{\rm tr}\mathop{\rm Diag}(s). Finally, by optimality of QQ, we have γPU(UU⊤)=trUU⊤Q=trA⊤MA=trDiag(s)\gamma_{{\mathcal{P}}_{\mathcal{U}}}(UU^{\top})=\mathop{\rm tr}UU^{\top}Q=\mathop{\rm tr}A^{\top}MA=\mathop{\rm tr}\mathop{\rm Diag}(s) and γPV(VV⊤)=trVV⊤S=trB⊤NB=trDiag(s)\gamma_{{\mathcal{P}}_{\mathcal{V}}}(VV^{\top})=\mathop{\rm tr}VV^{\top}S=\mathop{\rm tr}B^{\top}NB=\mathop{\rm tr}\mathop{\rm Diag}(s), and thus (U,V)(U,V) is an optimal decomposition for Eq. (6).

The duality properties presented in Prop. 10 are valid for all convex sets of positive semi-definite matrices that contain zero (and in particular for the ones in Section 5). This will be used in Section 5 to get computable approximations (with guarantees).

5 Non-convex approaches

Finding approximations of the gauge function Θ\Theta or its polar Θ∘\Theta^{\circ} has been tackled thoroughly through non-convex approaches, which may only find stationary points of the associated optimization problems. In this paper, we briefly review two approaches based on alternating minimization.

In Section 6, obtaining approximate maximizers (u,v)(u,v) will be key for the generalized conditional gradient algorithm and simplicial methods. In our experiments in Section 7, the power method leads to better values once initialized from results of the convex relaxations presented in Section 5.

Alternating optimization.

We consider r⩽npr\leqslant np elements and minimize either the sum ∑m=1rγU(um)γV(vm)\sum_{m=1}^{r}\gamma_{\mathcal{U}}(u_{m})\gamma_{\mathcal{V}}(v_{m}) or \sum_{m=1}^{r}\big{\{}\frac{1}{2}\gamma_{\mathcal{U}}(u_{m})^{2}+\frac{1}{2}\gamma_{\mathcal{V}}(v_{m})^{2}\big{\}}, subject to X=∑m=1rumvm⊤X=\sum_{m=1}^{r}u_{m}v_{m}^{\top}. This may be done by alternating minimization, with no guarantees, beyond decreasing the value of the upper-bound on the norm.

Non-convex optimization without local minima.

Nevertheless, in practice, alternating optimization techniques, although they come with no guarantees, still perform well when they can be applied (see Section 7).

Computable convex relaxations of decomposition norms

Before describing computable relaxations, we first refine the computability requirement which is weaker than having a good representation of PU∘∘{\mathcal{P}}_{\mathcal{U}}^{\circ\circ} or its polar PU∘{\mathcal{P}}_{\mathcal{U}}^{\circ}. We start with a simple lemma:

Let A,B⊂Sn+\mathcal{A},\mathcal{B}\subset{\mathcal{S}}_{n}^{+}. The following statements are equivalent:

(a) ∀M∈Sn+\forall M\in{\mathcal{S}}_{n}^{+}, sup⁡N∈AtrMN⩽sup⁡N∈BtrMN\sup_{N\in\mathcal{A}}\mathop{\rm tr}MN\leqslant\sup_{N\in\mathcal{B}}\mathop{\rm tr}MN,

(b) A∘∘−Sn+⊂B∘∘−Sn+\mathcal{A}^{\circ\circ}-{\mathcal{S}}_{n}^{+}\subset\mathcal{B}^{\circ\circ}-{\mathcal{S}}_{n}^{+},

(c) A∘∩Sn+⊃B∘∩Sn+\mathcal{A}^{\circ}\cap{\mathcal{S}}_{n}^{+}\supset\mathcal{B}^{\circ}\cap{\mathcal{S}}_{n}^{+}.

Moreover, if A\mathcal{A} is composed of rank-one matrices, then they are also equivalent to

Proof (a) is equivalent to (c) by simple properties of gauge functions. (b) is equivalent to (c) because of polar calculus and (Sn+)∘=−Sn+({\mathcal{S}}_{n}^{+})^{\circ}=-{\mathcal{S}}_{n}^{+}, which imply (A∘∩Sn+)∘=hull(A∘∘∪(−Sn+))=A∘∘−Sn+(\mathcal{A}^{\circ}\cap{\mathcal{S}}_{n}^{+})^{\circ}={\rm hull}(\mathcal{A}^{\circ\circ}\cup(-{\mathcal{S}}_{n}^{+}))=\mathcal{A}^{\circ\circ}-{\mathcal{S}}_{n}^{+}. (a) trivially implies (d). We only need to show (d) implies (a). Assume (d) is true and let M=YY⊤∈Sn+M=YY^{\top}\in{\mathcal{S}}_{n}^{+}, we have

Given the representation of Θ~\widetilde{\Theta} in Prop. 10, a key consequence of the previous lemma is that in the representation of Θ~(X)\widetilde{\Theta}(X) and Θ~∘(Y)\widetilde{\Theta}^{\circ}(Y) in Section 4, we may replace PU∘∘{\mathcal{P}}_{{\mathcal{U}}}^{\circ\circ} by any set CU{\mathcal{C}}_{\mathcal{U}} such that

For all other cases, we need computable convex approximations CU{\mathcal{C}}_{\mathcal{U}} of PU∘∘{\mathcal{P}}_{\mathcal{U}}^{\circ\circ} such that for

(if we impose the property above for all symmetric matrices MM, it is equivalent to PU∘∘⊂CU{\mathcal{P}}_{\mathcal{U}}^{\circ\circ}\subset{\mathcal{C}}_{\mathcal{U}}). Given Lemma 1, this is equivalent to

that is, we need an upper-bound on max⁡{γU∘(y),γU∘(−y)}2\max\{\gamma_{\mathcal{U}}^{\circ}(y),\gamma_{\mathcal{U}}^{\circ}(-y)\}^{2}, which is convex in yy⊤yy^{\top}. Note that given two sets CU{\mathcal{C}}_{\mathcal{U}} that satisfy Eq. (32), their intersection also does and the relaxation is then always tighter. We can therefore use several properties of the set U{\mathcal{U}}, that may then be combined together:

Linear inequalities: any matrix MM such that max⁡γU(u)⩽1u⊤Mu\max_{\gamma_{\mathcal{U}}(u)\leqslant 1}u^{\top}Mu may be computed or efficiently upper-bounded by hh adds another constraint of the form trMU⩽h\mathop{\rm tr}MU\leqslant h. Typically, M=IM=I is a good candidate, as this defines the equivalence between the gauge function γU\gamma_{\mathcal{U}} and ∥⋅∥2\|\cdot\|_{2}.

In Table 1, we present relaxations for the examples we consider in this paper. For all relaxations we consider (and also with positivity constraints), we have: CU∩{M, rank(M)=1}=PU{\mathcal{C}}_{\mathcal{U}}\cap\{M,\ {\rm rank}(M)=1\}={\mathcal{P}}_{\mathcal{U}}. This implies that for all u∈dom(γU)u\in{\rm dom}(\gamma_{\mathcal{U}}), γPU(uu⊤)=γCU(uu⊤)=min⁡{γU(u),γU(−u)}2\gamma_{{\mathcal{P}}_{\mathcal{U}}}(uu^{\top})=\gamma_{{\mathcal{C}}_{\mathcal{U}}}(uu^{\top})=\min\{\gamma_{\mathcal{U}}(u),\gamma_{\mathcal{U}}(-u)\}^{2}. Moreover, for all relaxations in Table 1 (without positivity constraints), we have equality in Eq. (32), a property that will be useful when deriving performance guarantees in the next sections (Prop. 12).

Because the bounds are also polar to each other (from Prop. 10), if we have a performance guarantee for Θ∘\Theta^{\circ}, i.e.,

for some κ⩾1\kappa\geqslant 1, we immediately get

Proof We only consider the case where V{\mathcal{V}} is the unit disk, which gives the main idea of the proof of . We consider a maximizer MM of Eq. (40) and a vector u=D1/2sign(v)u=D^{1/2}\mathop{\rm sign}(v) where vv is a random sample from a normal distribution with mean and covariance matrix D−1/2MD−1/2D^{-1/2}MD^{-1/2}, with D=Diag(diag(M))D=\mathop{\rm Diag}(\mathop{\rm diag}(M)). We have, using standard arguments from , uu⊤∈CU∩{rank(M)=1}=PUuu^{\top}\in{\mathcal{C}}_{\mathcal{U}}\cap\{{\rm rank}(M)=1\}={\mathcal{P}}_{\mathcal{U}} and

The general case is also based on sampling and normalizing Gaussian random variables (see more details in ).

Assume that for all w∈span(U)w\in{\rm span}({\mathcal{U}}) and z∈span(V)z\in{\rm span}({\mathcal{V}}), we have the representations max⁡{γU∘(w),γU∘(−w)}2=γCU∘(ww⊤)\max\{\gamma_{\mathcal{U}}^{\circ}(w),\gamma_{\mathcal{U}}^{\circ}(-w)\}^{2}=\gamma_{{\mathcal{C}}_{\mathcal{U}}}^{\circ}(ww^{\top}) and max⁡{γV∘(z),γV∘(−z)}2=γCV∘(zz⊤)\max\{\gamma_{\mathcal{V}}^{\circ}(z),\gamma_{\mathcal{V}}^{\circ}(-z)\}^{2}=\gamma_{{\mathcal{C}}_{\mathcal{V}}}^{\circ}(zz^{\top}). Then Eq. (41) and Eq. (42) are valid for κ=min⁡{n,d}\kappa=\min\{n,d\}. If moreover, V{\mathcal{V}} is the unit disk, they are also valid for κ=min⁡{n,d}\kappa=\sqrt{\min\{n,d\}}.

Proof Our assumptions imply by duality that for all w∈span(U)w\in{\rm span}({\mathcal{U}}), γU∪(−U)(w)2=inf⁡N∈CUw⊤N−1w=GU(ww⊤)\gamma_{{\mathcal{U}}\cup(-{\mathcal{U}})}(w)^{2}=\inf_{N\in{\mathcal{C}}_{\mathcal{U}}}w^{\top}N^{-1}w=G_{\mathcal{U}}(ww^{\top}), with GU(M)=inf⁡N∈CUMN−1G_{\mathcal{U}}(M)=\inf_{N\in{\mathcal{C}}_{\mathcal{U}}}MN^{-1}, which is defined as, with M=WW⊤∈CUM=WW^{\top}\in{\mathcal{C}}_{\mathcal{U}}, the minimum of trB\mathop{\rm tr}B s.t. \Big{(}\!\begin{array}[]{cc}A&W\\ W^{\top}&N\end{array}\!\Big{)}\succcurlyeq 0. By taking N=WW⊤N=WW^{\top} and A=IA=I, we obtain that GU(M)⩽rank(M)G_{\mathcal{U}}(M)\leqslant{\rm rank}(M).

Thus, there exists u,vu,v such that γU(u)>0\gamma_{\mathcal{U}}(u)>0, γV(v)>0\gamma_{\mathcal{V}}(v)>0 and u^{\top}Yv-\frac{\rho}{\min\{n,d\}}\big{(}\frac{1}{2}\gamma_{\mathcal{U}}(u)^{2}+\frac{1}{2}\gamma_{\mathcal{V}}(v)^{2}\big{)}. This implies that Eq. (41) is valid for κ=min⁡{n,d}\kappa=\min\{n,d\}.

Note that we may also obtain the same results without randomization. When γV=∥⋅∥2\gamma_{\mathcal{V}}=\|\cdot\|_{2}, we have:

Thus by taking vv the largest eigenvector of Y⊤MYY^{\top}MY and u∈argmaxu∈Uu⊤Yvu\in\mathop{\rm argmax}_{u\in{\mathcal{U}}}u^{\top}Yv, we have

with an approximation ratio of min⁡{n,p}\min\{n,p\} between the operator norm and the nuclear norm.

The two previous propositions lead to candidates for vectors uu and vv, through sampling from normal distributions with mean zero and covariance matrix equal to MM (or a normalized version thereof). Note that other possibilies are available, like taking a largest eigenvector (such as done in the proof of Prop. 12) and running the power method from Section 4.5 starting from any of the above candidates.

2 Random sampling

The following proposition provides an approximation ratio with high probability.

Note that the result above is rather weak as on top of the scaling in n\sqrt{n}, the equivalence constants between Θ\Theta and ∥⋅∥F\|\cdot\|_{F} and between γU\gamma_{\mathcal{U}} and ∥⋅∥2\|\cdot\|_{2} may be large as well (though finite because U∘{\mathcal{U}}^{\circ} is compact).

Upper bounds on approximation performance.

Obtaining decompositions from convex relaxations

The first possibility is to follow Section 4.4 and use solutions Q,SQ,S of

and decompose Q=AA⊤Q=AA^{\top} with AA full rank (i.e., A⊤AA^{\top}A invertible), and S=BB⊤S=BB^{\top} with BB full rank. Then U=(AA⊤)−1AGDiag(s)1/2U=(AA^{\top})^{-1}AG\mathop{\rm Diag}(s)^{1/2} and V=(BB⊤)−1BHDiag(s)1/2V=(BB^{\top})^{-1}BH\mathop{\rm Diag}(s)^{1/2}, are such that X=UV⊤X=UV^{\top} and 12γCU(UU⊤)+12γCV(VV⊤)\frac{1}{2}\gamma_{{\mathcal{C}}_{\mathcal{U}}}(UU^{\top})+\frac{1}{2}\gamma_{{\mathcal{C}}_{\mathcal{V}}}(VV^{\top}) is minimal (and equal to the value of the relaxation). However, for any orthogonal matrix RR, (UR,VR)(UR,VR) is also such a pair, and it is not possible to obtain a decomposition of XX such that the gauge functions γU\gamma_{\mathcal{U}} and γV\gamma_{\mathcal{V}} of all columns of UU and VV are small.

2 Conditional gradient algorithms

We may also find decompositions by approximately solving the following convex optimization problem (which is a generalized basis pursuit problem):

for λ\lambda small enough, using convex optimization techniques that only access Θ\Theta through computing Θ∘(Y)=sup⁡(u,v)∈U×Vu⊤Yv\Theta^{\circ}(Y)=\sup_{(u,v)\in{\mathcal{U}}\times{\mathcal{V}}}u^{\top}Yv, and the associated minimizers. This is exactly what generalized conditional gradient algorithms can do . However, we need an algorithm which is robust to obtaining only approximate maximizers (u,v)(u,v), with potentially multiplicative approximation guarantees for the computation of Θ∘\Theta^{\circ}.

We consider the following algorithm started from Z0=0Z_{0}=0, which iterates the following recursion, for ρt=2/(t+1)\rho_{t}=2/(t+1), t⩾1t\geqslant 1:

In Appendix A, we show that if we can find only approximate maximizers (u,v)(u,v) with approximation ratio κ⩾1\kappa\geqslant 1, then, if XλX_{\lambda} is the unique solution of Eq. (44), then we have

and ZtZ_{t} is a positive linear combination of matrices us−1vs−1⊤u_{s-1}v_{s-1}^{\top}, s⩽ts\leqslant t, with a sum of coefficients which is less than Θ(Zt)\Theta(Z_{t}). The previous inequality implies that Θ(Zt)⩽κΘ(X)+O(1/(λt)).\Theta(Z_{t})\leqslant\kappa\Theta(X)+O(1/(\lambda t)). Thus, when λ\lambda is small enough and tt is large enough, we obtain an approximation of Θ(X)\Theta(X) with approximation ratio which converges to a value less than κ\kappa.

Finding such decomposition by greedily and iteratively adding factors has been studied thoroughly in signal processing and statistics . In particular, if we assume that Θ\Theta is a norm, with our set of assumptions, the matching pursuit algorithm of may obtain an ε\varepsilon-approximation of XX with O(log⁡1ε)O(\log\frac{1}{\varepsilon}) rank-one factors, however, while the norm of the coefficients is bounded, it is not related to the decomposition gauge function Θ(X)\Theta(X). In Appendix A.1, we show how optimizing over the scalar ρ\rho in the algorithm above leads to a similar result, while the sum of coefficients converge to a value which is less than κΘ(X)\kappa\Theta(X).

3 Simplicial methods

Conditional gradient algorithms to solve Eq. (44) may be extended by simply replacing step (b), which is the minimization over a half-line, by the minimization with respect to the cone generated by the already obtained rank-one matrices, i.e.,

This approach is sometimes referred to as fully corrective and is an instance of a simplicial method (see, e.g., ); typically, it requires much fewer iterations while the cost of each iteration is higher. Note that when the algorithm stops (i.e., there is no further progress in reducing the cost function), then we have solved Eq. (44) up to λ(κ−1)Ω(x∗)\lambda(\kappa-1)\Omega(x_{\ast}). An inbetween alternative is to optimize only over α\alpha and ρ\rho. Note that the bound derived above also applies to these two extensions, which typically converge much quicker (see examples in Section 7).

Note that when (u,v)(u,v) is obtained from randomized rounding, the algorithm is related to what is proposed by , which uses non-adaptive random sampling for us−1vs−1⊤u_{s-1}v_{s-1}^{\top}. Moreover, the algorithm may be accelerated by only storing only vectors u1,…,ut−1u_{1},\dots,u_{t-1}, and replacing the subproblem in Eq. (45) by

Simulations

We first compared several approaches to estimating Θ∘(Y)=max⁡u∈U, v∈Vu⊤Yv=max⁡u∈{0,1}n∥Y⊤u∥2\Theta^{\circ}(Y)=\max_{u\in{\mathcal{U}},\ v\in{\mathcal{V}}}u^{\top}Yv=\max_{u\in\{0,1\}^{n}}\|Y^{\top}u\|_{2}, for YY a random matrix with independent and identically distributed components from a normal distribution with mean zero and variance one. We consider several strategies: (a) random sampling of uu, then running the power method to convergence, (b) sampling from the solution of the relaxed semi-definite program, with and without running the power method, and (c) taking the non-randomized approach described in the proof of Prop. 12. In Figure 1, we can see that (a) the performance of the semidefinite-relaxation, even without the power method, is typically much better than the guarantee, (b) that sampling from the relaxed solution and then running the power method outperforms random initiatializations, and (c) the non-randomized rounding based on eigenvectors has a less stable behavior and sometimes performs better.

In Figure 1, we report averaged value of the randomized rounding procedures. If we take the best values over more than a thousand samples, the obtained values of u⊤Yvu^{\top}Yv of all three schemes happens to be very close, with a slight advantage to the initializations of the power methods from the convex relaxation.

Finding decomposition.

We aim to solve the problem in Eq. (44) for λ=10−4\lambda=10^{-4}, and consider four approaches: (a) using the semidefinite relaxation to obtain a lower-bound (with no explicit decomposition), (b) using the conditional gradient algorithm, (c) using a simplicial method (the regular version based on Eq. (45) or the one adapted to storing only values of uu in Eq. (46)), (d) using alternating optimization and (e) random sampling (Section 5.2). In Figure 3, we compare these algorithms in two situations, one where the relaxation is tight (left: n=32n=32, d=1d=1) and one where it is not tight (right: n=32n=32, d=16d=16). The simplicial algorithms are the faster to converge, with a clear advantage to the one that stores only the vectors uu. The random selection procedure starts slow but eventually catches up, but never reaches the objective function of adaptive methods.

Finally, in Figure 2, we compare the result of using the simplicial method to obtain a decomposition to a simple alternating optimization method. We see that the convex relaxation outperforms significantly the non-convex approach.

Conclusion

We have presented a general framework for structured matrix decompositions based on gauge functions and semi-definite programming. Emphasis was put on situations where the rank-one factors belong to potentially non-convex and non-centrally symmetric sets. A series of algorithms and relaxations have been presented for a variety of structures.

Our limited experiments have focused on {0,1}\{0,1\}-valued factors. It would be worth studying more precisely recovery guarantees and potentially more scalable algorithms for such cases, in particular for additional constraints such as cardinality or submodularity. Similarly, our general framework leads to altenatives to non-convex algorithms for non-negative matrix factorization, for which it may be possible to show tightness under certain assumptions similar to .

Moreover, it would be interesting to see if techniques designed to adaptively reduce the rank for nuclear-norm penalized problems may be extended to our setting as well in order to enforce a smaller number of factors. Finally, the non-convex approaches (power method and alternating minimization) tend to work well in practice, and finding sufficient conditions or random instances (see, e.g., for such work for sparse principal component analysis) where they provably behave well would provide valuable additional insights into the problem of structured matrix factorizations.

Acknowledgments

This work was supported by the European Research Council (SIERRA project 239993). The author thanks Guillaume Obozinski and Alexandre d’Aspremont for fruitful discussions related to this work.

Appendix A Generalized conditional gradient with approximate oracle

This implies by recursion (see for details) that

If moreover, ff is μ\mu-strongly convex and the line search is performed exactly (and without a bound on α\alpha), then we may show a different bound. Up to the oracle with multiplicative approximation guarantees, the algorithm is then the same as what is proposed by , but our results use a slightly different set of assumptions.

First, we show that f(xt)f(x_{t}) remains bounded. Indeed, we have f(xt)⩽f((1−ρt)xt−1)⩽(1−ρt)f(xt−1)+ρtf(0)f(x_{t})\leqslant f((1-\rho_{t})x_{t-1})\leqslant(1-\rho_{t})f(x_{t-1})+\rho_{t}f(0), which leads to f(xt)⩽f(0)f(x_{t})\leqslant f(0) for all t⩾1t\geqslant 1. This implies that f(0)⩾f(0)+xt⊤f′(0)+μ2∥xt∥22⩾−∥xt∥2∥f′(0)∥2+μ2∥xt∥22f(0)\geqslant f(0)+x_{t}^{\top}f^{\prime}(0)+\frac{\mu}{2}\|x_{t}\|_{2}^{2}\geqslant-\|x_{t}\|_{2}\|f^{\prime}(0)\|_{2}+\frac{\mu}{2}\|x_{t}\|_{2}^{2}, leading to ∥xt∥2⩽2∥f′(0)∥2/μ\|x_{t}\|_{2}\leqslant 2\|f^{\prime}(0)\|_{2}/\mu.

We may then derive a different recursion, leading to

When applied to f(x)=12∥x−y∥22f(x)=\frac{1}{2}\|x-y\|_{2}^{2}, we obtain a decaying factor of

In this case, our algorithm is strongly related to the relaxed greedy algorithm of and our analysis provides an explicit link between these algorithms and basis pursuit .

We now assume that ff is μ\mu-strongly convex and that ff has a global minimum attained at x∗x_{\ast} such that γC(x∗)⩽ω\gamma_{\mathcal{C}}(x_{\ast})\leqslant\omega, and that we have a κ\kappa-approximate oracle for maximizing linear functions on C{\mathcal{C}}. We consider the algorithm:

Moreover, we have f(xt−1)−f(x∗)⩾μ2∥xt−1−x∗∥22⩾μ2γC(xt−1−x∗)2max⁡∥u∥2=1γC(u)2\displaystyle f(x_{t-1})-f(x_{\ast})\geqslant\frac{\mu}{2}\|x_{t-1}-x_{\ast}\|_{2}^{2}\geqslant\frac{\mu}{2}\frac{\gamma_{\mathcal{C}}(x_{t-1}-x_{\ast})^{2}}{\max_{\|u\|_{2}=1}\gamma_{\mathcal{C}}(u)^{2}}, leading to, with Δt=f(xt−1)−f(x∗)\Delta_{t}=f(x_{t-1})-f(x_{\ast}),

If \displaystyle\bigg{[}\Delta_{t-1}+\frac{\varepsilon\sqrt{\mu}\Delta_{t-1}^{1/2}}{\sqrt{2}\max_{\|u\|_{2}=1}\gamma_{\mathcal{C}}(u)}\bigg{]}\frac{1}{L\kappa^{2}\omega^{2}\max_{u\in{\mathcal{C}}}\|u\|^{2}}<1, we have a minimizer ρ∈[0,1)\rho\in[0,1), and

Thus, with \displaystyle\tau=\min\bigg{\{}\frac{1}{2},\frac{\varepsilon^{2}}{4\kappa^{2}\omega^{2}}\frac{\mu}{L}\frac{1}{\max_{\|u\|_{2}=1}\gamma_{\mathcal{C}}(u)\times\max_{u\in{\mathcal{C}}}\|u\|^{2}}\bigg{\}}, we have Δt⩽(1−τ)Δt−1\Delta_{t}\leqslant(1-\tau)\Delta_{t-1}, and hence a linear convergence rate.

Appendix B Maximum of beta random variables

Using standard results from probability (see, e.g., ), we get, with t=n/4t=n/4:

References