# Class-specific discriminant time-frequency analysis using novel jointly learnt non-negative matrix factorization

- Behnaz Ghoraani
^{1}Email author

**2016**:95

https://doi.org/10.1186/s13634-016-0393-4

© The Author(s) 2016

**Received: **9 March 2016

**Accepted: **29 August 2016

**Published: **7 September 2016

## Abstract

Time-frequency (TF) representation has found wide use in many challenging signal processing tasks including classification, interference rejection, and retrieval. Advances in TF analysis methods have led to the development of powerful techniques, which use non-negative matrix factorization (NMF) to adaptively decompose the TF data into TF basis components and coefficients. In this paper, standard NMF is modified for TF data, such that the improved TF bases can be used for signal classification applications with overlapping classes and data retrieval. The new method, called jointly learnt NMF (JLNMF) method, identifies both distinct and shared TF bases and is able to use the decomposed bases to successfully retrieve and separate the class-specific information from data. The paper provides the framework of the proposed JLNMF cost function and proposes a projected gradient framework to solve for limit point stationarity solutions. The developed algorithm has been applied to a synthetic data retrieval experiment and epileptic spikes in EEG signals of infantile spasms and discrimination of pathological voice disorder. The experimental results verified that JLNMF successfully identified the class-specific information, thus enhancing data separation performance.

## Keywords

## 1 Introduction

Automated decision-making systems have to deal with natural signals that are mostly nonstationary in nature, meaning that their statistics vary over time and frequency. Time-frequency (TF) analysis [1, 2] has been widely used over the past decades to characterize the nonstationary content of signals in the joint time and frequency domain. However, the main limitation of the TF-methods is the high dimensionality of the representation. Therefore, in order to effectively analyze natural signals in artificial intelligence and decision-making applications, it is necessary to develop dimensionality reduction techniques that can remove the redundant data and capture the essential features of the TF representation. Among such techniques, TF decomposition is an effective approach, where a matrix decomposition scheme is used to adaptively decompose the TF data into representing TF bases for further analysis [3–9].

Given the non-negative nature of TF data, non-negative matrix (NMF) decomposition has received favorable attention and been widely used for TF feature extraction applications. Several variants of NMF have been proposed over the past decade. Those methods impose additional constraints, such as localization or sparsity, on the NMF cost function to construct improved NMF decompositions. For example, Zafeiriou et al. [10] and Nikitidis et al. [11] have proposed a discriminative NMF, which adds additional constraints to maximize the between-class scatter and minimize the within-class scatter in the subspace spanned by the bases in order to improve the discrimination power of the decomposed TF bases. However, those methods are designed to classify the data to only one class at a time, and cannot separate the information as needed in data retrieval applications. In this paper, we developed a jointly learnt NMF (JLNMF) method to jointly learn the class-specific TF bases that are discriminant to each class as well as the ones that are shared between classes, and use the discriminant class-specific TF bases to identify how much of each class presents in a test data. To make this happen, a set of new constraints is enforced on the NMF cost function as explained in the present work.

The rest of the paper is organized as follows. The background and theoretical framework of the proposed algorithm is reviewed in Section 2. Section 3 introduces the proposed JLNMF algorithm and develops an optimization strategy to solve the proposed JLNMF cost function. Section 4 investigates the properties of the proposed JLNMF algorithm and presents the conducted experimental study and verifies the efficiency of JLNMF for data retrieval and information localization applications. Finally, concluding remarks are drawn in Section 5.

## 2 Theoretical framework

### 2.1 Time-frequency matrix decomposition

*x*(

*m*), its TF representation can be denoted by a matrix

**V**

_{ N×M }, where

*M*is the number of time samples in signal

*x*(

*m*) and

*N*is the number of frequency bins (i.e., frequency resolution). For example, the value of

**V**(

*n*,

*m*), for

*n*=1,…,

*N*and

*m*=1,…,

*M*, indicates the amount of signal energy at time sample

*m*and frequency of

*n*. The application of a matrix decomposition technique with

*r*coefficients on the TF matrix of

**V**can be written as following:

where the decomposed TF matrices, **W**
_{
N×r
} and **H**
_{
r×M
} (with *r*<<*N*,*M*) contain the TF bases and the corresponding coefficients of the linear combinations of the TF bases required to reconstruct each TF data, respectively, and are defined as **W**
_{
N×r
}=[*w*
_{1}
*w*
_{2} ⋯ *w*
_{
r
}] and **H**
_{
r×M
}=[*h*
_{1}
*h*
_{2} ⋯ *h*
_{
r
}]^{
T
}. In Eq. (1), TF matrix (**V**) is reduced to base and coefficient vectors ({*w*
_{
i
}}_{
i=1,…,r
} and \(\left \{{h^{T}_{i}}\right \}_{i=1,\ldots,r}\), respectively) subject to a given constraint on **W** and **H**.

**W**≥0 and

**H**≥0) and has been widely used in TF decomposition applications. NMF derives the TF bases and coefficients (i.e.,

**W**and

**H**, respectively) by minimizing the least-square cost function as shown below:

**W**and

**H**, and performs an iterative optimization to minimize the cost function in Eq. (2). In [12], Lee and Seung introduce two updating algorithms using the least square error and the Kullback-Leibler (KL) divergence as the cost functions. The equations, using the least square error method, are as follows:

