====== Differences ====== This shows you the differences between two versions of the page.
latentartists [2013/06/30 16:04] ben |
latentartists [2013/06/30 17:09] (current) ben |
||
---|---|---|---|
Line 1: | Line 1: | ||
- | ====== **Latent Artists** ====== | + | ====== Latent Artists ====== |
by Ben Swanson and Elif Yamangil | by Ben Swanson and Elif Yamangil | ||
- | Dataset - | + | ==== Dataset - ==== |
- | Assume a fixed vocabulary $V$, which in our experiments is a company internal list | + | |
+ | Assume a fixed vocabulary $V$, which in our experiments is a list compiled by the Echonest | ||
of music related multiword terms. | of music related multiword terms. | ||
- | Each item $x_i$, from an OO-programming point of view, has the following fields | + | Each item $d_i$ in our dataset $D$, from an OO-programming point of view, has the following fields |
* Artist Name | * Artist Name | ||
* Echonest ID | * Echonest ID | ||
- | * Echonest Genres | + | * Echonest Genres (used for qualitative evaluation) |
- | * Treating a sample of reviews for this artist as a bag of words $w \in V$ | + | * ML unigram model $x_i$, treating a sample of reviews for this artist as a bag of terms $w \in V$ |
+ | |||
+ | $|V| = 3368$ | ||
+ | |||
+ | $|D| = 23541$ | ||
+ | |||
+ | |||
+ | ==== Modeling Approach - ==== | ||
+ | |||
+ | Using Factor Analysis, each $x_i$ as | ||
+ | |||
+ | $z_i \sim \mathcal{N}(0,\mathbf{I})$ | ||
+ | |||
+ | $x_i \sim \mathcal{N}(Wz,\Psi)$ | ||
+ | |||
+ | ==== Hypothesis - ==== | ||
+ | |||
+ | Much work that discovers similarity through low-dimensional representations such as PCA or Neural Networks treat | ||
+ | each data point as a single point in space. By taking the Bayesian approach described above we can not only embed data in | ||
+ | a low dimensional space but also quantify our uncertainty about each dimension. | ||
+ | |||
+ | ==== Method - ==== | ||
+ | |||
+ | The above model can be used to predict similar artists based on distance in the latent space. The traditional | ||
+ | approach would be to represent artist $d_i$ with its posterior mean $\mathbb{E}[z_i]$, and measure Euclidian distance. | ||
+ | Our alternative computes distance with KL divergence between full posteriors. The posterior probability is given as | ||
+ | |||
+ | $z_i \sim \mathcal{N}(\mathbb{E}[z_i],G)$ | ||
+ | |||
+ | where | ||
+ | |||
+ | $G = (I + W^T\Psi^{-1}W)^{-1}$ | ||
+ | |||
+ | Distance between can be computed with KL-divergence, which for Multivariate Gaussian's is given as | ||
+ | |||
+ | $KL(\mathcal{N}_0||\mathcal{N}_1) \propto (\mathbb{E}[z_0] - \mathbb{E}[z_1])\Sigma^{-1}(\mathbb{E}[z_0] - \mathbb{E}[z_1])^T + C$ | ||
+ | |||
+ | if the covariance matrix $\Sigma$ is the same for both Gaussians. This shows that if $\Sigma^{-1}$ is a multiple | ||
+ | of the identity matrix, the ranking retrieved will be the same as that of Euclidian distance between posterior means. | ||
+ | |||
+ | We can calculate the artists that are similar to an arbitrary artist by calculating their distance to all other artists using one of these | ||
+ | metrics and applying a threshold. | ||
+ | |||
+ | ==== Evaluation - ==== | ||
+ | |||
+ | We evaluate prediction of similarity on the top 300 artists by Echonest "hotttness", a set we will call $\mathcal{H}$. | ||
+ | We use the official artists similars from the Echonest database for each artist as the ground truth, provided that these | ||
+ | similar artists are also in $\mathcal{H}$. By varying the threshold on KL divergence or Euclidian distance we can trace out | ||
+ | an ROC curve. | ||
+ | |||
+ | Our results, contained in the ROC plots below, correspond to training on the full dataset and only the top 1000 by hotttness. | ||
+ | In both experimental setups the same top 300 artists are used for evaluation, the only difference is the amount of information available | ||
+ | during training. | ||
+ | |||
+ | == Hottt 1000 == | ||
+ | |||
+ | {{::1000.jpg?600|}} | ||
+ | |||
+ | == Full Dataset == | ||
+ | |||
+ | {{::full.jpg?600|}} | ||
+ | |||
+ | The results do not support our hypothesis that taking uncertainty into account would create a more robust notion of similarity. | ||
+ | While both methods clearly capture the information in the Echonest artist similar lists, the area under the ROC curve is clearly | ||
+ | greater for the simple Euclidean distance based approach. | ||
+ | |||
+ | The reason that the experimental results do not match our intuition is unclear. One possibility is that KL divergence | ||
+ | is not an appropriate metric for similarity. | ||
+ |