Improving generative model-based unfolding with Schrödinger bridges

Sascha Diefenbacher,Guan-Horng Liu,Vinicius Mikuni,Benjamin Nachman,Weili Nie
DOI: https://doi.org/10.1103/physrevd.109.076011
IF: 5.407
2024-04-12
Physical Review D
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 Schrödinger 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. https://doi.org/10.1103/PhysRevD.109.076011 Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article's title, journal citation, and DOI. Funded by SCOAP 3 . Published by the American Physical Society
astronomy & astrophysics,physics, particles & fields
What problem does this paper attempt to address?