# A Survey on Surrogate Approaches to Non-negative Matrix Factorization

- 375 Downloads

## Abstract

Motivated by applications in hyperspectral imaging, we investigate methods for approximating a high-dimensional non-negative matrix * Y* by a product of two lower-dimensional, non-negative matrices

*and*

**K***. This so-called non-negative matrix factorization is based on defining suitable Tikhonov functionals, which combine a discrepancy measure for*

**X***≈*

**Y***with penalty terms for enforcing additional properties of*

**KX***and*

**K***. The minimization is based on alternating minimization with respect to*

**X***and*

**K***, where in each iteration step one replaces the original Tikhonov functional by a locally defined surrogate functional. The choice of surrogate functionals is crucial: It should allow a comparatively simple minimization and simultaneously its first-order optimality condition should lead to multiplicative update rules, which automatically preserve non-negativity of the iterates. We review the most standard construction principles for surrogate functionals for Frobenius-norm and Kullback–Leibler discrepancy measures. We extend the known surrogate constructions by a general framework, which allows to add a large variety of penalty terms. The paper finishes by deriving the corresponding alternating minimization schemes explicitly and by applying these methods to MALDI imaging data.*

**X**## Keywords

Non-negative matrix factorization Multi-parameter regularization Majorize-minimization algorithms Imaging mass spectrometry## Mathematics Subject Classification (2010)

15A23 68W25 65F22## 1 Introduction

Matrix factorization methods for large scale data sets have seen increasing scientific interest recently due to their central role for a large variety of machine learning tasks. The main aim of such approaches is to obtain a low-rank approximation of a typically large data matrix by factorizing it into two smaller matrices. One of the most widely used matrix factorization methods is the principal component analysis (PCA), which uses the singular value decomposition (SVD) of the given data matrix.

In this work, we review the particular case of non-negative matrix factorization (NMF), which is favorable for a range of applications where the data under investigation naturally satisfies a non-negativity constraint. These include dimension reduction, data compression, basis learning, and feature extraction as well as higher level tasks such as classification or clustering [11, 26, 27, 30]. PCA-based approaches without any non-negativity constraints would not lead to satisfactory results in this case since possible negative entries of the computed matrices cannot be easily interpreted for naturally non-negative datasets.

Typically, the NMF problem is formulated as a minimization problem. The corresponding cost function includes a suitable discrepancy term, which measures the difference between the data matrix and the calculated factorization, as well as penalty terms to tackle the non-uniqueness of the NMF, to deal with numerical instabilities but also to provide the matrices with desirable properties depending on the application task. The NMF cost functions are commonly non-convex and require tailored minimization techniques to ensure the minimization but also the non-negativity of the matrix iterates. This leads us to the so-called surrogate minimization approaches, which are also known as majorize-minimization algorithms [18, 23, 36]. Such surrogate methods have been investigated intensively for some of the most interesting discrepancy measures and penalty terms [11, 13, 16, 23, 25, 33, 36]. The idea is to replace the original cost function by a so-called surrogate functional, such that its minimization induces a monotonic decrease of the objective function. It should be constructed in such a way that it is easier to minimize and that the deduced update rules should preserve the non-negativity of the iterates, which typically leads to alternating, multiplicative update rules.

It appears that these constructions are obtained case-by-case employing different analytical approaches and different motivations for their derivation. The purpose of this paper, first of all, is to give a unified approach to surrogate constructions for NMF discrepancy functionals. This general construction principle is then applied to a wide class of functionals obtained by different combinations of divergence measures and penalty terms, thus extending the present state of the art for surrogate-based NMF constructions.

Secondly, one needs to develop minimization schemes for these functionals. Here, we develop concepts for obtaining multiplicative minimization schemes, which automatically preserve non-negativity without the need for further projections.

Finally, we exemplify some characteristic properties of the different functionals with MALDI imaging data, which are particularly high-dimensional and challenging hyperspectral datasets.

The paper is organized as follows. Section 2 introduces the basic definition of the considered NMF problems. Section 3 gives an overview about the theory of surrogate functionals as well as the construction principles. This is then exemplified in Section 4 for the most important cases of discrepancy terms, namely the Frobenius norm and the Kullback–Leibler divergence as well as for a variety of penalty terms. Section 5 discusses alternating minimization schemes for these general functionals with the aim to obtain non-negativity-preserving, multiplicative iterations. Finally, Section 6 contains numerical results for MALDI imaging data.

### 1.1 Notation

Throughout this work, we will denote matrices in bold capital Latin or Greek letters (e.g., * Y*,

*,*

**K****Ψ**,

**Λ**) while vectors will be written in small bold Latin or Greek letters (e.g.

*). The entries of matrices and vectors will be indicated in a non-bold format to distinguish between the*

**c, d, β, ζ***i*th entry

*x*

_{i}of a vector

*and*

**x***n*different vectors

**x**_{j}for

*j*= 1,…,

*n*. In doing so, we write for the entry of a matrix

*in the*

**M***i*th row and the

*j*th column

*M*

_{ij}and the

*i*th entry of a vector

*the symbol*

**x***x*

_{i}. The same holds for an entry of a matrix product: the

*ij*th entry of the matrix product

*will be indicated as (*

**MN***M*

*N*)

_{ij}.

Furthermore, we will use a dot notation to indicate rows and columns of matrices. For a matrix * M*, we will write

**M**_{∙,j}for the

*j*th column and

**M**_{i,∙}for the

*i*th row of the matrix.

What is more, we will use ∥⋅∥ for the usual Euclidean norm, \(\|\boldsymbol {M}\|_{1}:= {\sum }_{ij} |M_{ij}|\) for the 1-norm, and ∥* M*∥

_{F}for the Frobenius norm of a matrix

*.*

**M**Besides that, we will use equivalently the terms *function* and *functional* for a mapping into the real numbers.

Finally, the dimensions of the matrices in the considered NMF problem are reused in this work and will be introduced in the following section.

## 2 Non-negative Matrix Factorization

Before we introduce the basic NMF problem, we give the following definition to clarify the meaning of a non-negative matrix.

### **Definition 1**