In these equations, 〈. 〉 and \(\frac {\langle \: \ \rangle }{\langle \: \ \rangle }\) are element-wise multiplication and division of two matrices. Various alternative minimization strategies for NMF decomposition have been proposed [12–14]. In this work, we use a projected gradient bound-constrained optimization approach in [15]. The gradient-based NMF is computationally competitive and offers better convergence properties than [12].

### 2.2 Need for jointly learnt NMF (JLNMF) method

**V**

_{1}and

**V**

_{2}in Fig. 1 a) to decompose class-specific TF bases (i.e.,

**W**

_{1}and

**W**

_{2}) and create an overall TF basis matrix, consisting of all the class-specific TF bases. Then, a new test TF data is decomposed over the overall TF basis matrix to obtain the corresponding coefficients (i.e., \({h^{T}_{1}}\) and \({h^{T}_{2}}\)) of the linear combinations of the class-specific TF bases required to reconstruct the new TF data. Finally, the TF bases of each class and the corresponding coefficients can be used to reconstruct the part of the test data that belongs to each class (i.e., \(\hat {\textbf {V}}_{\text {Test}1}\) and \(\hat {\textbf {V}}_{\text {Test}2}\)). Figure 1 a shows a general schematic of this approach for a two-group classification scenario.

The approach in Fig. 1
a is effective if the decomposed TF bases are not only representative of the TF structure of each class but also discriminative to the TF structures from different classes. Inspired by image recognition applications, much outstanding research on the design of discriminant NMF techniques has been reported [11, 16]. For the most part, those methods have focused on increasing the discrimination by enforcing a penalty to maximize the discrimination between the decomposed bases of different classes. However, the decomposed bases could be shared between classes, and it is not always possible to identify class-specific TF bases as not necessarily there is a unique relationship between the decomposed bases and class labels. As a result, those methods could be effective in improving the accuracy rate in image classification applications, but cannot be employed in challenging signal processing applications, including classification of overlapping classes, and data retrieval. In this paper, a novel JLNMF method is proposed to decompose discriminant and class-specific TF bases. This new approach enforces the discrimination between the TF bases of different classes by modeling the similarities between classes with a set of shared TF bases (i.e., **W**
_{
j
} in Fig. 1
b) and decomposes class-specific TF bases (i.e., **W**
_{1d
} and **W**
_{2d
}) to model the distinct TF structure of each class. Decomposing a new data, with overlapping classes, over the entire discriminant and shared TF bases will identify the portion of the data that belongs to each class, and the corresponding coefficients can be used to successfully retrieve and separate the data from each class (i.e., \(\hat {\textbf {V}}_{\text {Test}1d}\) and \(\hat {\textbf {V}}_{\text {Test}2d}\)). Figure 1
b shows a general schematic of the proposed JLNMF approach for a two-group classification scenario.

The idea of including shared TF bases to enhance the discrimination power is inspired by an assumption made by the classification techniques, where they find a discriminating pattern between classes by dividing the feature space into non-overlapping subspaces, which each represents one of the classes. Although, this approach might be satisfactory in cases that the signals can be assumed to be completely separable in the feature space, and it seems to be too optimistic in applications where a natural and unavoidable overlap exists between different classes. In most real-world applications, specially in biomedical applications, the nature of signals from different classes is very similar, and there could only be slight changes in the TF patterns of signals from different classes. Natural similarities between different classes may result in some overlaps in the feature space, and the extracted features may not necessarily represent the discriminating structures in each class. This may cause some overlaps in the feature space, and thereby may degrade the performance of the classification task. Hence, identifying and removing those shared components in the TF bases from a classification task can improve the discrimination power between the remaining class-specific features. In order to improve the discrimination power of the TF bases of each class, in the present work, we account for those similarities between classes by inclusion of shared TF bases in the formulation of the NMF decomposition. It is worth mentioning that in a work by [17], the authors have proposed a non-negative matrix partial co-factorization approach for retrieving the information of drum track from a compound music of a drum and harmony track. The authors use pure drum music from a different source and assume that all the TF bases of the drum music are shared with the compound music. Then the two music (i.e., drum and compound music) are simultaneously decomposed by enforcing the constraint that all the bases of the drum track are shared with the music compound and there are only distinct bases of harmony tracks in the music compound. Basically, there is only **W**
_{
j
} and **W**
_{
d
} in the proposed model by Yoo et al. instead of **W**
_{
j
}, **W**
_{1d
}, and **W**
_{2d
}. Hence, the model by Yoo et al. can be considered as a subset of the developed framework in the present work and can be considered only if the information of one of the classes is fully shared with the other class.

## 3 Jointly learnt NMF (JLNMF) method

### 3.1 Formulation of JLNMF

**W**into two parts of

**W**

_{ d }and

**W**

_{ j }, where the former corresponds to the discriminant class-specific TF bases, and the latter represents the shared TF bases (see Fig. 1 b). Using the new notation, Eq. (4) can be rewritten as follows:

**W**

_{1},

**W**

_{1d }, and

**W**

_{ j }is formulated as follows:

*r*=

*r*

_{ d }+

*r*

_{ j }equals to the total number of discriminant and shared components (i.e.,

*r*

_{ d }and

*r*

_{ j }, respectively). Then, we can get the following:

**W**

_{1d },

**W**

_{2d }, and

**W**

_{ j }matrices as shown in Fig. 2.

where *δ* and *λ* are positive constants that control the trade-off between minimizing the reconstruction error and maximizing the discrimination power. The order of the terms in Eq. (9) follows the numbering of the arrows in Fig. 2.

### 3.2 Optimization of JLNMF

*g*in Eq. (9), the partial derivatives of

*g*with respect to each unknown variable is obtained and set equal to zero as follows:

**W**

_{ j }is the variable that the equation in being solved for, and the terms \(\left (\textbf {H}_{1}\textbf {H}_{1}^{\mathrm {T}}+\textbf {H}_{2}\textbf {H}_{2}^{\mathrm {T}}\right)\) and \(\left (1+\lambda \right)\textbf {W}_{1d}\textbf {H}_{1}\textbf {H}_{1}^{\mathrm {T}} + \left (1+\lambda \right)\textbf {W}_{2d}\textbf {H}_{2}\textbf {H}_{2}^{\mathrm {T}}- \textbf {V}_{1}\textbf {H}_{1}^{\mathrm {T}}-\textbf {V}_{2}\textbf {H}_{2}^{\mathrm {T}} \) are constant matrices. Thus, using the same proof that was shown in [12] to get the NMF solutions in Eq. (3), it can be concluded that Eq. (9) is convergent and non-increasing when alternatively one matrix is improved while the other matrices are fixed, and the solution of the optimization problem in Eq. (9) can be solved by iteratively solving the following sub-problems [18, 19]:

Although the optimization problem of Eq. (9) is not convex over variables **W**
_{
j
}, **W**
_{1d
}, **W**
_{2d
}, **H**
_{1}, and **H**
_{2}, the sub-problems in Eqs. (20) to (24) are convex and can be solved using projected gradient methods as follows.

**U**

^{(t)}to

**U**

^{(t+1)}by the following updating rule:

*t*) is the iteration order, ∇

*g*(

**U**) is the projected gradient of the function

*g*with respect to

**U**, while all the other matrices are constant, and

*α*

^{(t)}is the step size to update the matrix. The step size is found as \(\alpha ^{(t)}=\beta ^{k_{t}}\phantom {\dot {i}\!}\), where

*k*

_{ t }is the first non-negative integer

*k*for which the following equation holds:

where 0<*σ*<1 is a scalar parameter that is defined based on the convexity of the function *g*(**U**) [20]. The condition in Eq. (27) ensures the sufficient decrease of the function value per iteration. Bertsekas [20] has proved that *α*
^{(t)}>0 satisfying Eq. (27) always exists and every limit point of the solution \(\left \{\textbf {U}^{(t)}\right \}^{\infty }_{t=1}\) is a stationary point of Eq. (25). Therefore, by iteratively updating each matrix using updating rule in Eq. (26), while the other matrices are constant, we can iteratively solve for **W**
_{
j
}, **W**
_{1d
}, **W**
_{2d
}, **H**
_{1}, and **H**
_{2}, in the sub-problems of Eqs. (20) to (24).

### 3.3 Computation cost considerations

The computation cost for each iteration consists of computation of the gradient functions of Eqs. (15)-(19) to update current variables using Eq. (26) and the computational task of the sub-iteration to find a step size *α* such that the condition in Eq. (27) is satisfied. Let us first consider the computation of **W**
_{
j
}. The computation of constant matrices \(\textbf {H}_{1}\textbf {H}_{1}^{\mathrm {T}}\), \(\textbf {H}_{2}\textbf {H}_{2}^{\mathrm {T}}\), \(\textbf {W}_{1d} \left (\textbf {H}_{1}\textbf {H}_{1}^{\mathrm {T}}\right)\), \(\textbf {W}_{2d} \left (\textbf {H}_{2}\textbf {H}_{2}^{\mathrm {T}}\right)\), \(\textbf {V}_{1}\textbf {H}_{1}^{\mathrm {T}}\), and \(\textbf {V}_{2}\textbf {H}_{2}^{\mathrm {T}}\) is going to be performed only once per iteration and the computational cost is of the order of \(O\left (\bar {r}^{2}M + N{r^{2}_{d}}+NMr\right)\), where \(\bar {r} = max(r_{d},r_{j})\). Given that *r*
_{
d
}<*r*, *r*
_{
j
}<*r*, and *r* is a small value compared to *N* and *M*, the computational cost is of the order of *O*(*N*
*M*
*r*). Similarly, the computational cost of the fix matrices at each iteration for **W**
_{
d
} and **H** can be computed respectively in \(O \left (\bar {r}^{2}M+\bar (r)^{2}N+NMr \right)\) and \(O\left (\bar {r}^{2}N+{r^{2}_{d}}M+NMr \right)\) operations, which by considering the NMF criterion on r (*r*<<*N*,*M*), are computed to be of the order of *O*(*N*
*M*
*r*).

*g*(

*U*) and vector

*U*, using Taylor series, the following approximation can be considered:

See Appendix for ∇^{2}
*g*(**W**
_{
j
}), ∇^{2}
*g*(**W**
_{1d
}), ∇^{2}
*g*(**W**
_{2d
}), ∇^{2}
*g*(**H**
_{1}), and ∇^{2}
*g*(**H**
_{2}). Using the decreasing condition of Eq. (29) instead of Eq. (27), *g*(*U*
^{(k+1)}) does not need to be computed at every iteration to find a suitable step size. Although ∇^{2}
*g*(*U*
^{(k)}) has to be computed but it will happen only once. Hence, the computational cost at each sub-iteration can be computed in \(O\left (kN{r^{2}_{d}}\right)\), *O*(*k*
*N*
*r*
^{2}), and *O*(*k*
*M*
*r*
^{2}) for **W**
_{
j
}, **W**
_{
d
}, and **H**, respectively, where *k* is the average number of checking the decreasing condition in Eq. (27) at each iteration. Since *r*
_{
d
}<*r*, the overall computational cost to solve Eq. (9) is as follows: *#*
*i*
*t*
*e*
*r*
*a*
*t*
*i*
*o*
*n*
*s*×(*O*(*N*
*M*
*r*)+*#*
*s*
*u*
*b*_*i*
*t*
*e*
*r*
*a*
*t*
*i*
*o*
*n*
*s*×*O*(*k*
*N*
*r*
^{2}+*k*
*M*
*r*
^{2})).

### 3.4 JLNMF algorithms

The optimization algorithm of the proposed JLNMF method is summarized in Algorithms 1 and 2. Regarding the initialization of the algorithm, a common approach for NMF is initialization of the values of **W** and **H** matrices as non-negative random values. Following the same initialization method, all matrices **W**
_{1},**W**
_{2},**W**
_{
j
},**H**
_{1}, and **H**
_{2} are initialized to non-negative random values.

### 3.5 Parameter selection considerations

There are several parameters that need to be selected in the proposed JLNMF algorithm. The values of *σ* and *β* are set to 0.01 and 0.1, respectively, as commonly used in literature [15], and the values of *λ* and *γ* are dependent on the application. A suitable value of *λ* and *γ* can be selected from the range of [0.05 to 0.20] in order to balance off between the reconstruction error (i.e., the first two terms in Eq. 9) and the discrimination power (i.e., the remaining terms). Selection of *r* is also application depended and considerations similar to those for NMF can be applied to adjust *r* [21–23]. The selection of *r*
_{
j
} can be adjusted to increase the discrimination power between classes (i.e., maximize the difference between \(\hat {\textbf {V}}_{1d}\) and \(\hat {\textbf {V}}_{2d}\)). Furthermore, it is worth mentioning that although the proposed JLNMF algorithm has been formulated in the context of two classes, the extension of the method to multiple classes can be simply done. However, since the objective here is to address the challenges in biomedical signal classification applications, the paper has focused on two-class cases (i.e., abnormal vs. normal).

## 4 Experiments with JLNMF

### 4.1 Synthetic data

#### 4.1.1 Visualization of JLNMF

*x*(

*m*) and

*y*(

*m*), by combining a set of non-stationary components in the form of

*α*

*g*(

*μ*,

*σ*) sin(2

*π*(

*f*+

*θ*

*m*)

*m*) where

*g*(

*μ*,

*σ*) is a Gaussian function with mean

*μ*and variance of

*σ*, and the set (

*α*,

*μ*,

*σ*,

*f*,

*θ*) is the parameter for each component (a) to (g). The parameters of each component are manipulated to generate components (a) through (d) (see Fig. 3) as the shared structure by the two signals, components (e) and (f) as the distinct structure of signal

*x*(

*m*), and the chirp component (g) as the distinct structure of signal

*y*(

*m*). In this figure, parameters (

*α*,

*μ*,

*σ*,

*f*,

*θ*) for components (a) to (g) are, respectively, as follows: (1, 1.2, 0.05, 0.325, 0), (1,1.2, 0.05, 0.075, 0), (1, 3.4, 0.04, 0.125, 0), (1, 5.6, 0.03, 0.325, 0), (3, 0.9, 0.001, 0.450, 0), (3, 3.5, 0.001,0.213, 0), and (1, 5.3, 0.05, 0.005, 0.245), and Spectrogram with FFT size of 1024 points and Kaiser window with parameter of 5, length of 256 samples, and 220 samples overlap, is used to construct the TF representations. Each TF data (i.e.,

**V**

_{1}and

**V**

_{2}) was decomposed using standard NMF method with

*r*=6, and the decomposed TF basis and coefficient matrices were obtained and displayed in Fig. 3 c, d. It can be seen that NMF adaptively breaks down the non-stationary signal into TF basis and coefficient matrices representing the spectral and temporal information of the signal, respectively. For example, the first TF basis in

**W**

_{1}represents the spectral characteristics of components (f) and (c) in the TF signal

**V**

_{1}, and the corresponding TF coefficient in

**H**

_{1}represents the temporal locations of those components in the TF signal. The best scenario for this decomposition would be if the decomposed TF bases represent the discriminant structures of each TF signal separate from the shared ones. Then the discriminant TF bases would be powerful in differentiating between the two signals. However, there is no separation between the TF bases that represent the shared structure (i.e., components (a)–(d)) and the distinct structure of each signal; for example, TF basis 1 of

*x*(

*m*), contains the frequency structure of components (e) and (b), and similarly TF basis 2 models the components (a), (b), and (e). A similar behavior can be seen in the TF bases of signal

*y*(

*m*).

*r*=6 and

*r*

_{ j }=3. The values of

*σ*and

*β*were set to 0.01 and 0.1, respectively, as commonly used in literature [15], and a value of 0.05 was selected for

*λ*and

*γ*. As can be seen in Fig. 4 a, the first three (i.e.,

*r*

_{ d }=3) components of the decomposed TF bases

**W**

_{1d }and

**W**

_{2d }, respectively, represent the distinct structure of signals

*x*(

*m*) and

*y*(

*m*), and the last three (

*r*

_{ j }) are zero (see Eq. (7) for the structures of

**W**

_{ d }and

**W**

_{ j }with respect to

**W**). The shared structure is represented in the last three (

*r*

_{ j }=3) of the shared TF bases

**W**

_{ j }. It is interesting to point out that although

*r*

_{ d }was set to be three, the method adaptively identifies that only two distinct TF bases are sufficient to model the discriminate structure in

*x*(

*m*), and finds an empty (all zero) for the one of the TF basis vectors in

**W**

_{2d }. Additionally, unlike standard NMF TF bases, none of the JLNMF TF bases represents both the shared and distinct TF structure of the original signals

*x*(

*m*) and

*y*(

*m*). Comparing the decomposed TF basis and coefficient matrices of the proposed JLNMF and standard NMF (Figs. 3 c, d and 4 a, b), it can be observed that the JLNMF coefficients are noticeably more localized. Furthermore, the discriminant TF structures of signals

*x*(

*m*) and

*y*(

*m*) are successfully reconstructed as

**W**

_{1d }

**H**

_{1}and

**W**

_{ j }

**H**

_{2}, respectively, as shown in Fig. 4 c. The shared structures are also successfully reconstructed as

**W**

_{ j }

**H**

_{1}and

**W**

_{1d }

**H**

_{2}for each signal as shown in Fig. 4 d. This example demonstrates that the proposed JLNMF successfully separates the shared components and the distinct class-specific structures of each TF data. The only way to make this separation happen in the standard NMF is by identifying the TF bases of

**W**

_{1}and

**W**

_{2}that are correlated as the shared TF bases, and the uncorrelated ones as the class-specific TF bases. However, as can be seen in Fig. 3 c, the standard NMF TF bases are a mixture of shared and distinct structures and the TF bases of the two classes are correlated, and thereby, the standard NMF cannot achieve the results shown in Fig. 4.

#### 4.1.2 Properties of JLNMF

It is desirable that the extracted discriminant bases are robust to signal processing operations, which are not expected to affect the classification task. This section examines the properties of JLNMF for amplitude scale and time shift.

##### 4.1.2.1 Amplitude scaling:

If *y*(*m*) is the amplitude scaled version of signal *x*(*m*), such that *y*(*m*)=*a*
*x*(*m*), the TFM of the signal *y*(*m*), **V**
_{2}, can be written as **V**
_{2}=*a*
^{2}
**V**
_{1}, where **V**
_{1} is the TFM data of signal *x*(*m*) and *a* is the amplitude scale. Since the amplitude scaling does not affect the TF structure of the signal, it is expected that JLNMF does not identify any discriminant TF bases. To put it into test, an amplified version of signal *x*(*m*) in Fig. 3
a was generated and JLNMF was applied to the original and scaled TF data, and as expected, both of the discriminant matrices, **W**
_{1d
} and **W**
_{2d
}, were obtained empty, demonstrating that JLNMF was successful in identifying that there were no distinct differences between the two data, and verifying that JLNMF is invariant to amplitude scaling.

##### 4.1.2.2 Time shift:

*x*(

*m*) and

*y*(

*m*)=

*x*(

*m*−

*τ*). The temporal shift only shifts the TF structure of the TF data in time and does not change the TF components of the data (i.e.,

**V**

_{2}(

*n*,

*m*)=

**V**

_{1}(

*n*,

*m*−

*τ*). Hence, JLNMF is not expected to identify any distinct structural differences between

*x*(

*m*) and

*y*(

*m*). A signal

*x*(

*m*) and

*x*(

*m*−

*τ*) with a time shift of

*τ*=1.5 s is generated as shown in Fig. 5 a, and the TF data is displayed in Fig. 5 b.

JLNMF with *r*=9 and *r*
_{
j
}=6 was performed on the TF data, and the decomposed TF bases and coefficients are displayed in Fig. 5
c. The method did not identify any discriminant TF matrices (i.e., **W**
_{1d
} and **W**
_{2d
} are both zero). Hence, \(\hat {\textbf {V}}_{1d}\) and \(\hat {\textbf {V}}_{2d}\) are empty. **W**
_{
j
} contained the spectral vectors that are shared between *x*(*m*) and *x*(*m*−*τ*), and **H**
_{2} was equal to **H**
_{1}(*m*−*τ*). As can be seen, the JLNMF algorithm is invariant to time shifting.

#### 4.1.3 Data retrieval application using JLNMF algorithm

*x*and

*y*could be considered as the distinct structure in each class, and the box represents the shared structures between the two. Hence, each class consists of a class-specific distinct structure and one shared structure. Each signal is defined as the sum of two components as defined below:

*a*

_{0},

*b*

_{0}) belong to a uniform distribution U(0,1),

*a*

_{1}=0.25,

*b*

_{1}=0.40, and

*N*=1,000 is the signal length in samples with a sampling frequency of 1 kHz. Two classes are generated by selecting

*b*

_{2}from one of the following uniform distributions:

where *b*
_{21} and *b*
_{22} are selected using Eqs. (31a) and (31b), respectively. The TF representation for signals in each class is plotted in Fig. 6
c.

