Flow Matching Beyond Kinematics: Generating Jets with Particle-ID and Trajectory Displacement Information

• URL: http://arxiv.org/abs/2312.00123v1
• Date: Thu, 30 Nov 2023 19:00:02 GMT
• Title: Flow Matching Beyond Kinematics: Generating Jets with Particle-ID and Trajectory Displacement Information
• Authors: Joschka Birk, Erik Buhmann, Cedric Ewen, Gregor Kasieczka, David Shih
• Abstract summary: We introduce the first generative model trained on the JetClass dataset. Our model generates jets at the constituent level, and it is a permutation-equivariant continuous normalizing flow (CNF) trained with the flow matching technique. For the first time, we also introduce a generative model that goes beyond the kinematic features of jet constituents.
• Score: 0.0
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: We introduce the first generative model trained on the JetClass dataset. Our model generates jets at the constituent level, and it is a permutation-equivariant continuous normalizing flow (CNF) trained with the flow matching technique. It is conditioned on the jet type, so that a single model can be used to generate the ten different jet types of JetClass. For the first time, we also introduce a generative model that goes beyond the kinematic features of jet constituents. The JetClass dataset includes more features, such as particle-ID and track impact parameter, and we demonstrate that our CNF can accurately model all of these additional features as well. Our generative model for JetClass expands on the versatility of existing jet generation techniques, enhancing their potential utility in high-energy physics research, and offering a more comprehensive understanding of the generated jets.

Related papers

• Not-So-Optimal Transport Flows for 3D Point Cloud Generation [58.164908756416615]
  Learning generative models of 3D point clouds is one of the fundamental problems in 3D generative learning. In this paper, we analyze the recently proposed equivariant OT flows that learn permutation invariant generative models for point-based molecular data. We show that our proposed model outperforms prior diffusion- and flow-based approaches on a wide range of unconditional generation and shape completion.
  arXiv  Detail & Related papers  (2025-02-18T02:37:34Z)
• HEP-JEPA: A foundation model for collider physics using joint embedding predictive architecture [0.0]
  We present a transformer architecture-based foundation model for tasks at high-energy particle colliders. We train the model to classify jets using a self-supervised strategy inspired by the Joint Embedding Predictive Architecture. Our model fares well with other datasets for standard classification benchmark tasks.
  arXiv  Detail & Related papers  (2025-02-06T10:16:27Z)
• Jet: A Modern Transformer-Based Normalizing Flow [62.2573739835562]
  We revisit the design of the coupling-based normalizing flow models by carefully ablating prior design choices. We achieve state-of-the-art quantitative and qualitative performance with a much simpler architecture.
  arXiv  Detail & Related papers  (2024-12-19T18:09:42Z)
• JetFormer: An Autoregressive Generative Model of Raw Images and Text [62.2573739835562]
  We propose an autoregressive decoder-only transformer - JetFormer - which is trained to directly maximize the likelihood of raw data. We leverage a normalizing flow model to obtain a soft-token image representation that is jointly trained with an autoregressive multimodal transformer. JetFormer achieves text-to-image generation quality competitive with recent VQ-VAE- and VAE-based baselines.
  arXiv  Detail & Related papers  (2024-11-29T14:14:59Z)
• OmniJet-$α$: The first cross-task foundation model for particle physics [0.0]
  Foundation models are multi-dataset and multi-task machine learning methods that once pre-trained can be fine-tuned for a variety of downstream applications. We report significant progress on this challenge on several fronts. We demonstrate transfer learning between an unsupervised problem (jet generation) and a classic supervised task (jet tagging) with our new OmniJet-$alpha$ model.
  arXiv  Detail & Related papers  (2024-03-08T19:00:01Z)
• Guided Flows for Generative Modeling and Decision Making [55.42634941614435]
  We show that Guided Flows significantly improves the sample quality in conditional image generation and zero-shot text synthesis-to-speech. Notably, we are first to apply flow models for plan generation in the offline reinforcement learning setting ax speedup in compared to diffusion models.
  arXiv  Detail & Related papers  (2023-11-22T15:07:59Z)
• PC-JeDi: Diffusion for Particle Cloud Generation in High Energy Physics [0.8246494848934447]
  We present a new method to efficiently generate jets in High Energy Physics called PC-JeDi. This method uses score-based diffusion models in conjunction with transformers which are well suited to the task of generating jets as particle clouds. PC-JeDi achieves competitive performance with current state-of-the-art methods across several metrics that evaluate the quality of the generated jets.
  arXiv  Detail & Related papers  (2023-03-09T16:23:49Z)
• Particle Transformer for Jet Tagging [4.604003661048267]
  We present JetClass, a new comprehensive dataset for jet tagging. The dataset consists of 100 M jets, about two orders of magnitude larger than existing public datasets. We propose a new Transformer-based architecture for jet tagging, called Particle Transformer (ParT)
  arXiv  Detail & Related papers  (2022-02-08T10:36:29Z)
• Controllable and Compositional Generation with Latent-Space Energy-Based Models [60.87740144816278]
  Controllable generation is one of the key requirements for successful adoption of deep generative models in real-world applications. In this work, we use energy-based models (EBMs) to handle compositional generation over a set of attributes. By composing energy functions with logical operators, this work is the first to achieve such compositionality in generating photo-realistic images of resolution 1024x1024.
  arXiv  Detail & Related papers  (2021-10-21T03:31:45Z)
• Autoencoders for unsupervised anomaly detection in high energy physics [105.54048699217668]
  We study the tagging of top jet images in a background of QCD jet images. We show that the standard autoencoder setup cannot be considered as a model-independent anomaly tagger. We suggest improved performance measures for the task of model-independent anomaly detection.
  arXiv  Detail & Related papers  (2021-04-19T05:06:57Z)
• Wind Speed Prediction using Deep Ensemble Learning with a Jet-like Architecture [0.28675177318965034]
  The design of wings, tail, and nose of a jet improves aerodynamics resulting in a smooth and controlled flight of the jet. The diverse feature spaces of the base-regressors are exploited using the jet-like ensemble architecture. The proposed DEL-Jet technique is evaluated for ten independent runs and shows that the deep and jet-like architecture helps in improving the robustness and generalization of the learning system.
  arXiv  Detail & Related papers  (2020-02-28T08:33:41Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.