Linearized Optimal Transport for Jet Physics

Jul 13, 2021, 4:45 PM
15m
Track E (Zoom)

Track E

Zoom

talk Computation, Machine Learning, and AI Computation, Machine Learning, and AI

Speaker

Ms Tianji Cai (University of California, Santa Barbara)

Description

As an unsupervised machine learning strategy, optimal transport (OT) has been applied to jet physics for the computation of distance between collider events. Here we generalize the Energy Mover’s Distance to include both the balanced Wasserstein-2 (W2) distance and the unbalanced Hellinger-Kantorovich (HK) distance. Whereas the W2 distance only allows for mass to be transported, the HK distance allows mass to be transported, created and destroyed, therefore naturally incorporating the total pt difference of the jets. Both distances enjoy a weak Riemannian structure and thus admit linear approximation. Such a linear framework significantly reduces the computational cost and in addition provides a Euclidean embedding amenable to simple machine learning algorithms and visualization techniques downstream. Here we demonstrate the benefit of this linear approach for jet classification and study its behavior in the presence of pileup.

Are you are a member of the APS Division of Particles and Fields? No

Primary authors

Ms Tianji Cai (University of California, Santa Barbara) Ms Junyi Cheng (University of California, Santa Barbara) Prof. Nathaniel Craig (University of California, Santa Barbara) Prof. Katy Craig (University of California, Santa Barbara) Prof. Bernhard Schmitzer (UNIVERSITAT GOTTINGEN) Prof. Matthew Thorpe (UNIVERSITY OF MANCHESTER)

Presentation materials