*v*

_{18915×1}, and the training data sets are created as: \(\textbf {V}^{(18915\times 300)}_{1} = \left [v_{1}(1) \ v_{1}(2) \ \cdots \ v_{1}(300) \right ]\) and \(\textbf {V}^{(18915\times 300)}_{2} = \left [v_{2}(1) \ v_{2}(2) \ \cdots \ v_{2}(300) \right ]\). (3) JLNMF is applied to the TF data

**V**

_{1}and

**V**

_{2}and class-specific and shared TF bases are decomposed. The parameters

*r*and

*r*

_{ j }are selected 40 and 20, respectively, and

*λ*and

*γ*are varied over {0.05,0.1,0.15,0.2}. (4) for classification purposes, each test data,

*v*

_{test}(

*i*)

_{ i=1:40}, is projected over the JLNMF class-specific and shared TF bases obtained in the previous step. To make this happen, an overall TF bases,

**W**, is constructed in the form of [

*w*

_{1d }(1) ⋯

*w*

_{1d }(

*r*

_{ d })

*w*

_{2d }(1) ⋯

*w*

_{2d }(

*r*

_{ d })

*w*

_{ j }(

*r*−

*r*

_{ j }+1) ⋯

*w*

_{ j }(

*r*)], and coefficient vector \(h^{T}_{\text {test}}(i)_{i=1:40}\) is computed using the updating rule in Eq. (3). The first

*r*

_{ d }=20 elements in \(h^{T}_{\text {test}}(i)\) contain the corresponding coefficients of the class-specific TF bases of class 1, the next

*r*

_{ d }=20 elements contain the corresponding coefficients of the class-specific TF bases of class 2, and the last

*r*

_{ j }=20 elements contain the corresponding coefficients of the linear combinations of the shared TF bases. Hence, the distinct structures of class 1 and class 2 in test data

*i*are reconstructed as follows:

*v*

_{1d }(

*i*), and \(\hat {v}_{2d}(i)\) and the original

*v*

_{2d }(

*i*), are computed. The average of the correlation values, which we denote as class-specific recovery percentage (CRP), is computed as shown in the following equation to assess the method’s success in identifying and recovering the distinct structures of each class in the presence of 100 % overlap (see Fig. 6 c for the test data).

*λ*and

*γ*over {0.05,0.1,0.15,0.2}. The average CRP

_{1}and CRP

_{2}values are shown in Fig. 7.

*λ*and

*γ*with (0.10 and 0.05), being the best average CRP value (97 %) for the given example. Figure 8 displays the reconstructed class-specific structures for two examples with (CRP

_{1}, CRP

_{2}) values of (96 %, 98 %) and (72 %, 99 %) for the example in the top and bottom, respectively. As can be seen in this plot, although there is a relatively low correlation between \(\hat {v}_{1d}(i)\) and

*v*

_{1d }(

*i*), the distinct structure of class 1 has been mostly recovered. To compare JLNMF with NMF, the experiment is repeated using standard NMF, where NMF with

*r*=30 is separately applied to each TF data and the shared TF bases are identified as the TF bases with a correlation value of greater than 0.9. The average CRP value of 88 % is obtained, which is significantly less the CRP value of 97 %, obtained using the proposed JLNMF method.

### 4.2 Real data

#### 4.2.1 Localization of epileptic spikes in EEGs

The application of the proposed JLNMF method to localize the epileptic discharges associated with infantile spasms in hypsarrhythmia (HYPS) is explored. Infantile spasms refer to a catastrophic form of epilepsy occurring in infancy that is diagnosed based on the findings of HYPS in EEG recordings combined with epileptic spasms [26]. HYPS is characterized by a chaotic and high voltage background with multifocal, discharges [27]. However, identifying this pattern of activity in a conventional EEG recording is challenging in the presence of HYPS due to the abundance of epileptiform discharges with varying focality and morphology [28]. An experienced electroencephalographer interprets an EEG by inspecting and approximating the characteristics of HYPS subjectively rather than through objective quantification. Due to complex nature of these signals, even experienced EEG readers tend to interpret HYPS differently, which can have serious implications in the treatment of the infant [28]. Several algorithms have been developed to detect epileptic discharges during epilepsy. Some of those algorithms include temporal analysis based on template matching and mimetic analysis methods, or TF method based on wavelet analysis. However, those methods have been developed for epileptic discharge detection in EEG signals associated with other types of epilepsy and not in the presence HYPS. The existing methods are either based on supervised classifiers in which the presence of true spikes has to be readily available and identifiable to train the algorithm during a learning phase, or they are template based, which rely on pre-specified spike characteristics such as amplitude and duration of the discharges. Given the chaotic appearance of EEG during HYPS, the manual localization of true spikes is not always possible. The spikes of interest are characteristically non-uniform, which presents a challenge for temporal-based methods. Hence, there is a need to develop an semi-supervised feature extraction and classification method to assist in localization of spikes with multiple foci and varying morphologies during HYPS.