A matrix \(\boldsymbol {M}\in \mathbb {R}^{m\times n}\) is called **non-negative** if \(\boldsymbol {M}\in \mathbb {R}_{\geq 0}^{m\times n}\), where \( \mathbb {R}_{\geq 0} := \{x\in \mathbb {R} : x\geq 0\}\).

The non-negativity of an arbitrary matrix * M* will be abbreviated for simplicity as

*≥ 0 in the later sections of this work.*

**M***p*≪ min(

*n*,

*m*) and

*m*data vectors \(\boldsymbol {Y}_{\bullet , j} \in \mathbb {R}^{n}\) for

*j*= 1,…,

*m*, which are stored column-wise in the matrix

*. Similarly for*

**Y***k*= 1,…,

*p*, we denote by

**K**_{∙,k}, respectively

**X**_{k,∙}, the column vectors of

*, respectively the row vectors of*

**K***. We then obtain the following approximation for the column vectors*

**X**

**Y**_{∙,j}as well as the row vectors

**Y**_{i,∙}:

**K**_{∙,k}

**X**_{k,∙}on the right-hand side of the third equation yields rank-one matrices for every

*k*.

By these representations, we can regard the rows **X**_{k,∙} as a low-dimensional set of basis vectors, which are tailored for approximating the high-dimensional data vectors, i.e., NMF solves the task of basis learning with non-negativity constraints.

Following the interpretation given above, we can also regard NMF as a basis for compression. * K* and

*are determined by storing (*

**X***n*+

*m*) ⋅

*p*coefficients, as opposed to

*n*⋅

*m*coefficients for

*. The columns of*

**Y***can be regarded as characteristic components of the given data set {*

**K**

**Y**_{∙,j}}

_{j}. If these data vectors are input for a classification task, one can use the

*p*correlation values with the column vectors of

*as features for constructing the classification scheme.*

**K**The standard variational approach for constructing an NMF is to define a suitable discrepancy measure *D*(⋅,⋅) between * Y* and

*and to minimize the resulting functional. Despite their seemingly simple structure, NMF problems are ill-posed, non-linear, and non-convex, i.e., they require stabilization techniques as well as tailored approaches for minimization. In this paper, we consider discrepancy measures based on divergences [17].*

**KX**### **Definition 2**

**divergence**D is a map \(D: {\Omega } \times {\Omega } \to \mathbb {R}\), which fulfills the following properties:

- (i)
*D*(*x*,*y*) ≥ 0 ∀(*x*,*y*) ∈ Ω×Ω, - (ii)
*D*(*x*,*y*) = 0 ⇔*x*=*y*.

### **Definition 3**

*β*-divergence) The

*β*

**-divergence**\(d_{\beta } : \mathbb {R}_{>0} \times \mathbb {R}_{>0} \rightarrow \mathbb {R}_{\geq 0}\) for \(\beta \in \mathbb {R}\) is defined as

The corresponding matrix divergences are defined componentwise, i.e., *β* = 2 yields the Frobenius norm and *β* = 1 the Kullback–Leibler divergence.

These discrepancy measures are typically amended by so-called penalty terms for stabilization and for enforcing additional properties such as sparsity or orthogonality. This yields the following general minimization task.

### **Definition 4**

**cost functional**. Furthermore, we call

- (i)
*D*_{β}(,**Y**) the**KX****discrepancy term**, - (ii)
*α*_{ℓ}the**regularization parameters**or**weights**, - (iii)
and

*φ*_{ℓ}(,**K**) the**X****penalty terms**.

*,*

**K***). Hence, algorithms based on alternating minimization with respect to*

**X***and*

**K***are favorable, i.e.,*

**X***d*denotes the iteration index of the corresponding matrices.

This yields simpler, often convex restricted problems with respect to either * K* or

*. Considering for example the minimization of the NMF functional with Frobenius norm and without any penalty term yields a high-dimensional linear system \(\boldsymbol {K}^{\intercal } \boldsymbol {Y} = \boldsymbol {K}^{\intercal }\boldsymbol {K} \boldsymbol {X}\), which, however, would need to be solved iteratively.*

**X**Instead, the so-called surrogate methods for computing NMF decompositions have been proposed recently and are introduced in the next section. They also consider alternating minimization steps for * K* and

*, but they replace the restricted minimization problems in (3) and (4) by simpler minimization tasks, which are obtained by locally replacing*

**X***F*by surrogate functionals for

*and*

**K***separately.*

**X**## 3 Surrogate Functionals

In this section, we discuss general surrogate approaches for minimizing general non-convex functionals, which are then exemplified for specific NMF functionals in later sections.

*surrogate minimization*(or

*maximization*)

*algorithms*(SM) or also as

*MM algorithms*, where the first M stands for majorize and the second M for minimize (see also [18, 23, 36]). Such approaches have been demonstrated to be very useful in many fields of inverse problems, in particular for hyperspectral imaging [11], medical imaging applications such as transmisson tomography [13, 14] and MALDI imaging and tumor typing applications [26].

Replacing a non-convex functional by a series of convex problems is the main motivation for such surrogate approaches. However, if constructed appropriately, they can also be used to replace non-differentiable functionals by a series of differentiable problems and they can be tailored such that gradient descent methods for minimization yield multiplicative update rules which automatically incorporate non-negativity constraints without further projections.

From this point on, it is important to note that possible zero denominators during the derivation of the NMF algorithms as well as in the multiplicative update rules themselves will not be discussed explicitly throughout this work. Usually, this issue is handled in practice by adding a small positive constant in the denominator during the iteration scheme. In fact, the instability of NMF algorithms due to the convergence of some entries in the matrices to zero has not been sufficiently discussed in the literature and still needs proper solution techniques. We will not focus on this problem and turn now to the basic definition and properties of surrogate functionals.

### 3.1 Definitions and Basic Properties

As in [25], we use the following definition of a surrogate functional.

### **Definition 5**

**surrogate functional**or a

**surrogate**for F, if it satisfies the following conditions:

- (i)
*Q*_{F}(,**x**) ≥**a***F*() for all**x**,**x**∈ Ω,**a** - (ii)
*Q*_{F}(,**x**) =**x***F*() for all**x**∈ Ω.**x**

