Title: MULTI-LABEL LEARNING WITH SEMANTIC EMBED-DINGS

Abstract: Multi-label learning aims to automatically assign to an instance (e.g., an image or a document) the most relevant subset of labels from a large set of possible labels. The main challenge is to maintain accurate predictions while scaling efficiently on data sets with extremely large label sets and many training data points. We propose a simple but effective neural net approach, the Semantic Embedding Model (SEM), that models the labels for an instance as draws from a multinomial distribution parametrized by nonlinear functions of the instance features. A Gauss-Siedel mini-batch adaptive gradient descent algorithm is used to fit the model. To handle extremely large label sets, we propose and experimentally validate the efficacy of fitting randomly chosen marginal label distributions. Experimental results on eight real-world data sets show that SEM garners significant performance gains over existing methods. In particular, we compare SEM to four recent state-ofthe-art algorithms (NNML, BMLPL, REmbed, and SLEEC) and find that SEM uniformly outperforms these algorithms in several widely used evaluation metrics while requiring significantly less training time.

1. INTRODUCTION:

The multi-label learning problem is to learn to predict potentially multiple relevant labels given an instance. Instances that have multiple labels naturally occur in many application domains, including multimedia information retrieval, tag recommendation, semantic scene classification, query categorization, gene function prediction, medical diagnosis, drug discovery, and marketing.

A popular approach to the multi-label learning problem is to embed the labels in a low-dimensional latent space via linear or local non-linear embeddings. The approach of Hsu et al. (2009) projects the label vectors to a random low-dimensional space, fits a regression model in this space, then projects these predictions back to the original label space. Balasubramanian & Lebanon (2012) use a sparsity-regularized least squares reconstruction objective to select a small set of landmark labels that are used to predict the remaining labels. Bi & Kwok (2013) take a similar approach, with a greatly decreased computation cost, by posing the problem of selecting the landmark labels as one of column subset selection and adopting the leverage score sampling approach (Boutsidis et al., 2009) . Recently, Yu et al. (2014) and Jing et al. (2015) propose using trace norm regularization to identify a low-dimensional representation of the original large label space. Mineiro & Karampatziakis (2015) use randomized dimensionality reduction to learn a low-dimensional embedding that explicitly captures correlations between the instance features and their labels. These approaches, like other linear embedding methods, assume that the label matrix is low-rank. However, the label matrix in most applications of multi-label learning is a sparse binary matrix, and thus is extremely likely to violate this low-rank assumption (Bhatia et al., 2015) .

Rather than working with the original label and feature matrices, some methods work instead with label or feature similarity matrices, and seek to preserve the local structure of the data in the learned low-dimensional latent space. Tai & Lin (2010) use PCA on the label covariance matrix to extract a low-dimensional latent space for labels and Chen & Lin (2012) extend this method to integrate feature information. Lin et al. (2014) apply PCA to a similarity matrix constructed using both label and feature information; this approach is time-consuming as it requires computing a large similarity matrix. Nam et al. (2014) introduce a neural network model to capture non-linear relationships between the input features and the labels. However, this approach is computationally infeasible when the number of possible labels is large. Similarly, Cissé et al. (2016) shows that using a deep learning approach built on top of an informative partitioning of the label space gives good performance; the scalability of this method was not characterized. Prabhu & Varma (2014) propose a method to efficiently train a classification tree by minimizing the Normalized Discounted Cumulative Gain. Rai et al. (2015) assumes that the label vectors are generated by sampling from a weighted combination of label topics, where the mixture coefficients are determined by the instance features. Bhatia et al. (2015) proposes a multi-phase algorithm (SLEEC) that first clusters the instances into a number of relatively small groups, learns label embeddings for each group via an SVD, and then trains linear regressors from the input features to the latent label factors for each group. SLEEC empirically outperforms previous state-of-the-art multi-label classifiers, but the label embedding in each group is learned from a nearest neighbor graph that is constructed solely from labelling information, ignoring the available feature matrix; the feature matrix has been shown repeatedly to be a source of useful information for label embedding (Chen & Lin, 2012; Lin et al., 2014; Yu et al., 2014; Jing et al., 2015) .