A 5-min section of awake EEG recording from an infant with infantile spasms is used to explore the application of JLNMF method for semi-supervised localization of epileptic spikes. The subject consent was obtained through the Infantile Spasms Registry and Genetic Studies via a protocol approved by the University of Rochester’s Research Subjects Review Board. The EEG signals were recorded based on the international standard 10–20 system with sampling frequency of 512 Hz. The recording EEGs were imported to Persyst EEG software (Persyst, San Diego, CA) for artifact reduction and then were imported into MATLAB and bandpass filtered (0.5–30 Hz) for further analysis. All the epilepticform discharges were manually marked by an epileptologist.

The data is divided into 10-s windows and an epileptologist labels each 10-s EEG window as non-spike (NS) if there are no epiletiform spikes in that window; otherwise, the window is marked as possibly-spike (PS). The objective is to characterize the structure of the epileptiform spikes and localize them during each 10-s window. The localization task is formulated as a classification task, where the objective is to decompose the class-specific (i.e., epileptic spikes) and shared bases (i.e., the common EEG baseline), which can be used to reconstruct the class-specific data in each class. The distinct structure of the PS recordings is expected to indicate the spike locations during the EEG recordings. There are two main challenges: the first challenge is that the EEG recordings are strongly non-stationary, which can be addressed by using the TF data of the EEG recordings to improve the representation of the non-stationary information. The second challenge is the substantial amount of similarity (i.e., overlap) between the two NS and PS classes. This is mainly because the exact locations of the epileptiform spikes are not specified, and the only available information is that the PS class contains several spikes at some unknown locations. The proposed JLNMF method is used to address this challenge, as it is able to decompose the TF data to the spike-specific TF bases from the common EEG baseline in a semi-supervised fashion. The details of the application of JLNMF on the EEG recording is as follows:

*v*

_{16,000×1}. The training data set is created by collecting the collapsed TF vectors of half of the NS and PS data. (3) JLNMF is applied to the TF data and class-specific and shared TF bases are decomposed. The parameters

*r*and

*r*

_{ j }are selected as 40 and 20, respectively, and the values of 0.10 and 0.05 are, respectively, selected for

*λ*and

*γ*. Figure 9 a shows the shared (on top) and PS-specific (on bottom) TF bases. To visualize the TF bases, each decomposed base is restructured back to the original size. The NS-specific TF matrix is found empty, which means that JLNMF does not identify any distinct structure to the NS class. (4) For classification purposes, each test data is projected over the JLNMF PS-specific and shared TF bases obtained in the previous step, and the coefficient vector is computed using the updating rule in Eq. (3), and the distinct structures corresponding to the PS-specific TF bases are reconstructed for all the test data.

*r*=40. The TF bases of each class is shown in Fig. 10. The NS and PS TF bases are compared to separate the spike-specific TF bases from the shared ones, but none had a correlation value of greater than 0.9 and only three had a value of greater than 0.8. Hence, the standard NMF is unable to locate any spike-related TF bases without any further analysis, while JLNMF showed to be successful in identifying the epileptic spikes. Such a method is deemed to be necessary for reliable evaluation of features associated with HYPS, which could potentially improve the assessment of infantile spasms, which is of significant importance in the therapy, management and ultimately the success of the prescribed treatment.

#### 4.2.2 Discrimination of pathological voice disorder

## 5 Conclusions

In most real-world applications, the nature of signals from different classes are very similar, and there could only be slight changes in the TF patterns of signals from different classes. Inspired by this observation, a set of shared TF bases along with class-specific TF bases, was considered in the proposed JLNMF algorithm, and a new cost function was introduced to enforce discrimination between shared and class-specific TF bases. A projected gradients framework was performed to optimization the JLNMF cost function. It was shown that the proposed JLNMF is invariant to amplitude scaling and time shifting, which is required for signal recognition applications. The performance of the proposed JLNMF was evaluated and compared to the results of standard NMF for data retrieval and localization applications, verifying the effectiveness of JLNMF.

## 6 Appendix

## Declarations

### Acknowledgements

The author would like to thank Drs. L.E. Seltzer and A.R. Paciorkowski from the Department of Neurology from University of Rochester Medical Center for providing the EEG data and their insight and assessment of the epileptic spikes in the EEG signals.

### Competing interests

The author declares that he has no competing interests.

### Ethics approval and consent to participate

The Ethical approval was given by the Institutional Review Board at University of Rochester Medical Center (Rochester, NY) under the reference number of 43225.

**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.

## Authors’ Affiliations

## References