*F*(see also Fig. 1).

### **Lemma 1**

*Let*\({\Omega }\subseteq \mathbb {R}^{N}\)

*denote*

*an open set,*\(F: {\Omega }\rightarrow \mathbb {R}\)

*a*

*given function and*

*Q*

_{F}

*a surrogate functional for*

*F. Assume that*argmin

_{x∈ Ω}

*Q*

_{F}(

*,*

**x***)*

**a***is well defined for all*

*∈ Ω.*

**a***Define the iterated updates by*

*with*

**x**^{[0]}:= argmin

_{x∈ Ω}

*Q*

_{F}(

*,*

**x***)*

**a***for an arbitrary*

*∈ Ω.*

**a***Then*,

*F*(

**x**^{[d]})

*is a monotonically decreasing sequence, i.e.*,

### Proof

**x**^{[d+ 1]}in (5). □

### Remark 1

(Addition of surrogate functionals) Let \({\Omega }\subseteq \mathbb {R}^{n}\) be an open set, \(F, G:{\Omega }\rightarrow \mathbb {R}\) pointwise defined functionals and *Q*_{F}, *Q*_{G} corresponding surrogates. Then, *Q*_{F} + *Q*_{G} is a surrogate functional for *F* + *G*.

For each functional *F*, there typically exist a large variety of surrogate functionals and we can aim at optimizing their structure. The following additional property is the key to simple and efficient minimization schemes for surrogate functionals.

### **Definition 6**

*Q*

_{F}a surrogate for

*F*. The surrogate

*Q*

_{F}is called

**separable**, if there exist functions \(g_{i}:\mathbb {R} \times {\Omega } \rightarrow \mathbb {R}\), such that

Lemma 1 above only ensures the monotonic decrease of the cost functional, which is not sufficient to guarantee convergence of the sequence {**x**^{[d]}} to a minimizer of *F* or at least to a stationary point of *F*. The convergence theory for surrogate functionals is far from complete (see also the works [23] and [36]).

Despite this lack of theoretical foundation, surrogate-based minimization yields strictly decreasing sequences for a large variety of applications. In particular, surrogate-based methods can be constructed such that first-order optimality conditions lead to multiplicative update rules, which—in view of the NMF—is a very desirable property.

We now turn to discussing three different construction principles for surrogate functionals.

### 3.2 Jensen’s Inequality

The starting point is the well-known Jensen’s inequality for convex functions (see [10]).

### **Lemma 2**

*(Jensen’s inequality) Let*\({\Omega } \subseteq \mathbb {R}^{N}\)

*denote*

*a convex set,*\(F:{\Omega } \rightarrow \mathbb {R}\)

*a*

*convex function and*

*λ*

_{i}∈ [0,1]

*non-negative numbers for*

*i*∈{1,…,

*k*}

*with*\({\sum }_{i = 1}^{k} \lambda _{i} = 1\).

*Then, for all*

**x**_{i}∈ Ω

*,*

*it holds that*

*F*which are derived from continuously differentiable and convex functions \( f:\mathbb {R}_{>0} \rightarrow \mathbb {R}\) via

*∈ Ω. This also implies that*

**c***F*is convex, since

*λ*

_{i}∈ [0,1] with \({\sum }_{i = 1}^{N} \lambda _{i} = 1\) and \(\boldsymbol {\alpha } \in \mathbb {R}^{N}\) and define

*∈ Ω. This implies*

**b***F*, which can be seen by the inequality above and by observing

### 3.3 Low Quadratic Bound Principle

This concept is based on a Taylor expansion of *F* in combination with a majorization of the quadratic term. This so-called low quadratic bound principle (LQBP) has been introduced in [5] and was used in particular for the computation of maximum likelihood estimators. These methods do not require that *F* itself is convex and its construction is based on the following lemma.

### **Lemma 3**

*Let*\({\Omega } \subseteq \mathbb {R}^{N}\)

*denote*

*an open and convex set and*\(f:{\Omega } \rightarrow \mathbb {R}\)

*a*

*twice continuously differentiable functional. Assume that a*

*matrix*\(\boldsymbol {{\Lambda }} (\boldsymbol {x}) \in \mathbb {R}^{N\times N}\)

*exists,*

*such that*

**Λ**(

*) −∇*

**x**^{2}

*f*(

*)*

**x***is positive semi-definite for all*

*∈ Ω.*

**x***We then obtain a quadratic majorization*

*and*

*Q*

_{f}

*is a surrogate functional for f*.

### Proof

The proof of this classical result is based on the second-order Taylor polynomial of f and shall be left to the reader. □

The related update rule for surrogate minimization can be stated explicitly under natural assumptions on the matrix **Λ**.

### **Corollary 1**

*Assume that the assumptions of Lemma 3 hold. In addition, assume that*

**Λ**

*is a positive definite and symmetric matrix. Then, the corresponding surrogate*

*Q*

_{f}

*is strictly convex in its first variable and we have from*(5)

### Proof

*α*∈{1,…,

*N*}, we have that

**Λ**. Hence, it holds that ∇

_{x}

*Q*

_{f}(

*,*

**x***) = ∇*

**a***f*(

*) +*

**a****Λ**(

*)(*

**a***−*

**x***). The Hessian matrix of*

**a***Q*

_{f}then satisfies

**Λ**is particularly simple if

**Λ**is a diagonal matrix. Furthermore, the diagonal structure ensures the separability of the surrogate functional mentioned in Definition 6. Therefore, we consider matrices of the form

*κ*

_{i}≥ 0 has to be chosen individually depending on the considered cost function. We will see that an appropriate choice of

*κ*

_{i}will lead finally to the desired multiplicative update rules of the NMF algorithm.

The matrix **Λ**(* a*) in (9) fulfills the conditions in Corollary 1 as it can be seen by the following lemma. Therefore, if

**Λ**is constructed as in (9), the update rule in (8) can be applied immediately.

### **Lemma 4**

*Let*\(\boldsymbol {M}\in \mathbb {R}_{\geq 0}^{N \times N}\)

*denote*

*a symmetric matrix. With*\(\boldsymbol {a}\in \mathbb {R}_{>0}^{N}\)

*and*

*κ*