The contribution of this paper is a scalable, accurate, and simple neural network approach to multilabel learning. Experiments establish that our method is faster and more accurate than SLEEC, the current state-of-the-art scalable algorithm.

Notation: In the sequel, n is the number of training instances, c is the cardinality of the set of possible labels, d is the dimensionality of the feature vectors, and r is the dimension of the learned latent space. The matrix X ∈ R n×d contains the instance features, and Y ∈ 0, 1 n×c indicates the labels assigned to each instance. We denote the number of observed labels for instance i with i = c k=1 y ik . The notations A i• and A •j respectively refer to the ith row and jth column of the matrix A. Unless otherwise specified, the notation f (A) denotes the elementwise application of an arbitrary function f to the A, so for example exp(A) ij = exp(a ij ).

2. THE SEMANTIC EMBEDDING MODEL:

Our Semantic Embedding Model (SEM) assumes that the underlying parameters determining the observed labels are low-rank rather than that the observed label matrix is itself low-rank, and it uses a nonlinear model to fit the probability distributions over the labels, conditioned on the instance features.

SEM models the i-th row of Y as the result of i draws from a multinomial distribution:

Y i• ∼ Multinomial( i ; P i ), where P = exp(h ij ) c k=1 exp(h ik ) i=1,...,n j=1,...,c

.

(1)

The parameter matrix H = UV T + 1 n b T is the sum of label priors b ∈ R c and the product of explanatory latent factors associated with the instances (U ∈ R n×r ) and the labels (V ∈ R c×r ). Further, we allow the latent factors associated with each instance to be a nonlinear function of the features associated with that instance, U = f (X, W) for some W to be learned. We note that if f (X, W) = XW, SEM could be viewed as fitting a Bayesian Exponential Family PCA (Mohamed et al., 2009) . However, throughout this paper we take f (XW) = σ(XW), where σ(X) = (1 + exp(-X)) -1 denotes the elementwise application of the sigmoid function, as we find this gives good results; with this choice, SEM is more naturally viewed as a neural network model.

We fit the SEM parameters by maximizing the likelihood of the observed labels. This is equivalent to minimizing the sum of the KL divergences between the empirical label distributions for each instance and the label distributions predicted by the model (Pawitan, 2001) . Accordingly, we define the empirical label distribution matrix G, whose ith row satisfies G i = Y i / i , then minimize the row-wise Kullback-Leibler distance (Yang et al., 2011) between G and P:

EQUATION

Recalling that

P ij = exp(h ij ) c k=1 exp(h ik ) = exp (σ(XW)V T ) ij + b j c k=1 exp ((σ(XW)V T ) ik + b k )

, some algebraic manipulations give the final objective

EQUATION

Thus the SEM parameters are learned by solving the optimization problem min

EQUATION

Here V ∈ R c×r are the representations of the labels in a latent semantic space, W ∈ R d×r controls the nonlinear mapping from the instance features to the same semantic space, and the offsets b ∈ R c allow for label-specific offsets in the mapping from the semantic space to the log probabilities.

3. MODEL FITTING:

The optimization problem (4) is non-convex. To solve it efficiently, we use a Gauss-Siedel approach combined with mini-batching.

Namely, we cyclically update each of W, V, and b using AdaGrad (Duchi et al., 2011) while keeping the other two variable fixed. We compute the gradients using mini-batches. To state the expressions for the gradients with respect to the model parameters, we introduce some helpful notation: A B denotes the entry-wise product of two matrices, M = σ(XW) (1 -σ(XW)),

and D = Diag exp (σ(XW)V T + 1 n b T )1 c .

