Reuse & Permissions

It is not necessary to obtain permission to reuse this article or its components as it is available under the terms of the Creative Commons Attribution 4.0 International license. This license permits unrestricted use, distribution, and reproduction in any medium, provided attribution to the author(s) and the published article's title, journal citation, and DOI are maintained. Please note that some figures may have been included with permission from other third parties. It is your responsibility to obtain the proper permission from the rights holder directly for these figures.

Export citation

Export citation

Choose format for download:

Download Citation
  • Open Access

Multiscale optimal transport for complete collider events

Tianji Cai*

Nathaniel Craig†

Katy Craig‡

Xinyuan Lin§

  • School of Physical Science and Engineering, Tongji University, Shanghai 200092, China and State Key Laboratory of Autonomous Intelligent Unmanned Systems, MOE Frontiers Science Center for Intelligent Autonomous Systems, Tongji University, Shanghai 200092, China

  • *Contact author: tianjiresearch@gmail.com
  • †Contact author: ncraig@ucsb.edu
  • ‡Contact author: kcraig@math.ucsb.edu
  • §Contact author: xil244@ucsd.edu

Phys. Rev. D 112, 036021 – Published 29 August, 2025

DOI: https://doi.org/10.1103/7nhv-hndt

Abstract

Building upon the success of optimal transport metrics defined for single collinear jets, we develop a multiscale framework that models entire collider events as distributions on the manifold of their constituent jets, which are themselves distributions on the ground space of the calorimeter. This hierarchical structure of optimal transport effectively captures relevant physics at different scales. We demonstrate the versatility of our method in two event classification tasks, which respectively emphasize intrajet substructure and interjet spatial correlations. Our results highlight the importance of a nested structure of manifolds in the treatment of full collider events, broadening the applicability of optimal transport methods in collider analyses.

View figure in article

Physics Subject Headings (PhySH)

Article Text

