Skip to yearly menu bar Skip to main content


Poster
in
Workshop: Physics for Machine Learning

Lorentz Group Equivariant Autoencoders

Zichun Hao · Raghav Kansal · Javier Duarte · Nadya Chernyavskaya


Abstract: We develop the Lorentz group autoencoder (LGAE), an autoencoder that is equivariant with respect to the proper, orthochronous Lorentz group $\mathrm{SO}^+(3,1)$, with a latent space living in the representations of the group. We present our architecture and several experimental results on data at the Large Hadron Collider and find it outperforms a graph neural network baseline model on several compression, reconstruction, and anomaly detection tasks. The PyTorch code for our models is provided in Hao et al. (2022a).

Chat is not available.