The gradients are readily computed from (2):

EQUATION

Using AdaGrad, the update rule for W (τ ) is

EQUATION

)

where τ is the timestep and α W is a matrix of step sizes computed via

EQUATION

where ε and the learning rate ρ determine how much an entry W ij is updated during the first timestep.

V (τ ) and b (τ ) are computed according to similar updating rules obtained from ( 8) and ( 9) by substituting G(W) with G(V) (or G(b)), W with V (or b), and α W with α V (or α b ).

A listing of the proposed algorithm is given in Algorithm 1. Its computational complexity is O(T nr(d + c)), where T is the number of epochs. We note that the gradient calculations in lines 7-9 of Algorithm 1 are amenable to parallelization.

Algorithm 1 Mini-Batched Gauss-Siedel Adaptive Gradient Descent for learning SEM parameters Input: Instance feature matrix X ∈ R n×d , observed label matrix Y ∈ R n×c , dimensionality of the latent space r, learning rate ρ and > 0, mini-batch size m < n, and number of epochs T . 1: Initialize W (0) , V (0) , and b (0) 2: for t = 1, 2, . . . , T do Select the data instances in the z-th mini-batch by working with X Iz,• in lieu of X 7:

Update W (τ ) via (8) while fixing V = V (τ -1) and b = b (τ -1)

8::

Update V (τ ) via the analog of (8) for V while fixing W = W (τ ) and b = b (τ -1) 9:

Update b (τ ) via the analog of ( 8) for b while fixing W = W (τ ) and V = V (τ ) 10:

end for 11: end for Output: W (τ ) , V (τ ) and b (τ )

3.1. INCREASED EFFICIENCY BY FITTING MARGINALS:

Although Algorithm 1 runs in time linear in the dimensions of the model parameters and the input datasets, it can be computationally expensive when there are more than a few thousand labels. To further reduce the running time of our algorithm, we note that in practice, each instance is often associated with i c labels.

To speed up the training, at each timestep τ , rather than attempting to minimize the divergence between the entire empirical and predicted label distributions of each instance, we sample a set of labels L (τ ) i

for each instance and attempt to minimize the empirical and predicted marginal label distributions over that set of labels L (τ ) i . Let PL i denote the set of labels assigned to the ith instance, and AL i denote the set of labels not assigned to that instance. We sample from AL i to form a set NL i , and use L

(τ ) i = PL i ∪ NL (τ )

i . This leads to the modified objective

EQUATION

)

Note that J (τ ) is a random function that changes at each timestep. Minimizing this stochastic objective effectively seeks SEM parameters which fit all the randomly sampled marginals encountered during training. Thus it is important to sample the sets NL i so that the selected marginals capture non-trivial information about the label distributions. One can imagine that uniformly sampling from AL i will not provide very informative marginals. As an improvement on this naïve scheme, we sample labels from AL i with probability proportional to their frequency of occurrence in the training data set. The number of negative labels is set to be β times the number of positive labels i.e., |NL i | = β|PL i | = β i . Further, when m > 1, to faciliate efficient BLAS operations while mini-batching, we use the same marginals for each instance in the same minibatch, i.e., we fit marginals over L (τ ) := i∈Iz L (τ )

i , where I z denotes the set of instances in the current minibatch. In the experiments presented in Section 4, we found that β around 10 suffices when c is relatively small, and β around 100 suffices when c is on the order of tens of thousands.

3.2. LABEL PREDICTION:

We present two methods for predicting the labels for a new instance x ∈ R d given the fitted SEM parameters.

The first uses the generative model behind SEM: form h = σ(x T W)V T +b T and note the probability that the jth label is assigned to that instance is given by

EQUATION

)

Accordingly we assign the most probable labels to x. We call this prediction scheme the direct SEM method; it simply requires choosing the labels corresponding to the largest entries of h.