References (55)

  1. A. J. Larkoski, I. Moult, and B. Nachman, Jet substructure at the large hadron collider: A review of recent advances in theory and machine learning, Phys. Rep. 841, 1 (2020).
  2. D. Guest, K. Cranmer, and D. Whiteson, Deep learning and its application to LHC physics, Annu. Rev. Nucl. Part. Sci. 68, 161 (2018).
  3. K. Albertsson et al., Machine learning in high energy physics community white paper, J. Phys. Conf. Ser. 1085, 022008 (2018).
  4. A. Radovic, M. Williams, D. Rousseau, M. Kagan, D. Bonacorsi, A. Himmel, A. Aurisano, K. Terao, and T. Wongjirad, Machine learning at the energy and intensity frontiers of particle physics, Nature (London) 560, 41 (2018).
  5. G. Carleo, I. Cirac, K. Cranmer, L. Daudet, M. Schuld, N. Tishby, L. Vogt-Maranto, and L. Zdeborová, Machine learning and the physical sciences, Rev. Mod. Phys. 91, 045002 (2019).
  6. D. Bourilkov, Machine and deep learning applications in particle physics, Int. J. Mod. Phys. A 34, 1930019 (2019).
  7. M. D. Schwartz, Modern machine learning and particle physics, Harvard Data Sci. Rev. 3 (2021).
  8. G. Karagiorgi, G. Kasieczka, S. Kravitz, B. Nachman, and D. Shih, Machine learning in the search for new fundamental physics, arXiv:2112.03769.
  9. P. Shanahan, K. Terao, and D. Whiteson, Snowmass 2021 computational frontier compf03 topical group report: Machine learning, arXiv:2209.07559.
  10. P. T. Komiske, E. M. Metodiev, and J. Thaler, Metric space of collider events, Phys. Rev. Lett. 123, 041801 (2019).
  11. P. T. Komiske, E. M. Metodiev, and J. Thaler, The hidden geometry of particle collisions, J. High Energy Phys. 07 (2020) 006.
  12. T. Cai, J. Cheng, N. Craig, and K. Craig, Linearized optimal transport for collider events, Phys. Rev. D 102, 116019 (2020).
  13. T. Cai, J. Cheng, K. Craig, and N. Craig, Which metric on the space of collider events?, Phys. Rev. D 105, 076003 (2022).
  14. P. T. Komiske, R. Mastandrea, E. M. Metodiev, P. Naik, and J. Thaler, Exploring the space of jets with CMS open data, Phys. Rev. D 101, 034009 (2020).
  15. C. Cesarotti and J. Thaler, A robust measure of event isotropy at colliders, J. High Energy Phys. 08 (2020) 084.
  16. C. Cesarotti, M. Reece, and M. J. Strassler, The efficacy of event isotropy as an event shape observable, J. High Energy Phys. 07 (2021) 215.
  17. T. Cheng, J.-F. Arguin, J. Leissner-Martin, J. Pilette, and T. Golling, Variational autoencoders for anomalous jet tagging, Phys. Rev. D 107, 016002 (2023).
  18. M. Crispim Romão, N. F. Castro, J. G. Milhano, R. Pedro, and T. Vale, Use of a generalized energy Mover’s distance in the search for rare phenomena at colliders, Eur. Phys. J. C 81, 192 (2021).
  19. G. Aad, B. Abbott et al. (The ATLAS Collaboration), Measurements of multijet event isotropies using optimal transport with the ATLAS detector, J. High Energy Phys. 10 (2023) 060.
  20. N. Craig, J. N. Howard, and H. Li, Exploring optimal transport for event-level anomaly detection at the large hadron collider, arXiv:2401.15542.
  21. L. Gouskos, F. Iemmi, S. Liechti, B. Maier, V. Mikuni, and H. Qu, Optimal transport for a novel event description at hadron colliders, Phys. Rev. D 108, 096003 (2023).
  22. C. Cesarotti and M. LeBlanc, A field guide to event-shape observables using optimal transport, arXiv:2409.13150.
  23. R. Gambhir, A. J. Larkoski, and J. Thaler, SPECTER: Efficient evaluation of the spectral EMD, J. High Energy Phys. 12 (2025) 219.
  24. D. Ba, A. S. Dogra, R. Gambhir, A. Tasissa, and J. Thaler, SHAPER: Can you hear the shape of a jet?, J. High Energy Phys. 06 (2023) 195.
  25. T. Cai, Optimal transport for high energy physics, Ph.D. thesis, UC, Santa Barbara (main) (2023), https://escholarship.org/uc/item/93n2s2ff.
  26. A. Mullin, S. Nicholls, H. Pacey, M. Parker, M. White, and S. Williams, Does SUSY have friends? A new approach for LHC event analysis, J. High Energy Phys. 02 (2021) 160.
  27. T. Cai, J. Cheng, N. Craig, G. Koszegi, and A. J. Larkoski, The phase space distance between collider events, J. High Energy Phys. 09 (2024) 054.
  28. G. Stein, U. Seljak, and B. Dai, Unsupervised in-distribution anomaly detection of new physics through conditional density estimation, arXiv:2012.11638.
  29. G. Kasieczka et al., The LHC olympics 2020 a community challenge for anomaly detection in high energy physics, Rep. Prog. Phys. 84, 124201 (2021).
  30. J. N. Howard, S. Mandt, D. Whiteson, and Y. Yang, Learning to simulate high energy particle collisions from unlabeled data, Sci. Rep. 12, 7567 (2022).
  31. G. D. Guglielmo, F. Fahim, C. Herwig, M. B. Valentin, J. Duarte, C. Gingu, P. Harris, J. Hirschauer, M. Kwok, V. Loncar, Y. Luo, L. Miranda, J. Ngadiuba, D. Noonan, S. Ogrenci-Memik, M. Pierini, S. Summers, and N. Tran, A reconfigurable neural network ASIC for detector front-end data compression at the HL-LHC, IEEE Trans. Nucl. Sci. 68, 2179 (2021).
  32. R. Kansal, J. Duarte, H. Su, B. Orzari, T. Tomei, M. Pierini, M. Touranakou, J.-R. Vlimant, and D. Gunopulos, Particle cloud generation with message passing generative adversarial networks, arXiv:2106.11535.
  33. J. H. Collins, An exploration of learnt representations of w jets, arXiv:2109.10919.
  34. T. Cai, J. Cheng, B. Schmitzer, and M. Thorpe, The linearized Hellinger-Kkantorovich distance, SIAM J. Imag. Sci. 15, 45 (2022).
  35. J. Lott, Some geometric calculations on wasserstein space, arXiv:math/0612562.
  36. L. Ambrosio, N. Gigli, and G. Savaré, Gradient Flows: In Metric Spaces and in the Space of Probability Measures (Springer Science & Business Media, New York, 2008).
  37. F. Otto, The geometry of dissipative evolution equations: The porous medium equation, Commun. Partial Differ. Equations 26, 101 (2001).
  38. D. Bertsekas, The auction algorithm: A distributed relaxation method for the assignment problem, Ann. Oper. Res. 14, 105 (1988).
  39. J. Altschuler, J. Weed, and P. Rigollet, Near-linear time approximation algorithms for optimal transport via Sinkhorn iteration, Adv. Neural Inf. Process. Syst. 30, 1961 (2018).
  40. M. Cuturi, Sinkhorn distances: Lightspeed computation of optimal transportation distances, Adv. Neural Inf. Process. Syst. 26, 2292 (2013).
  41. W. Wang, D. Slepčev, S. Basu, J. A. Ozolek, and G. K. Rohde, A linear optimal transportation framework for quantifying and visualizing variations in sets of images, Int. J. Comput. Vis. 101, 254 (2013).
  42. G. Gur-Ari, M. Papucci, and G. Perez, Classification of energy flow observables in narrow jets, arXiv:1101.2905.
  43. J. Gallicchio and M. D. Schwartz, Quark and gluon jet substructure, J. High Energy Phys. 04 (2013) 090.
  44. M. Cacciari, G. P. Salam, and G. Soyez, The anti-kt jet clustering algorithm, J. High Energy Phys. 04 (2008) 063.
  45. L. Ambrosio and N. Gigli, A user’s guide to optimal transport, in Modelling and Optimisation of Flows on Networks: Cetraro, Italy 2009, Editors: Benedetto Piccoli, Michel Rascle (Springer, Berlin, Heidelberg, 2013), pp. 1–155.
  46. J. Cuesta-Albertos, C. Matrán, and J. Rodríguez-Rodríguez, Approximation to probabilities through uniform laws on convex sets, J. Theor. Probab. 16, 363 (2003).
  47. T. M. Cover and P. E. Hart, Nearest neighbor pattern classification, IEEE Trans. Inf. Theory 13, 21 (1967).
  48. I. A. Bertram (D0 Collaboration), Jet results at the D0 experiment, Acta Phys. Pol. B 33, 3141 (2002).
  49. Z. Nagy, Next-to-leading order calculation of three-jet observables in hadron-hadron collisions, Phys. Rev. D 68, 094002 (2003).
  50. A. Banfi, G. P. Salam, and G. Zanderighi, Resummed event shapes at hadron-hadron colliders, J. High Energy Phys. 08 (2004) 062.
  51. J. Alwall, R. Frederix, S. Frixione, V. Hirschi, F. Maltoni, O. Mattelaer, H.-S. Shao, T. Stelzer, P. Torrielli, and M. Zaro, The automated computation of tree-level and next-to-leading order differential cross sections, and their matching to parton shower simulations, J. High Energy Phys. 07 (2014) 079.
  52. C. Bierlich, S. Chakraborty, N. Desai, L. Gellersen, I. Helenius, P. Ilten, L. Lönnblad, S. Mrenna, S. Prestel, C. T. Preuss, T. Sjöstrand, P. Skands, M. Utheim, and R. Verheyen, A comprehensive guide to the physics and usage of pythia 8.3, SciPost Phys. Codebases 2022, 8 (2022).
  53. M. Cacciari, G. P. Salam, and G. Soyez, fastjet user manual, Eur. Phys. J. C 72, 1896 (2012).
  54. T. Cai, N. Craig, K. Craig, and X. Lin, Multi-scale optimal transport for complete collider events, arXiv:2501.10681.
  55. T. Manole, P. Bryant, J. Alison, M. Kuusela, and L. Wasserman, Background modeling for double Higgs boson production: Density ratios and optimal transport, arXiv:2208.02807.

Outline

Information

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation