11–15 Mar 2024
Charles B. Wang Center, Stony Brook University
US/Eastern timezone

Generic representations of jets at detector-level with self-supervised learning

13 Mar 2024, 15:50
20m
Lecture Hall 2 ( Charles B. Wang Center, Stony Brook University )

Lecture Hall 2

Charles B. Wang Center, Stony Brook University

100 Circle Rd, Stony Brook, NY 11794
Oral Track 2: Data Analysis - Algorithms and Tools Track 2: Data Analysis - Algorithms and Tools

Speaker

Patrick Rieck (New York University (US))

Description

Supervised learning has been used successfully for jet classification and to predict a range of jet properties, such as mass and energy. Each model learns to encode jet features, resulting in a representation that is tailored to its specific task. But could the common elements underlying such tasks be combined in a single foundation model to extract features generically? To address this question, we explore self-supervised learning (SSL), inspired by its applications in the domains of computer vision and natural language processing. Besides offering a simpler and more resource-effective route when learning multiple tasks, SSL can be trained on unlabeled data, e.g. large sets of collision data. We demonstrate that a jet representation obtained through SSL can be readily fine-tuned for downstream tasks of jet kinematics prediction and jet classification. Compared to existing studies in this direction, we use a realistic full-coverage calorimeter simulation, leading to results that more faithfully reflect the prospects at real collider experiments.

Significance

Going beyond previous work, we present novel approaches to the training of our foundation model, with the aim of leveraging large unlabeled datasets (opening the door to novel data-driven analysis techniques) and learning transferable jet representations that are invariant to detector properties.

References

  • Presentation at the ML4Jets Workshop, Hamburg, 2023: https://indico.cern.ch/event/1253794/contributions/5588641/

  • Paper presenting the collider detector simulation used in this work: “Configurable calorimeter simulation for AI applications”

Primary authors

Dmitrii Kobylianskii (Weizmann Institute of Science (IL)) Eilam Gross (Weizmann Institute of Science (IL)) Etienne Dreyer (Weizmann Institute of Science (IL)) Garrett Merz Kyle Stuart Cranmer (University of Wisconsin Madison (US)) Nathalie Soybelman (Weizmann Institute of Science (IL)) Nilotpal Kakati (Weizmann Institute of Science (IL)) Patrick Rieck (New York University (US))

Presentation materials