The second method builds a kernel classifier in the semantic space obtained from the SEM factorization. Following Mineiro & Karampatziakis (2015) , a classifier is trained on these semantic representations by solving the optimization problem

EQUATION

where is the log-loss penalty and ψ : R r → R s is an s-dimensional Random Fourier Feature (RFF) map (Rahimi & Recht, 2007) :

EQUATION

where Φ ∈ R s×r is a matrix of i.i.d. standard Gaussians and θ ∈ [0, 2π) s is a vector of i.i.d uniform samples from [0, 2π).

At test time, the predicted label probabilities for an instance x are given by Zψ(xW), so we assign the most probable labels according to this model. We refer to this scheme as the kernelized SEM method.

4. EXPERIMENTS:

In the sequel we refer to the direct SEM scheme as simply SEM, and the kernelized SEM scheme as SEM-K. We compare SEM and SEM-K with several alternative multi-label learning algorithms: NNML (Nam et al., 2014) , REmbed (Mineiro & Karampatziakis, 2015) , SLEEC (Bhatia et al., 2015) , and BMLPL (Rai et al., 2015) . We do not compare to the models proposed in (Tai & Lin, 2010; Chen & Lin, 2012; Bi & Kwok, 2013; Yu et al., 2014; Prabhu & Varma, 2014) because earlier works (Yu et al., 2014; Bhatia et al., 2015) have shown that they are inferior to SLEEC.

4.1. DATASETS:

Table 1 summarizes the eight datasets used in our experiments. Here n train and n test are the numbers of training and testing instances, d is the number of features, c is the number of labels/classes, and the avg( i ) column reports the average number of labels per instance. In these datasets, the number of labels varies from 23 to 30938, the average label cardinality varies from 2.508 to 19.020, and the number of instances in different classes varies over a large range. Thus predicting the labels assignments correctly over this collection of datasets is a challenging task. 

4.2. METHODOLOGY:

The codes of the methods we compare to are provided by the authors, in particular, we note that the computationally intensive portions of REmbed, SLEEC and NNML are implemented in C; by way of comparison, our algorithms are entirely implemented in Matlab. Due to there being several parameters for each method, we hand-tuned the parameters for each dataset as suggested by the authors. All methods were run in MATLAB on a Windows server with 4GB memory and four 2.3GHz CPUs with eight cores.

The prediction performance for each algorithm is evaluated according to widely-used metrics in the field of multi-label classification, viz., label-based Macro-F1 (MaF1) and Micro-F1 (MiF1) and instance-based Precision-at-k (P@k, esp. P@1 and P@3) (Zhang & Zhou, 2014) . MaF1 and MiF1 require predefining a threshold to determine the number of labels to be assigned to the testing data.

In our experiments, the number of labels assigned to each testing instance was set according to its ground truth. First we compare the performance on six multi-label learning problems with c < 1000. To fit both SEM models, we take the number of epochs be 30 and the mini-batch size be 200-i.e., T = 30 and m = 200 in Algorithm 1-and because c is small, we fit the full label distributions. The classification performances of our SEM algorithms and the baseline methods are shown in Table 2 . SEM or SEM-K outperform the alternative algorithms in most cases.

Table 3 compares the running times of the algorithms as the size of the dataset is increased, using MediaMill. We see that SEM is the fastest model, followed by REMBED, then closely by SEM-K; the remaining three models are significantly more costly. It is clear that NNML, the previous neural network approach to multi-label learning costs the most. In the other five algorithms, the latent space dimensionality (r) is set to be 50. SLEEC is expensive because it constructs the nearest neighbor graph among training data and computes the top r eigenvectors of the corresponding similarity matrix, which costs O(n 2 r + d 2 r). REmbed is efficient because its main cost is to find the singular vectors of a c × (r + q) matrix (here c is the number of labels and q is a small integer), but its performance is inferior to SEM-K. The BMLPL code provided by the author applies SVD to the training data to initialize than model parameters and then uses conjugate gradient to update the parameters, thus it costs much more than REmbed and our proposed methods.