_{i}≥ 0

*,*

*we define the diagonal matrix*

**Λ**

*by*

*for*

*i*= 1,…,

*N*.

*Then*,

**Λ**

*and*

**Λ**−

**M***are positive semi-definite*.

### Proof

*denote the Kronecker symbol. Then,*

**δ****Λ**follows from its diagonal structure. □

### 3.4 Further Construction Principles

So far we have discussed two major construction principles based on either Jensen’s inequality or upper bounds for the quadratic term in Taylor expansions. Lange [23] lists further construction principles, which however will not be used for NMF constructions in the subsequent sections of this paper. For completeness, we briefly list their main properties.

*α*

_{i}≥ 0,

*i*∈{1,…,

*N*} such that \({\sum }_{i = 1}^{N} \alpha _{i} = 1\) and

*α*

_{i}> 0 if

*c*

_{i}≠ 0, which yields

*p*≥ 0. This type of surrogate was originally introduced in the context of medical imaging (see [12]), for positron emission tomography.

## 4 Surrogates for NMF Functionals

In this section, we apply the general construction principles of Section 3 to the NMF problem as stated in (1). The resulting functional *F*(* K*,

*) depends on both factors of the matrix decomposition and minimization is attempted by alternating minimization with respect to*

**X***and*

**K***as described in (3) and (4).*

**X**However, we replace the functional *F* in each iteration by suitable surrogate functionals, which allow an explicit minimization. Hence, we avoid the minimization of *F* itself, which even for the most simple quadratic formulation requires solving a high-dimensional linear system.

We start by considering the discrepancy terms for *β* = 2 (Frobenius norm) and *β* = 1 (Kullback–Leibler divergence) and determine surrogate functionals with respect to * X* and

*. We then add several penalty terms and develop surrogate functionals accordingly. With regard to the construction of surrogates for the case of*

**K***β*= 0 (Itakura–Saito divergence), we refer to the works [15, 16, 32].

### 4.1 Frobenius Discrepancy and Low Quadratic Bound Principle

*for the Frobenius discrepancy*

**X**

**Y**_{∙,j}, resp.

**X**_{∙,j}, denote the column vectors of

*, resp.*

**Y***. The separability of*

**X***F*yields

*κ*

_{k}= 0 yields

*κ*

_{k}ensures the multiplicativity of the final NMF algorithm. In the case of the Frobenius discrepancy term, we will see that suitable

*κ*

_{k}can be chosen dependent on

*ℓ*

_{1}regularization terms in the cost function, which are not included up to now (see Section 4.4 and Appendix 1.1 for more details on this issue). Due to the absent

*ℓ*

_{1}terms, we set

*κ*

_{k}= 0 to get the desired multiplicative update rules. Summing up the contributions of the columns of

*yields the final surrogate*

**X***can be obtained by regarding the rows of*

**K***separately, which for*

**K**### **Theorem 1**

*We consider the*

*cost functionals*\(F(\boldsymbol {X}) := 1/2 \|\boldsymbol {Y}-\boldsymbol {KX}\|_{F}^{2}\)

*and*\(G(\boldsymbol {K}) := 1/2 \|\boldsymbol {Y}-\boldsymbol {KX}\|_{F}^{2}\).

*Then,*

*define separable and convex surrogate functionals.*

### 4.2 Frobenius Discrepancy and Jensen’s Inequality

*; the construction for*

**X***will be very similar. Expanding the Frobenius discrepancy yields*

**K***Q*

_{F,2}(

*,*

**X***) by*

**A***proceeds in the same way. We summarize the results in the following theorem.*

**K**### **Theorem 2**

*We consider the cost*

*functionals*\(F(\boldsymbol {X}) := 1/2 \|\boldsymbol {Y}-\boldsymbol {KX}\|_{F}^{2} \)

*and*\(G(\boldsymbol {K}) := 1/2 \|\boldsymbol {Y}-\boldsymbol {KX}\|_{F}^{2} \).

*Then,*

*define separable and convex surrogate functionals.*

### 4.3 Surrogates for Kullback–Leibler Divergence

*β*= 1 in Definition 3 yields the so-called Kullback–Leibler divergence (KLD). For matrices \(\boldsymbol {M}, \boldsymbol {N}\in \mathbb {R}^{n\times m}_{>0}, \) it is defined as

We will focus in this subsection on Jensen’s inequality for constructing surrogates for the KLD since they will lead to the known classical NMF algorithms (see also [11, 24, 25]). However, it is also possible to use the LQBP principle to construct a suitable surrogate functional for the KLD which leads to different, multiplicative update rules (see Appendix 2).

*, i.e., we consider*

**X***λ*

_{k}and

*α*

_{k}as in the section above and applying it to the convex function

*f*(

*t*) := −ln(

*t*), we obtain

*Y*

_{ij}≥ 0 and the addition of appropriate terms yield

*Q*

_{F,1}(

*,*

**X***) =*

**X***F*(

*) follows by simple algebraic manipulations, such that*

**X***Q*

_{F,1}is a valid surrogate functional for

*F*.

*Q*

_{F,2}and

*Q*

_{F,3}by using, i.e.,

*f*

_{1}(

*t*) =

*Y*

_{ij}ln(

*Y*

_{ij}/

*t*) −

*Y*

_{ij}+

*t*or

*f*

_{2}(

*t*) = −

*Y*

_{ij}ln(

*t*) +

*t*instead of

*f*. Inserting the same

*λ*

_{k}and

*α*

_{k}as before in (7), we obtain immediately the surrogates

*yields the following theorem.*

**K**### **Theorem 3**

*We consider the cost functionals*

*F*(

*) := KL(*

**X***,*

**Y***)*

**KX***and*

*G*(

*) := KL(*

**K***,*

**Y***).*

**KX***Then,*

*define separable and convex surrogate functionals.*

### 4.4 Surrogates for *ℓ* _{1}- and *ℓ* _{2}-Norm Penalties

*ℓ*

_{1}- and

*ℓ*

_{2}-terms for the matrix factors leading to

*β*∈{1,2}.

