Aspen Open Jets: a real-world ML-ready dataset for jet physics

DOI

This dataset contains approximately 180 M boosted jets, derived from open data collected by the CMS experiment at the Large Hadron Collider (LHC) in 2016 — specifically the JetHT datastream — and presented in a format suitable for Machine Learning (ML) applications. A detailed description of the dataset and how it was produced can be found in the companion paper, arxiv 2412.10504.

For each jet we store its transverse momentum (p_T), pseudorapidity (eta), and azimuthal angular coordinate (phi). We also store its mass, groomed with the softdrop algorithm as computed within the CMS reconstruction. Up to 150 constituents of the jet are stored. For each constituent, its 4-momentum is stored in the format (p_x, p_y, p_z, E). We additionally store its transverse impact parameter (d_0) and longitudinal impact parameter (d_z) with their uncertainties, the charge of the candidate, its particle-ID (PID) in the PDG format (note that neutral hadrons are assigned the PID=130 of the neutral kaon K_L^0, while positively/negatively charged hadrons are assigned PID=211 of the charged pion) and its weight from the PUPPI algorithm. We also include additional jet substructure quantities computed within the CMS reconstruction, including the number of constituents in the jet, N-subjettiness variables, various jet-tagging observables from the CMS implementation of ParticleNet and a regression of the jet mass from ParticleNet.

Events are stored in h5 format with 4 keys:

'event_info', shape (N_jets, 3): [Run Number, LumiBlock, Event Number]
'jet_kinematics', shape (N_jets, 4): [pt, eta, phi, softdrop mass]
'PFCands', shape (N_jets, 150, 11): Zero padded list of up to 150 PFcandidates inside the jet.
Info for each candidate is [px, py, pz, E, d0, d0Err, dz, dzErr, charge, PDG ID, PUPPI weight]
'jet_tagging', shape (N_jets, 13): Tagging info/scores for the AK8 jet.
Info for each jet: [nConstituents, tau1, tau2, tau3, tau4, ParticleNet H4q vs QCD, ParticleNet Hbb vs QCD, ParticleNet Hcc vs QCD, ParticleNet QCD score, ParticleNet T vs QCD, ParticleNet W vs QCD, ParticleNet Z vs QCD, ParticleNet regressed mass]

The code that was used to create Aspen Open Jets from CMS open data can be found at https://github.com/OzAmram/AOJProcessing, and the code used for the OmniJet-alpha model and its training can be found at https://github.com/uhh-pd-ml/omnijet_alpha.

 

This work was initiated at the Aspen Center for Physics, supported by National Science Foundation grant PHY-2210452. We would like to thank Alexander Mück for discussions. The research of MK and HR-G is supported by the Deutsche Forschungsgemeinschaft DFG under grant 396021762 -- TRR 257: Particle physics phenomenology after the Higgs discovery. JB, AH and GK are supported by the DFG under the German Excellence Initiative -- EXC 2121 Quantum Universe – 390833306, and by PUNCH4NFDI – project number 460248186. DAF, IP, and DS are supported by DOE grant DOE-SC0010008. OA is supported by Fermi Research Alliance, LLC under Contract No. DE-AC02-07CH11359 with the U.S. Department of Energy, Office of Science, Office of High Energy Physics. LA is supported by the University of Bologna. Additionally, we acknowledge support from the Maxwell computational resources at Deutsches Elektronen-Synchrotron DESY, Hamburg, Germany, and computing resources provided by RWTH Aachen University under project rwth0934. This research used resources of the National Energy Research Scientific Computing Center, a DOE Office of Science User Facility supported by the Office of Science of the U.S. Department of Energy under Contract No. DE-AC02-05CH11231 using NERSC award HEP-ERCAP0027491.

Identifier
DOI https://doi.org/10.25592/uhhfdm.16505
Related Identifier IsPartOf https://doi.org/10.25592/uhhfdm.16504
Metadata Access https://www.fdr.uni-hamburg.de/oai2d?verb=GetRecord&metadataPrefix=oai_datacite&identifier=oai:fdr.uni-hamburg.de:16505
Provenance
Creator Amram, Oz ORCID logo; Anzalone, Luca ORCID logo; Birk, Joschka ORCID logo; Faroughy, Darius A. ORCID logo; Hallin, Anna (ORCID: 0000-0002-1551-814X); Kasieczka, Gregor ORCID logo; Krämer, Michael ORCID logo; Pang, Ian ORCID logo; Reyes-Gonzalez, Humberto ORCID logo; Shih, David ORCID logo
Publisher Universität Hamburg
Publication Year 2024
Rights Creative Commons Attribution 4.0 International; Open Access; https://creativecommons.org/licenses/by/4.0/legalcode; info:eu-repo/semantics/openAccess
OpenAccess true
Representation
Resource Type Dataset
Version 1.0
Discipline Natural Sciences; Physics