4.4. PERFORMANCE ON DATASETS WITH LARGE LABEL SETS:

We proposed using SEM to fit marginals rather than the entire label distribution when c is large, for computational efficiency. To judge the effectiveness of this proposal, we compare the accuracy and running times of the SEM and SEM-K models with baselines on EurLex-des and Wiki10K, two datasets with c > 1000. As baselines, we use REmbed and SLEEC in accordance with the above discussion which showed that these two methods are efficient and/or have good performance.

The hyperparameters in SLEEC were set according to the original authors' code: r for EurLex-des and Wiki10K is 100 and 75 respectively, and 3 clusters are used for Eurlex-des and 5 are used for Wiki10K. To fit the SEM models, we used the same value of r as SLEEC on these two datasets and used 10 training epochs. For REmbed, the latent space size r was tuned via cross-validation; r = 300 for Eurlex-des and r = 150 for Wiki10K. The number of Random Fourier Features is 2000 for both REmbed and SEM-K. The latent space size r in SEM is same with SLEEC. The mini-batch sizes and number of epochs are set to be 200 and 10 respectively when fitting the SEM models. The number of threads is set to be 8 for all methods. Table 4 compares the classification performances of the methods on these two datasets. It is clear that SEM-K with a small set of negative labels obtains better performance than both REmbed and SLEEC.

Table 5 shows that, additionally, the SEM-K models are fit much faster than than the other models.

4.5. IMPACT OF THE SIZE OF THE MARGINALS:

Figure 1 illustrates the impact of the choice of β on the prediction performance (in terms of P@1) of SEM and SEM-K. The performances of SLEEC and REmbed are included for comparison. The hyperparameters of SLEEC, REmbed and SEM were set as in Section 4.4.

It is evident that the performance of SEM increases significantly in a monotonic fashion with β. However, SEM-K is insensitive to β once it passes a dataset-dependent threshold (e.g., β = 60 for Eurlex-des and β = 100 for Wiki10K). Note that on Wiki10K, even the simpler direct SEM outperforms REmbed when there are sufficient negative labels.

Figure 2 illustrates the effect of β on the running times of SEM and SEM-K. Note that the additional time to fit the classifier in the semantic space required by SEM-K is negligible compared to the time it takes to first fit the direct SEM model. 

4.6. ADDITIONAL CONSIDERATIONS:

There are other important ways in which the proposed SEM methods can be compared to the baseline multi-label learning methods, including their performance as a function of the latent space dimensionality and as a function of the amount of training. Due to space constraints, a discussion of these two concerns and the convergence behavior of Algorithm 1 is provided in the Supplementary material.

5. CONCLUSION:

We proposed a new semantic embedding model (SEM) for handling the multi-label learning task. A framework based on Gauss-Siedel mini-batched adaptive gradient descent was proposed for efficiently solving the non-convex optimization problem required to learn the SEM parameters. For large label sets, we proposed fitting the SEM to marginal distributions rather than the full label distribution. A series of experiments on eight real-world datasets empirically demonstrated that the proposed method is superior to state-of-the-art methods in terms of prediction performance and running time.

sampled ten times, and the resulting average P@1 and MiF1 on the testing data are recorded. During training, the latent dimensionality parameter r is selected via 5-fold cross-validation.

Figure 4 shows these results for the Mediamill dataset which contains the largest number of instances.

As expected, the performance of all the methods is positive correlated with the size of the training data set, and we also see that the proposed SEM-K uniformly outperforms the other methods regardless of the training data size. As it is often expensive to obtain large labeled data sets in real applications, this observation suggests that SEM-K is a better choice for these situations.

C. Convergence:

In order to demonstrated the convergence of the proposed method, we show the value of objective function (4) (at r = 45) via Figure 5 (a) and the prediction result (P@1) via Figure 5 (b) along with the number of passes to the dataset (i.e., τ in Algorithm 1). It can be seen that SEM could be convergent and the prediction performance becomes stable in less than 50 epochs, which will leverage SEM dealing with large-scale data. 

:



Figures and Tables:

n/m mini-batchs I z ⊂ {1, ..., n} of size m 4: for b = 1, 2, . . . n/m do 5: Set τ = (t -1)(n/m) + b 6:

Figure 1: The P@1 performance of SEM and SEM-K as a function of β, in comparison to the performances of SLEEC and REmbed on the (a) Eurlex-des and (b)Wiki10K datasets.

Figure 2: Running time of SEM-K under varying β.

Figure 5: Performance of the proposed SEM method (with r = 45, ρ = 0.1) on the Delicious dataset, a) objective function value to minimum and b) prediction result in terms of P@1, where x-axis represents the number of passes to the dataset.

Multi-label dataset summary.

Table content: <table><tr><td>Dataset</td><td>Domain</td><td>ntrain</td><td>ntest</td><td>d</td><td>c</td><td>avg( i)</td></tr><tr><td>MSRC</td><td>image</td><td>296</td><td>295</td><td>512</td><td>23</td><td>2.508</td></tr><tr><td>Corel5K</td><td>image</td><td>4500</td><td>500</td><td>499</td><td>374</td><td>3.522</td></tr><tr><td>SUN</td><td>image</td><td>12906</td><td>1434</td><td>512</td><td>102</td><td>15.526</td></tr><tr><td>Delicious</td><td>text</td><td>12920</td><td>3185</td><td>500</td><td>983</td><td>19.020</td></tr><tr><td>EurLex-sub</td><td>text</td><td>17413</td><td>1935</td><td>5000</td><td>201</td><td>2.213</td></tr><tr><td>Mediamill</td><td>video</td><td colspan="2">30993 12914</td><td>210</td><td>101</td><td>4.736</td></tr><tr><td>Eurlex-des</td><td>text</td><td>17413</td><td>1935</td><td>5000</td><td>3993</td><td>5.31</td></tr><tr><td>Wiki10K</td><td>text</td><td>14146</td><td>6616</td><td colspan="2">101938 30938</td><td>18.64</td></tr></table>

The classification performance of six multi-label classification algorithms (NNML, BMLPL, REmbed, SLEEC and the proposed SEM and SEM-K). The best and second best results are respectively bolded and underlined for each evaluation measure.