The *ℓ*_{2}-penalty prohibits exploding norms for each matrix factor and the *ℓ*_{1}-term promotes sparsity in the minimizing factors (see [21, 28] for a general exposition). Combinations of *ℓ*_{1}- and *ℓ*_{2}-norms are sometimes called elastic net regularization [20] due to their importance in medical imaging.

*:*

**X***Q*

_{KL}is the surrogate for the KLD of Theorem 3 for

*.*

**X**The Frobenius case cannot be treated in the same way. If we use the penalty terms as surrogates themselves and obtain the standard minimization algorithm by first-order optimality conditions, then this does not lead to a multiplicative algorithm, which preserves the non-negativity of the iterates. It can be easily seen that the *ℓ*_{1}-penalty term causes this difficulty. For a more extended discussion on this, see Appendix 1.1.

*κ*

_{k}is done dependent on the

*ℓ*

_{1}regularization term of the cost function

*f*

_{y}as already described in Section 4.1. It can be shown in the derivation of the NMF algorithm that

*κ*

_{k}=

*λ*for all

*k*leads to multiplicative update rules. A more general cost function is considered in Appendix 1.1, where the concrete effect of

*κ*

_{k}is described in more detail.

*is then*

**X***we obtain the surrogate by using the diagonal matrix*

**K**### 4.5 Surrogates for Orthogonality Constraints

The observation that a non-negative matrix with pairwise orthogonal rows has at most one non-zero entry per column is the motivation for introducing orthogonality constraints for * K* or

*. This will lead to strictly uncorrelated feature vectors, which is desirable in several applications, e.g., for obtaining discriminating biomarkers from mass spectra (see Section 6 on MALDI imaging).*

**X***and*

**V***and split the orthogonality condition into two second-order terms leading to*

**W**### **Theorem 4**

*We consider the cost*

*functionals*

*with*

*σ*

_{X,1},

*σ*

_{X,2},

*σ*

_{K,1},

*σ*

_{K,2}≥ 0.

*Then,*

*define separable and convex surrogate functionals.*

### 4.6 Surrogates for Total Variation Penalties

Total variation (TV) penalty terms are the second important class of regularization terms besides *ℓ*_{p}-penalty terms. TV penalties aim at smooth or even piecewise constant minimizers; hence, they are defined in terms of first-order or higher-order derivatives [7].

Originally, they were introduced for denoising applications in image processing [31] but have since been applied to inpainting, deconvolution, and other inverse problems (see, e.g., [8]). The precise mathematical formulation of the total variation in the continuous case is described in the following definition.

### **Definition 7**

**total variation**of a function \(u\in L^{1}_{\text {loc}}({\Omega })\) is defined as

There exist several analytic relaxations of TV based on *ℓ*_{1}-norms of the gradient, which are more tractable for analytical investigations. For numerical implementations, one rather uses the *L*_{1}-norm of the gradient \(\|\nabla f\|_{L_{1}}\) as a more computationally tractable substitute. For discretization, the gradient is typically replaced by a finite difference approximation [9].

For applying TV norms to data, we assume that the row index in the data matrix * Y* refers to spatial locations and the column index to so-called channels. In this case, we consider the most frequently used isotropic TV for applying it to measured, discretized hyperspectral data.

### **Definition 8**

*ε*

_{TV}> 0, the

**total variation**of a matrix \(\boldsymbol {K}\in \mathbb {R}_{\geq 0}^{n\times p}\) is defined as

*k*th data channel and

*N*

_{i}⊆{1,…,

*n*}∖{

*i*} denotes the index set referring to spatially neighboring pixels.

**K**_{∙,k}at pixel

*K*

_{ik}for some neighborhood pixels defined by

*N*

_{i}. A typical choice for neighborhood pixels in two dimensions for the pixel (0,0) is

*N*

_{(0,0)}:= {(1,0),(0,1)} to get an estimate of the gradient components along both axes. Finally, by introducing the positive constant

*ε*

_{TV}> 0, we get a differentiable approximation of the total variation penalty.

In Section 6, we will discuss the application of NMF methods to hyperspectral MALDI imaging datasets, which has a natural “spatial structure” in its columns.

In this section, we stay with a generic choice of *N*_{i} as well as of the *ψ*_{k} and we construct a surrogate following the approach of the groundbreaking works of [13] and [29].

*t*≥ 0 and

*s*> 0, we use the inequality (linear majorization)

_{ik}

*with values obtained by an arbitrary non-negative matrix*

**K***:*

**A***i*, multiplication with

*ψ*

_{k}, and summation with respect to

*k*lead to

### **Theorem 5**

*We consider the cost functional*

*F*(

*) := TV(*

**K***)*

**K***with the total variation defined in*(10)

*. Then,*

*defines a separable and convex surrogate functional*.

*C*(

*) denotes some function depending on*

**A***. The description of*

**A***Q*

_{TV}with the help of

*P*(

*)*

**A**_{ik}and

*Z*(

*)*

**A**_{ik}will also allow us to compute the partial derivatives in a more comfortable way (see also Appendix 1.2).

### 4.7 Surrogates for Supervised NMF

As a motivation for this section, we consider classification tasks. We view the data matrix * Y* as a collection of

*n*data vectors, which are stored in the rows of

*. Moreover, we do have an expert annotation*

**Y***u*

_{i}for

*i*= 1,…,

*n*,which assigns a label to each data vector. For a classification problem with two classes, we have

*u*

_{i}∈{0,1}.

As already stated, the rows of the matrix * X* of an NMF decomposition can be regarded as a basis for approximating the rows of

*. Hence, one assumes that the correlations between a row*

**Y**

**Y**_{i,∙}of

*and all row vectors of*

**Y***, i.e., computing \(\boldsymbol {Y}_{i,\bullet }\boldsymbol {X}^{\intercal }\), contain the relevant information of*

**X**

**Y**_{i,∙}. The vector of correlations yields the so-called feature vector of length

*p*. A classical linear regression model, which uses these feature vectors, then asks to compute weights

*β*

_{k}for

*k*= 1,…,

*p*, such that \(\boldsymbol {Y}_{i,\bullet } \boldsymbol {X}^{\intercal }\boldsymbol {\beta } \approx u_{i}\) (for more details on linear discriminant analysis methods, we refer to [4, Chapter 4]).

