This repository contains the experiments code for Lorentz Local Canonicalization (LLoCa) by Jonas Spinner, Luigi Favaro, Peter Lippmann, Sebastian Pitz, Gerrit Gerhartz, Tilman Plehn, and Fred A. Hamprecht. Relevant links:
- Package: https://github.com/heidelberg-hepml/lloca (documentation)
- Paper: Lorentz Local Canonicalization: How to make any Network Lorentz-Equivariant (computer science audience)
- Paper: Lorentz-Equivariance without Limitations (high-energy physics audience)
For instructions on how to run the code, loading the datasets, and commands for reproducing the results in the papers, have a look at REPRODUCE.md.
More packages for Lorentz-equivariant networks used in this repo:
- https://github.com/heidelberg-hepml/lgatr (documentation)
- https://github.com/heidelberg-hepml/pelican-lite (documentation)
If you find this code useful in your research, please cite our papers
@article{Spinner:2025prg,
author = "Spinner, Jonas and Favaro, Luigi and Lippmann, Peter and Pitz, Sebastian and Gerhartz, Gerrit and Plehn, Tilman and Hamprecht, Fred A.",
title = "{Lorentz Local Canonicalization: How to Make Any Network Lorentz-Equivariant}",
eprint = "2505.20280",
archivePrefix = "arXiv",
primaryClass = "stat.ML",
month = "5",
year = "2025"
}
@article{Favaro:2025pgz,
author = "Favaro, Luigi and Gerhartz, Gerrit and Hamprecht, Fred A. and Lippmann, Peter and Pitz, Sebastian and Plehn, Tilman and Qu, Huilin and Spinner, Jonas",
title = "{Lorentz-Equivariance without Limitations}",
eprint = "2508.14898",
archivePrefix = "arXiv",
primaryClass = "hep-ph",
month = "8",
year = "2025"
}