Table content: <table><tr><td/><td>MaF1</td><td>MiF1</td><td>P@1</td><td>P@3</td><td>MaF1</td><td>MiF1</td><td>P@1</td><td>P@3</td></tr><tr><td/><td/><td colspan="2">MSRC</td><td/><td/><td colspan="2">Corel5K</td></tr><tr><td>NNML</td><td colspan="4">0.4086 0.5944 0.7356 0.5073</td><td colspan="4">0.0547 0.2967 0.4020 0.3047</td></tr><tr><td colspan="5">BMLPL 0.4592 0.6199 0.7017 0.5288</td><td colspan="4">0.0315 0.2779 0.3940 0.2820</td></tr><tr><td colspan="5">REmbed 0.3537 0.5128 0.5322 0.4384</td><td colspan="4">0.0450 0.2144 0.3060 0.2247</td></tr><tr><td>SLEEC</td><td colspan="4">0.4973 0.6314 0.7353 0.5243</td><td colspan="4">0.0534 0.3188 0.4360 0.3287</td></tr><tr><td>SEM</td><td colspan="4">0.5064 0.6173 0.7220 0.5333</td><td colspan="4">0.0623 0.3188 0.4320 0.3293</td></tr><tr><td>SEM-K</td><td colspan="4">0.5770 0.6492 0.7458 0.5525</td><td colspan="4">0.0589 0.2649 0.3600 0.2773</td></tr><tr><td/><td/><td>SUN</td><td/><td/><td/><td colspan="2">Mediamill</td></tr><tr><td>NNML</td><td colspan="4">0.2807 0.5248 0.9421 0.8580</td><td colspan="4">0.0819 0.5890 0.8260 0.6675</td></tr><tr><td colspan="5">BMLPL 0.1897 0.4766 0.9024 0.8001</td><td colspan="4">0.0855 0.6012 0.8478 0.6854</td></tr><tr><td colspan="5">REmbed 0.3408 0.5125 0.9393 0.8591</td><td colspan="4">0.2634 0.6371 0.8741 0.6988</td></tr><tr><td>SLEEC</td><td colspan="4">0.2935 0.5256 0.9484 0.8656</td><td colspan="4">0.2851 0.6546 0.8899 0.7158</td></tr><tr><td>SEM</td><td colspan="4">0.3648 0.5486 0.9365 0.8642</td><td colspan="4">0.1593 0.6296 0.8746 0.6996</td></tr><tr><td>SEM-K</td><td colspan="4">0.3703 0.5466 0.9575 0.8787</td><td colspan="4">0.2570 0.6717 0.8953 0.7278</td></tr><tr><td/><td/><td colspan="2">Delicious</td><td/><td/><td colspan="2">Eurlex-sub</td></tr><tr><td>NNML</td><td colspan="4">0.1721 0.3963 0.6687 0.6169</td><td colspan="4">0.5761 0.8487 0.9173 0.6267</td></tr><tr><td colspan="5">BMLPL 0.1061 0.3739 0.6378 0.5772</td><td colspan="4">0.1459 0.6011 0.6789 0.4697</td></tr><tr><td colspan="4">REmbed 0.1549 0.3713 0.6353</td><td>0.572</td><td colspan="4">0.5335 0.8031 0.8785 0.5977</td></tr><tr><td>SLEEC</td><td colspan="4">0.1257 0.3859 0.6674 0.6112</td><td colspan="4">0.5433 0.8461 0.9152 0.6191</td></tr><tr><td>SEM</td><td colspan="4">0.1941 0.3980 0.6727 0.6162</td><td colspan="4">0.5652 0.8339 0.8971 0.6188</td></tr><tr><td>SEM-K</td><td colspan="4">0.1675 0.3886 0.6658 0.6112</td><td colspan="4">0.5807 0.8494 0.9188 0.6269</td></tr></table>

The running times, in seconds, of six multi-label classification algorithms (NNML, BMLPL, REmbed, SLEEC and the proposed SEM and SEM-K) for differing training sizes on the Mediamill dataset.

Table content: <table><tr><td>ntrain</td><td colspan="4">NNML BMLPL REMBED SLEEC</td><td colspan="2">SEM SEM-K</td></tr><tr><td>439</td><td>327.57</td><td>10.29</td><td>2.07</td><td>16.11</td><td>0.60</td><td>1.50</td></tr><tr><td>1756</td><td>1333.91</td><td>20.35</td><td>3.02</td><td>57.16</td><td>2.41</td><td>4.29</td></tr><tr><td>3073</td><td>2363.02</td><td>48.2</td><td>4.14</td><td>145.88</td><td>4.36</td><td>6.99</td></tr><tr><td>4391</td><td>3264.79</td><td>41.72</td><td>5.45</td><td>227.76</td><td>6.65</td><td>10.10</td></tr><tr><td>8781</td><td>4428.09</td><td>84.09</td><td>10.83</td><td>815.66</td><td>12.29</td><td>21.73</td></tr><tr><td colspan="2">13172 5170.00</td><td>119.09</td><td>17.04</td><td colspan="2">1041.07 18.39</td><td>26.49</td></tr><tr><td colspan="2">17563 5170.17</td><td>185.05</td><td>20.90</td><td>1692.7</td><td>24.22</td><td>42.21</td></tr><tr><td colspan="2">21954 5297.75</td><td>225.96</td><td>44.20</td><td colspan="2">1772.52 30.10</td><td>50.64</td></tr><tr><td colspan="2">26344 5947.94</td><td>235.93</td><td>52.75</td><td colspan="2">1985.82 35.95</td><td>59.42</td></tr><tr><td colspan="2">30735 6604.93</td><td>275.06</td><td>58.74</td><td colspan="2">2181.48 41.37</td><td>61.30</td></tr><tr><td colspan="5">4.3 PERFORMANCE ON DATASETS WITH SMALL LABEL SETS</td><td/><td/></tr></table>