*as a minimizer of*

**β***and*

**X***to define*

**β***is obtained by MALDI measurements, the vector*

**Y**

**x**^{∗}can be interpreted as a characteristic mass spectrum of some specific tumor type and can be directly used for classification tasks in the arising field of digital pathology (see also Section 6 and [26]).

The classification of a new data vector * y* is then simply obtained by computing the scalar product \(w={\boldsymbol {x}^{\ast }}^{\intercal } \boldsymbol {y}\) and assigning either the class label 0 or 1 by comparing

*w*with a pre-assigned threshold

*s*. This threshold is typically obtained in the training phase of the classification procedure by computing \(\boldsymbol {YX}^{\intercal }\boldsymbol {\beta }\) for some given training data

*and choosing*

**Y***s*, such that a performance measure of the classifier is optimized.

*and*

**K***, and then computing the weights*

**X***of the classifier. Hence, the computation of the NMF is done independently of the class labels*

**β***, which is also referred to as an unsupervised NMF approach. We might expect that computing the NMF by minimizing a functional which includes the class labels, i.e.,*

**u***and*

**X***:*

**β***f*and auxiliary variables \(\boldsymbol {c}\in \mathbb {R}^{N}_{>0}\) and \(\boldsymbol {d}\in \mathbb {R}^{M}_{>0}\). We now use the following generalized Jensen’s inequality

*i*∈{1,…,

*n*} yields by inserting

*λ*

_{jk}and

*α*

_{jk}into (11)

*proceeds analogously. We summarize the results in the following theorem.*

**β**### **Theorem 6**

*Let*\( F(\boldsymbol {X}) := 1/2\|\boldsymbol {u} - \boldsymbol {YX}^{\intercal } \boldsymbol {\beta }\|^{2}\)

*and*\( G(\boldsymbol {\beta }) := 1/2\|\boldsymbol {u} - \boldsymbol {YX}^{\intercal } \boldsymbol {\beta } \|^{2} \)

*denote*

*a cost functional with repect to*

**X***and*

*.*

**β***Then,*

*define separable and convex surrogate functionals.*

A big advantage of linear regression models is their simplicity and manageability. However, they are by far not the optimal approach to approximate the binary output data * u* with a continuous input. Logistic regression models offer a way more natural method for binary classification tasks. Together with the supervised NMF as a feature extraction method, this overall workflow leads in [26] to excellent classification results and outperformed classical approaches.

However, the proposed model is based on a gradient descent approach, such that the non-negativity of the iterates can only be guaranteed by a projection step. Appropriate surrogate functionals for this workflow are still ongoing research and could lead to even better outcomes (see also [35, 36]).

## 5 Surrogate-Based NMF Algorithms

In the previous section, we have defined surrogate functionals for various NMF cost functions. Besides the necessary surrogate properties, we also expect that the minimization of these surrogates is straightforward and can be computed efficiently.

In our case, we demand additionally that the minimization schemes based on solving the first-order optimality conditions leads to a separable algorithm and that it only requires multiplicative updates, which automatically preserve the non-negativity of its iterates. Let us start with denoting the most general functional with Kullback–Leibler divergence, the Frobenius case follows similarly.

For constructing non-negative matrix factorizations, we incorporate *ℓ*_{2}-, sparsity-, orthogonality-, and TV-constraints and also the penalty terms coming from the supervised NMF. Of course, in most applications, one only uses a subset of these constraints for stabilization and for enhancing certain properties. These algorithms can readily be obtained from the general case by putting the respective regularization parameters to zero. The corresponding update rules are classical results and can be found in numerous works [11, 24, 25].

### **Definition 9**

*λ*,

*μ*,

*ν*,

*ω*,

*τ*,

*σ*

_{K,1},

*σ*

_{K,2},

*σ*

_{X,1},

*σ*

_{X,2},

*ρ*≥ 0, we define the NMF minimization problem by

The choice of the various regularization parameters occurring in Definition 9 is often based on heuristic approaches. We will not focus on that issue in this work and refer instead to [19] and the references therein, where two methods are investigated for the general case of multi-parameter Tikhonov regularization.

The algorithms studied in this section will start with positive initializations for * K*,

*,*

**X***,*

**V***, and*

**W***. These matrices are updated alternatingly, i.e., all matrices except one matrix are kept fixed and only the selected matrix is updated by solving the respective first-order optimality condition.*

**β**We will focus in this section on the derivation of the update rules of * K* (see also Appendix 1.2). The iteration schemes for the other matrices follow analogously.

*, i.e., we aim at determining a minimizer for*

**K***Q*

_{KL}for the Kullback–Leibler divergence in Theorem 3,

*Q*

_{TV}for the TV penalty term in Theorem 5 and

*Q*

_{Orth}for the orthogonality penalty terms in Theorem 4.

*ξ*∈{1,…,

*n*} and

*ζ*∈{1,…,

*p*}. This leads to a system of quadratic equations

*K*

_{ξζ}and denoting the Hadamard product by ∘ as well as the matrix division for each entry separately by a fraction line yield the following update rule for

*. (Note that the notation for*

**K***P*(

*) and*

**A***Z*(

*) was introduced in the section on TV regularization above.)*

**A**

**1**_{n×p}denotes an

*n*×

*p*matrix with ones in every entry and \(\boldsymbol {\Psi }\in \mathbb {R}^{n\times p}_{\geq 0}\) is defined as

*are computed similarly. Defining*

**X***,*

**V***are straight forward and we obtain the following theorem.*

**W**### **Theorem 7**

*The*

*initializations*\(\boldsymbol {K}^{[0]}, \boldsymbol {V}^{[0]} \in \mathbb {R}_{> 0}^{n\times p}\)

*,*\(\boldsymbol {X}^{[0]}, \boldsymbol {W}^{[0]} \in \mathbb {R}_{> 0}^{p\times m}\)

*,*\(\boldsymbol {\beta }^{[0]}\in \mathbb {R}_{> 0}^{p}\)

*and*

*the iterative updates*

*lead to a monotonic decrease of the cost functional*

*ℓ*

_{1}- and

*ℓ*

_{2}-regularized NMF, this leads to the cost function

