# JUNIPR: a Framework for Unsupervised Machine Learning in Particle Physics

arXiv: https://arxiv.org/abs/1804.09720 · HTML: https://mlsanity.com/paper/1804.09720/

## Key facts

- Cited by 18 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 25 Apr 2018.

## Built on by (most cited first, 18 of 18)

- [A Living Review of Machine Learning for Particle Physics](https://mlsanity.com/paper/2102.02770/) (2021)
- [A comprehensive review of Quantum Machine Learning: from NISQ to Fault Tolerance](https://mlsanity.com/paper/2401.11351/) (2024)
- [Generative Networks for Precision Enthusiasts](https://mlsanity.com/paper/2110.13632/) (2021)
- [CaloClouds II: Ultra-Fast Geometry-Independent Highly-Granular Calorimeter Simulation](https://mlsanity.com/paper/2309.05704/) (2023)
- [Understanding Event-Generation Networks via Uncertainties](https://mlsanity.com/paper/2104.04543/) (2021)
- [E Pluribus Unum Ex Machina: Learning from Many Collider Events at Once](https://mlsanity.com/paper/2101.07263/) (2021)
- [Convolutional L2LFlows: Generating Accurate Showers in Highly Granular Calorimeters Using Convolutional Normalizing Flows](https://mlsanity.com/paper/2405.20407/) (2024)
- [High-dimensional and Permutation Invariant Anomaly Detection](https://mlsanity.com/paper/2306.03933/) (2023)
- [A Primer on Quantum Machine Learning](https://mlsanity.com/paper/2511.15969/) (2025)
- [Learning Likelihood Ratios with Neural Network Classifiers](https://mlsanity.com/paper/2305.10500/) (2023)
- [A Review of Physics-based Machine Learning in Civil Engineering](https://mlsanity.com/paper/2110.04600/) (2021)
- [A Comparison Between Invariant and Equivariant Classical and Quantum Graph Neural Networks](https://mlsanity.com/paper/2311.18672/) (2023)
- [ELSA -- Enhanced latent spaces for improved collider simulations](https://mlsanity.com/paper/2305.07696/) (2023)
- [Sequence-based Machine Learning Models in Jet Physics](https://mlsanity.com/paper/2102.06128/) (2021)
- [The use of Generative Adversarial Networks to characterise new physics in multi-lepton final states at the LHC](https://mlsanity.com/paper/2105.14933/) (2021)
- [Extrapolating Jet Radiation with Autoregressive Transformers](https://mlsanity.com/paper/2412.12074/) (2024)
- [BitHEP -- The Limits of Low-Precision ML in HEP](https://mlsanity.com/paper/2504.03387/) (2025)
- [Likelihood-Based Unsupervised Anomaly Detection in CMS Dijet Events](https://mlsanity.com/paper/2609.06686/) (2026)

## Abstract (arXiv)

In applications of machine learning to particle physics, a persistent challenge is how to go beyond discrimination to learn about the underlying physics. To this end, a powerful tool would be a framework for unsupervised learning, where the machine learns the intricate high-dimensional contours of the data upon which it is trained, without reference to pre-established labels. In order to approach such a complex task, an unsupervised network must be structured intelligently, based on a qualitative understanding of the data. In this paper, we scaffold the neural network's architecture around a leading-order model of the physics underlying the data. In addition to making unsupervised learning tractable, this design actually alleviates existing tensions between performance and interpretability. We call the framework JUNIPR: "Jets from UNsupervised Interpretable PRobabilistic models". In this approach, the set of particle momenta composing a jet are clustered into a binary tree that the neural network examines sequentially. Training is unsupervised and unrestricted: the network could decide that the data bears little correspondence to the chosen tree structure. However, when there is a correspondence, the network's output along the tree has a direct physical interpretation. JUNIPR models can perform discrimination tasks, through the statistically optimal likelihood-ratio test, and they permit visualizations of discrimination power at each branching in a jet's tree. Additionally, JUNIPR models provide a probability distribution from which events can be drawn, providing a data-driven Monte Carlo generator. As a third application, JUNIPR models can reweight events from one (e.g. simulated) data set to agree with distributions from another (e.g. experimental) data set.