The classification performance of five methods (REmbed, SLEEC and the proposed SEM and SEM-K with two values of β) on the Eurlex-des and Wiki10K datasets. The best and second best results are respectively bolded and underlined for each evaluation metric.

Table content: <table><tr><td/><td/><td colspan="2">REmbed SLEEC</td><td>SEM</td><td>SEM-K</td><td>SEM-K</td></tr><tr><td/><td/><td/><td/><td>(β = 500)</td><td>(β = 10)</td><td>(β = 60)</td></tr><tr><td/><td>P@1</td><td>0.7299</td><td>0.8017</td><td>0.7107</td><td>0.8024</td><td>0.8135</td></tr><tr><td colspan="2">Eurlex-des P@3</td><td>0.6064</td><td>0.6539</td><td>0.5874</td><td>0.6621</td><td>0.6714</td></tr><tr><td/><td>P@5</td><td>0.5060</td><td>0.5375</td><td>0.4916</td><td>0.5493</td><td>0.5563</td></tr><tr><td/><td/><td colspan="2">REmbed SLEEC</td><td>SEM</td><td>SEM-K</td><td>SEM-K</td></tr><tr><td/><td/><td/><td/><td colspan="3">(β = 1200) (β = 10) (β = 100)</td></tr><tr><td/><td>P@1</td><td>0.6963</td><td>0.8554</td><td>0.8517</td><td>0.8582</td><td>0.8671</td></tr><tr><td>Wiki10K</td><td>P@3</td><td>0.5790</td><td>0.7359</td><td>0.7133</td><td>0.7278</td><td>0.7385</td></tr><tr><td/><td>P@5</td><td>0.4929</td><td>0.6310</td><td>0.6171</td><td>0.6236</td><td>0.6353</td></tr></table>

The running times, in seconds, of five methods (REmbed, SLEEC and the proposed SEM and SEM-K for two values of β) on the Eurlex-des and Wiki10K datasets.

Table content: <table><tr><td/><td colspan="2">REmbed SLEEC</td><td>SEM</td><td>SEM-K</td><td>SEM-K</td></tr><tr><td/><td/><td/><td colspan="3">(β = 500) (β = 10) (β = 60)</td></tr><tr><td colspan="2">Eurlex-des 358.63</td><td>1571.30</td><td>1210.30</td><td>167.10</td><td>250.77</td></tr><tr><td>Wiki10K</td><td>2858.96</td><td>2497.00</td><td>2003.43</td><td>646.48</td><td>769.18</td></tr></table>

Peer review (iclr2017):
The paper presents the semantic embedding model for multi-label prediction. In my questions, I pointed that the proposed approach assumes the number of labels to predict is known, and the authors said this was an orthogonal question, although I don*t think it is! I was trying to understand how different is SEM from a basic MLP with softmax output which would be trained with a two step approach instead of stochastic gradient descent. It seems reasonable given their similarity to compare to this very basic baseline. Regarding the sampling strategy to estimate the posterior distribution, and the difference with Jean et al, I agree it is slightly different but I think you should definitely refer to it and point to the differences. One last question: why is it called *semantic* embeddings? usually this term is used to show some semantic meaning between trained embeddings, but this doesn*t seem to appear in this paper.