*is obtained by setting*

**X***of the previous theorem—leads to*

**X***,*

**V***,*

**W***as in Theorem 7 (see also Appendix 1.1 for more details on the derivation of these algorithms.)*

**β**## 6 MALDI Imaging

As a test case, we analyze MALDI imaging data (matrix-assisted laser desorption/ionization) of a rat brain. MALDI imaging is a comparatively novel modality, which unravels the molecular landscape of tissue slices and allows a subsequent proteomic or metabolic analysis [1, 6, 22]. Clustering this data reveals for example different metabolic regions of the tissue, which can be used for supporting pathological diagnosis of tumors.

In our numerical experiments, we used a classical rat brain dataset which has been used in several data processing papers before [2, 3, 22]. It constitutes a standard test set for hyperspectral data analysis.

The tissue slice was scanned at 20,185 positions. At each position, a full mass spectrum with 2974 *m*/*z* (mass over charge) values was collected, i.e., instead of three color channels, as it is usual in image processing, this data has 2974 channels, each channel containing the spatial distribution of molecules having the same *m*/*z* value.

*as the pseudo channels of the NMF decomposition or the rows of*

**K***as pseudo spectra characterizing the different metabolic processes present in the tissue slice (see the Figs. 3, 4, 5, and 6).*

**X**Both ways of visualization do have their respective value. Looking at the pseudo spectra in connection with orthogonality constraints leads to a clustering of the spectra and to a subdivision of the tissue slice in regions with potentially different metabolic activities (see [22]). Considering instead the different pseudo spectra, which were constructed in order to have a base which allows a low-dimensional approximation of the dataset, is the basis for subsequent proteomic analysis, e.g., one may target pseudo spectra where the related pseudo channels are concentrated in regions, which were annotated by pathological experts. Mass values which are dominant in those spectra may stem from proteins/peptides relating to biomarkers as indicators for certain diseases. Hence, classification schemes based on NMF decompositions have been widely investigated [26, 30, 34].

## 7 Conclusion

In this paper, we investigated methods based on surrogate minimization approaches for the solution of NMF problems. The interest in NMF methods is related to its importance for several machine learning tasks. Application for large datasets requires that the resulting algorithms are very efficient and that iteration schemes only need simple matrix-vector multiplications.

The state of the art for constructing appropriate surrogates is based on case-by-case studies for the different considered NMF models. In this paper, we embedded the different approaches in a general framework, which allowed us to analyze several extensions to the NMF cost functional, including *ℓ*_{1}- and *ℓ*_{2}-regularization, orthogonality constraints, and total variation penalties as well as extensions, which led to supervised NMF concepts.

Secondly, we analyzed surrogates in the context of the related iteration schemes, which are based on first-order optimality conditions. The requirement of separability as well as the need of having multiplicative updates, which preserve non-negativity without additional projections, was analyzed. This resulted in a general description of algorithms for alternating minimization of constrained NMF functionals. The potential of these methods is confirmed by numerical tests using hyperspectral data from a MALDI imaging experiment.

Several further directions of research would be of interest. First of all, besides the most widely used penalty terms discussed in this paper, further penalty terms, e.g., higher-order TV terms, could be considered. Secondly, construction principles for more general discrepancy terms could be analyzed (see also [16]).

Potentially more importantly, this paper contains only very first results for combining NMF constructions directly with subsequent classification tasks. The question of an appropriate surrogate functional for the supervised NMF model with logistic regression used in [26] remains unanswered and also the comparison with algorithmic alternatives such as ADMM methods needs to be explored.

## Notes

### Acknowledgements

The authors want to thank Christine De Mol for her excellent presentations at several conferences and our joint discussions, which were the starting point for this paper. The presented results are based on the Master Thesis of the first author.

### Funding Information

This project was supported by the Deutsche Forschungsgemeinschaft (DFG) within the framework of GRK 2224/1 “*π*^{3}: Parameter Identification - Analysis, Algorithms, Applications”.

## References

