Improving Generative Model-based Unfolding with Schrödinger Bridges

Abstract

Machine learning-based unfolding has enabled unbinned and high-dimensional differential cross section measurements. Two main approaches have emerged in this research area: one based on discriminative models and one based on generative models. The main advantage of discriminative models is that they learn a small correction to a starting simulation while generative models scale better to regions of phase space with little data. We propose to use Schroedinger Bridges and diffusion models to create SBUnfold, an unfolding approach that combines the strengths of both discriminative and generative models. The key feature of SBUnfold is that its generative model maps one set of events into another without having to go through a known probability density as is the case for normalizing flows and standard diffusion models. We show that SBUnfold achieves excellent performance compared to state of the art methods on a synthetic Z+jets dataset.

9 pages, 5 figures

References in corpus (3)

• PYTHIA 6.4 Physics and Manual
• An Introduction to PYTHIA 8.2
• Herwig++ Physics and Manual

Cited by in corpus (21)

• CaloDREAM -- Detector Response Emulation via Attentive flow Matching
• The Landscape of Unfolding with Machine Learning
• Precision-Machine Learning for the Matrix Element Method
• A simultaneous unbinned differential cross section measurement of twenty-four $Z$+jets kinematic observables with the ATLAS detector
• A Lorentz-Equivariant Transformer for All of the LHC
• Normalizing Flows for High-Dimensional Detector Simulations
• Differentiable MadNIS-Lite
• Full Event Particle-Level Unfolding with Variable-Length Latent Variational Diffusion
• CaloChallenge 2022: A Community Challenge for Fast Calorimeter Simulation
• Accurate Surrogate Amplitudes with Calibrated Uncertainties
• Unsupervised and lightly supervised learning in particle physics
• Moment Unfolding
• Unifying Simulation and Inference with Normalizing Flows
• Generative Unfolding with Distribution Mapping
• Machine Learning-Assisted Unfolding for Neutrino Cross-section Measurements with the OmniFold Technique
• A Practical Guide to Unbinned Unfolding
• Extrapolating Jet Radiation with Autoregressive Transformers
• Tools for Unbinned Unfolding
• How to Unfold Top Decays
• Response Matrix Estimation in Unfolding Differential Cross Sections
• ScatterPrism: convergence for generative simulation and inverse problems in particle and nuclear physics