- L Cohen, Time-frequency distributions—a review. Proc. IEEE.
**77:**, 941–981 (1989).View ArticleGoogle Scholar - SG Mallat, Z Zhifeng, Matching Pursuits with Time-Frequency Dictionaries. IEEE Trans. Signal Process.
**41**(12), 3397–3415 (1993).View ArticleMATHGoogle Scholar - B Boashash, G Azemi, NA Khan, Principles of time–frequency feature extraction for change detection in non-stationary signals: Applications to newborn EEG abnormality detection. Pattern Recognit.
**48**(3), 616–627 (2015).View ArticleGoogle Scholar - O Yilmaz, S Rickard, Blind separation of speech mixtures via time-frequency masking. IEEE Trans. Signal Process.
**52**(7), 1830–1847 (2004).MathSciNetView ArticleGoogle Scholar - R Hennequin, B David, R Badeau, in Proc. ICASSP. Score informed audio source separation using a parametric model of non-negative spectrogram (Prague, Czech Republic, 2011), pp. 45–48.Google Scholar
- T Virtanen, Monaural sound source separation by nonnegative matrix factorization with temporal continuity and sparseness criteria. IEEE Trans. Audio, Speech Lang. Process.
**15**(3), 1066–1074 (2007).View ArticleGoogle Scholar - C Févotte, N Bertin, J-L Durrieu, Nonnegative matrix factorization with the itakura-saito divergence: With application to music analysis. Neural Comput.
**21**(3), 793–830 (2009).View ArticleMATHGoogle Scholar - B Ghoraani, S Krishnan, Time-frequency feature extraction and classification of environmental audio signals. IEEE Trans. Audio, Speech Lang. Process.
**19**(7), 2197–2209 (2011).View ArticleGoogle Scholar - B Ghoraani, S Krishnan, A joint time-frequency and matrix decomposition feature extraction methodology for pathological voice classification. EURASIP J. Adv. Signal Process.
**2009**(ID 928974), 11–1011552009928974 (2009).MATHGoogle Scholar - S Zafeiriou, A Tefas, I Buciu, I Pitas, Exploiting discriminant information in nonnegative matrix factorization with application to frontal face verification. IEEE Trans. Neural Netw.
**17**(3), 683–695 (2006).View ArticleGoogle Scholar - S Nikitidis, A Tefas, I Pitas, Projected gradients for subclass discriminant nonnegative subspace learning. IEEE Trans. Cybernet.
**44**(12), 2806–2819 (2014).View ArticleGoogle Scholar - DD Lee, HS Seung, Algorithms for non-negative matrix factorization (Adv Neural Inf Proces Syst, 2001).Google Scholar
- MW Berry, M Browne, AN Langville, VP Pauca, RJ Plemmons, Algorithms and applications for approximate nonnegative matrix factorization. Comput. Stat. Data Anal.
**52**(1), 155–173 (2007).MathSciNetView ArticleMATHGoogle Scholar - I Buciu, Non-negative matrix factorization, a new tool for feature extraction: Theory and applications. Proc. Int. J. Comput. Commun. Control.
**3:**, 67–74 (2008).Google Scholar - C-J Lin, Projected gradient methods for nonnegative matrix factorization. Neural Comput.
**19**(10), 2756–2779 (2007).MathSciNetView ArticleMATHGoogle Scholar - N Guan, X Zhang, Z Luo, D Tao, X Yang, Discriminant projective non-negative matrix factorization. PloS one.
**8**(12), 8329 (2013).Google Scholar - J Yoo, M Kim, K Kang, S Choi, Nonnegative matrix partial co-factorization for drum source separation, (Dallas, Texas, USA, 2010).Google Scholar
- P Paatero, The multilinear enginea table-driven, least squares program for solving multilinear problems, including the n-way parallel factor analysis model. J. Comput. Graph. Stat.
**8:**, 854–888 (1999).MathSciNetGoogle Scholar - M Chu, F Diele, R Plemmons, S Ragni, Optimality, computation, and interpretation of nonnegative matrix factorizations. SIAM J Matrix Anal, 4–8030 (2004). Citeseer.Google Scholar
- DP Bertsekas, On the goldstein-levitin-polyak gradient projection method. IEEE Trans. Automatic Control.
**21:**, 174–184 (1976).MathSciNetView ArticleMATHGoogle Scholar - AT Cemgil, Bayesian inference for nonnegative matrix factorisation models. Computational Intelligence and Neuroscience.
**2009:**, 17 (2009). article ID 785152.View ArticleGoogle Scholar - M Morup, LK Hansen, in Proc. 17th European Signal Processing Conference (EUSIPCOŠ09). Tuning pruning in sparse non-negative matrix factorization (Glasgow, Scotland, 2009), pp. 1923–1927.Google Scholar
- VY Tan, C Févotte, Automatic relevance determination in nonnegative matrix factorization with the/spl beta/-divergence. IEEE Trans. Pattern Anal. Mach. Intell.
**35**(7), 1592–1605 (2013).View ArticleGoogle Scholar - M Davy, C Doncarli, GF Boudreaux-Bartels, Improved optimization of time-frequency-based signal classifiers. Signal Process. Lett. IEEE.
**8**(2), 52–57 (2001).View ArticleGoogle Scholar - B Ghoraani, S Krishnan, Discriminant non-stationary signal features’ clustering using hard and fuzzy cluster labeling. EURASIP J. Adv. Signal Process.
**2012**(1), 1–20 (2012).View ArticleGoogle Scholar - JM Pellock, R Hrachovy, et al, Infantile spasms: a us consensus report. Epilepsia.
**51**(10), 2175–2189 (2010).View ArticleGoogle Scholar - RA Hrachovy, JD Frost, P Kellaway, Hypsarrhythmia variations on the theme. Epilepsia.
**25**(3), 317–325 (1984).View ArticleGoogle Scholar - SA Hussain, G Kwong, SA Hussain, G Kwong, JJ Millichap, JR Mytinger, N Ryan, JH Matsumoto, JY Wu, JT Lerner, R Sankar, Hypsarrhythmia assessment exhibits poor interrater reliability: a threat to clinical trial validity. Epilepsia.
**56**(1), 77–81 (2014).View ArticleGoogle Scholar - K Lee, HF Schuknecht, Results of tympanoplasty and mastoidectomy at the Massachusetts Eye and Ear Infirmary. Laryngoscope.
**81:**, 529–43 (1971).View ArticleGoogle Scholar