- 1.Aebersold, R., Goodlett, D.R.: Mass spectrometry in proteomics. Chem. Rev.
**101**, 269–296 (2001)CrossRefGoogle Scholar - 2.Alexandrov, T., Bartels, A.: Testing for presence of known and unknown molecules in imaging mass spectrometry. Bioinformatics
**29**, 2335–2342 (2013)CrossRefGoogle Scholar - 3.Alexandrov, T., Becker, M., Deininger, S.-O., Ernst, G., Wehder, L., Grasmair, M., von Eggeling, F., Thiele, H., Maass, P.: Spatial segmentation of imaging mass spectrometry data with edge-preserving image denoising and clustering. J. Proteome Res.
**9**, 6535–6546 (2010)CrossRefGoogle Scholar - 4.Bishop, C.: Pattern Recognition and Machine Learning. Springer-Verlag, New York (2006)zbMATHGoogle Scholar
- 5.Böhning, D., Lindsay, B.G.: Monotonicity of quadratic-approximation algorithms. Ann. Inst. Stat. Math.
**40**, 641–663 (1988)MathSciNetCrossRefGoogle Scholar - 6.Caprioli, R.M., Farmer, T.B., Gile, J.: Molecular imaging of biological samples: localization of peptides and proteins using MALDI-TOF MS. Anal. Chem.
**69**, 4751–4760 (1997)CrossRefGoogle Scholar - 7.Chambolle, A., Caselles, V., Cremers, D., Novaga, M., Cremers, D., Pock, T.: An introduction to total variation for image analysis. In: Fornasier, M. (ed.) Theoretical Foundations and Numerical Methods for Sparse Recovery. Radon Series on Computational and Applied Mathematics, vol. 9, pp 263–340. Walter de Gruyter, Berlin (2010)Google Scholar
- 8.Chan, T., Esedoglu, S., Park, F., Yip, A.: Total variation image restoration: Overview and recent developments. In: Paragios, N., Chen, Y., Faugeras, O. (eds.) Handbook of Mathematical Models in Computer Vision, pp 17–31. Springer, Boston (2006)Google Scholar
- 9.Condat, L.: Discrete total variation: New definition and minimization. SIAM J. Imaging Sci.
**10**, 1258–1290 (2017)MathSciNetCrossRefGoogle Scholar - 10.Cvetkovski, Z.: Inequalities - Theorems, Techniques and Selected Problems. Springer-Verlag, Berlin Heidelberg (2012)zbMATHGoogle Scholar
- 11.De Mol, C.: Regularized Multiplicative Algorithms for Nonnegative Matrix Factorization. Methodological Aspects of Hyperspectral Imaging Workshop, Nice (2013)Google Scholar
- 12.De Pierro, A.R.: On the relation between the ISRA and the EM algorithm for positron emission tomography. IEEE Trans. Méd. Imaging
**12**, 328–333 (1993)CrossRefGoogle Scholar - 13.Defrise, M., Vanhove, C., Liu, X.: An algorithm for total variation regularization in high-dimensional linear problems. Inverse Probl.
**27**, 065002 (2011)MathSciNetCrossRefGoogle Scholar - 14.Fessler, J.A.: Statistical image reconstruction methods for transmission tomography. In: Sonka, M., Fitzpatrick, J. (eds.) Medical Image Processing and Analysis. Handbook of Medical Imaging, vol. 2, pp 1–70. SPIE Press, Bellingham (2000)Google Scholar
- 15.Févotte, C., Bertin, N., Durrieu, J.-L.: Nonnegative matrix factorization with the Itakura-Saito divergence: With application to music analysis. Neural Comput.
**21**, 793–830 (2009)CrossRefGoogle Scholar - 16.Févotte, C., Idier, J.: Algorithms for nonnegative matrix factorization with the
*β*-divergence. Neural Comput.**23**, 2421–2456 (2011)MathSciNetCrossRefGoogle Scholar - 17.Hennequin, R., David, B., Badeau, R.: Beta-divergence as a subclass of Bregman divergence. IEEE Signal Process. Lett.
**18**, 83–86 (2011)CrossRefGoogle Scholar - 18.Hunter, D.R., Lange, K.: A tutorial on MM algorithms. Am. Stat.
**58**, 30–37 (2004)MathSciNetCrossRefGoogle Scholar - 19.Ito, K., Jin, B., Takeuchi, T.: Multi-parameter Tikhonov regularization—an augmented approach. Chin. Ann. Math. Ser. B
**35**, 383–398 (2014)MathSciNetCrossRefGoogle Scholar - 20.Jin, B., Lorenz, D.A., Schiffler, S.: Elastic-net regularization: error estimates and active set methods. Inverse Probl.
**25**, 115022 (2009)MathSciNetCrossRefGoogle Scholar - 21.Jin, B., Maass, P.: Sparsity regularization for parameter identification problems. Inverse Probl.
**28**, 123001 (2012)MathSciNetCrossRefGoogle Scholar - 22.Kobarg, J.H., Maass, P., Oetjen, J., Trop, O., Hirsch, E., Sagiv, C., Golbabaee, M., Vandergheynst, P.: Numerical experiments with MALDI imaging data. Adv. Comput. Math.
**40**, 667–682 (2014)MathSciNetCrossRefGoogle Scholar - 23.Lange, K.: Optimization, 2nd edn. Springer Texts in Statistics, vol. 95. Springer-Verlag, New York (2013)Google Scholar
- 24.Lecharlier, L., De Mol, C.: Regularized blind deconvolution with Poisson data. J. Phys.: Conf. Ser.
**464**, 012003 (2013)Google Scholar - 25.Lee, D.D., Seung, H.S.: Learning the parts of objects by non-negative matrix factorization. Nature
**401**, 788–791 (1999)CrossRefGoogle Scholar - 26.Leuschner, J., Fernsel, P., Schmidt, M., Lachmund, D., Boskamp, T., Maass, P.: Supervised non-negative matrix factorization methods with MALDI-imaging applications. Bioinformatics (in review) (2018)Google Scholar
- 27.Li, T., Ding, C.: Non-negative matrix factorization for clustering: a survey. In: Aggarwal, C. C., Reddy, C. (eds.) Data Clustering: Algorithms and Applications, pp 149–176. CRC Press, Boca Raton (2013)Google Scholar
- 28.Louis, A.K.: Inverse und Schlecht Gestellte Probleme. Vieweg+Teubner, Verlag (1989)CrossRefGoogle Scholar
- 29.Oliveira, J.P., Bioucas-Dias, J.M., Figueiredo, M.A.T.: Review: Adaptive total variation image deblurring: a majorization–minimization approach. Signal Process.
**89**, 1683–1693 (2009)CrossRefGoogle Scholar - 30.Phon-Amnuaisuk, S.: Applying non-negative matrix factorization to classify superimposed handwritten digits. Proced. Comput. Sci.
**24**, 261–267 (2013)CrossRefGoogle Scholar - 31.Rudin, L.I., Osher, S., Fatemi, E.: Nonlinear total variation based noise removal algorithms. Phys. D
**60**, 259–268 (1992)MathSciNetCrossRefGoogle Scholar - 32.Sun, D.L., Févotte, C.: Alternating direction method of multipliers for non-negative matrix factorization with the beta-divergence. In: 2014 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), pp. 6201–6205 (2014)Google Scholar
- 33.Tan, V.Y.F., Févotte, C.: Automatic relevance determination in nonnegative matrix factorization with the
*β*-divergence. arXiv:1111.6085v3 (2012) - 34.Tang, J., Ceng, X., Peng, B.: New methods of data clustering and classification based on NMF. In: 2011 International Conference on Business Computing and Global Informatization, pp. 432–435 (2011)Google Scholar
- 35.Zhang, Z., Kwok, J.T., Yeung, D.-Y.: Surrogate maximization/minimization algorithms for AdaBoost and the logistic regression model. In: Proceedings of the Twenty-first International Conference on Machine Learning, ICML ’04 (2004)Google Scholar
- 36.Zhang, Z., Kwok, J.T., Yeung, D.-Y.: Surrogate maximization/minimization algorithms and extensions. Mach. Learn.
**69**, 1–33 (2007)CrossRefGoogle Scholar

## Copyright information

**Open Access** This article is distributed under the terms of the Creative Commons Attribution 4.0 International License (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made.