Evaluating the performance of variational autoencoder on unlabeled data

Viewed 39

I've designed a variational autoencoder (VAE) that clusters sequential time series data. To evaluate the performance of VAE on labeled data, First, I run KMeans on the raw data and compare the generated labels with the true labels using Adjusted Mutual Info Score (AMI). Then, after the model is trained, I pass validation data to it, run KMeans on latent vectors, and compare the generated labels with the true labels of validation data using AMI. Finally, I compare the two AMI scores with each other to see if KMeans has better performance on the latent vectors than the raw data.

My question is this: How can we evaluate the performance of VAE when the data is unlabeled?

I know we can run KMeans on the raw data and generate labels for it, but in this case, since we consider the generated labels as true labels, how can we compare the performance of KMeans on the raw data with KMeans on the latent vectors?

Note: The model is totally unsupervised. Labels (if exist) are not used in the training process. They're used only for evaluation.

0 Answers
Related