The β-VAE network compresses each trajectory’s time series of x,y positions and velocities into a compact set of learned coordinates (a latent space) that preserve reconstructability. We further incorporated a triplet loss term (42) to encourage the learning of discriminate features, which help separate the latent representation of each search algorithm. Visualizing the resulting latent representation by projecting it into 2-D using Principal Component Analysis (PCA) demonstrates that the six algorithm classes form distinct, well-separated clusters (Figure 4B).
How much variance in the 70-dimensions do the first 2 PCs account for?
The discrete structure of the PC space is interesting. I wonder how the space would change if you were to train the model on non-instantaneous windows. I would hypothesize a smoother space but, maybe paradoxically, the possibility of greater classification accuracy given the autocorrelation of the velocity components.
Also, the 70-dimensional latent space post-optimization is intriguing. Any idea why that size is optimal with respect to the input features?