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

Integrating particle flavor into deep learning models for hadronization

Jay Chan1, Xiangyang Ju1, Adam Kania2, Benjamin Nachman3,4, Vishnu Sangli5,3, and Andrzej Siodmok2

Phys. Rev. D 111, 116015 – Published 17 June, 2025

DOI: https://doi.org/10.1103/hgbg-k7js

Abstract

Hadronization models used in event generators are physics-inspired functions with many tunable parameters. Since we do not understand hadronization from first principles, there have been multiple proposals to improve the accuracy of hadronization models by utilizing more flexible parametrizations based on neural networks. These recent proposals have focused on the kinematic properties of hadrons, but a full model must also include particle flavor. In this paper, we show how to build a deep learning-based hadronization model that includes both kinematic (continuous) and flavor (discrete) degrees of freedom. Our approach is based on generative adversarial networks and we show the performance within the context of the cluster hadronization model within the herwig event generator.

View figure in article

Physics Subject Headings (PhySH)

Article Text

References (33)

  1. B. R. Webber, A QCD model for jet fragmentation including soft gluon interference, Nucl. Phys. B238, 492 (1984).
  2. G. Corcella, I. G. Knowles, G. Marchesini, S. Moretti, K. Odagiri, P. Richardson, M. H. Seymour, and B. R. Webber, herwig 6: An event generator for hadron emission reactions with interfering gluons (including supersymmetric processes), J. High Energy Phys. 01 (2001) 010.
  3. M. Bahr et al., herwig++ physics and manual, Eur. Phys. J. C 58, 639 (2008).
  4. J. Bellm et al., herwig 7.0/herwig++ 3.0 release note, Eur. Phys. J. C 76, 196 (2016).
  5. J. Bellm et al., herwig 7.2 release note, Eur. Phys. J. C 80, 452 (2020).
  6. T. Gleisberg, S. Hoeche, F. Krauss, M. Schonherr, S. Schumann, F. Siegertand J. Winter, Event generation with sherpa 1.1, J. High Energy Phys. 02 (2009) 007.
  7. E. Bothmann et al. (sherpa Collaboration), Event generation with sherpa 2.2, SciPost Phys. 7, 034 (2019).
  8. B. Andersson, G. Gustafson, G. Ingelman, and T. Sjostrand, Parton fragmentation and string dynamics, Phys. Rep. 97, 31 (1983).
  9. T. Sjostrand, Jet fragmentation of nearby partons, Nucl. Phys. B248, 469 (1984).
  10. T. Sjöstrand, S. Mrenna, and P. Z. Skands, A brief introduction to pythia 8.1, Comput. Phys. Commun. 178, 852 (2008).
  11. T. Sjöstrand, S. Mrenna, and P. Z. Skands, pythia 6.4 physics and manual, J. High Energy Phys. 05 (2006) 026.
  12. S. Badger et al., Machine learning and LHC event generation, SciPost Phys. 14, 079 (2023).
  13. R. D. Ball et al. (NNPDF Collaboration), The path to proton structure at 1% accuracy, Eur. Phys. J. C 82, 428 (2022).
  14. P. Ilten, T. Menzo, A. Youssef, and J. Zupan, Modeling hadronization using machine learning, SciPost Phys. 14, 027 (2023).
  15. A. Ghosh, X. Ju, B. Nachman, and A. Siodmok, Towards a deep learning model for hadronization, Phys. Rev. D 106, 096020 (2022).
  16. J. Chan, X. Ju, A. Kania, B. Nachman, V. Sangli, and A. Siodmok, Fitting a deep generative hadronization model, J. High Energy Phys. 09 (2023) 084.
  17. C. Bierlich, P. Ilten, T. Menzo, S. Mrenna, M. Szewc, M. K. Wilkinson, A. Youssef, and J. Zupan, Towards a data-driven model of hadronization using normalizing flows, SciPost Phys. 17, 045 (2024).
  18. I. J. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde-Farley, S. Ozair et al., Generative adversarial nets, in Proceedings of the 27th International Conference on Neural Information Processing Systems—Volume 2, NIPS’14 (MIT Press, Cambridge, MA, USA, 2014), pp. 2672–2680, http://dl.acm.org/citation.cfm?id=2969033.2969125.
  19. A. Creswell, T. White, V. Dumoulin, K. Arulkumaran, B. Sengupta, and A. A. Bharath, Generative adversarial networks: An overview, IEEE Signal Process. Mag. 35, 53 (2018).
  20. D. Rezende and S. Mohamed, Variational inference with normalizing flows, in Proceedings of the 32nd International Conference on Machine Learning (2015), Vol. 37, pp. 1530–1538, arXiv:1505.05770.
  21. I. Kobyzev, S. Prince, and M. Brubaker, Normalizing flows: An introduction and review of current methods, IEEE Trans. Pattern Anal. Mach. Intell. 43, C1 (2021).
  22. M. Arjovsky, S. Chintala, and L. Bottou, Wasserstein generative adversarial networks, in Proceedings of the 34th International Conference on Machine Learning, edited by D. Precup and Y. W. Teh, Vol. 70 of Proceedings of Machine Learning Research (PMLR, 2017), pp. 214–223, https://proceedings.mlr.press/v70/arjovsky17a.html.
  23. L. Mescheder, A. Geiger, and S. Nowozin, Which training methods for GANs do actually converge?, in Proceedings of the 35th International Conference on Machine Learning, edited by J. Dy and A. Krause, Vol. 80 of Proceedings of Machine Learning Research (PMLR, 2018), pp. 3481–3490, https://proceedings.mlr.press/v80/mescheder18a.html.
  24. B. Xu, N. Wang, T. Chen, and M. Li, Empirical evaluation of rectified activations in convolutional network, arXiv:1505.00853.
  25. E. Jang, S. Gu, and B. Poole, Categorical reparameterization with gumbel-softmax, in International Conference on Learning Representations (2017), https://openreview.net/forum?id=rkE3y85ee.
  26. A. Paszke, S. Gross, F. Massa, A. Lerer, J. Bradbury, G. Chanan et al., pytorch: An imperative style, high-performance deep learning library, in Advances in Neural Information Processing Systems 32, edited by H. Wallach, H. Larochelle, A. Beygelzimer, F. d’ Alché-Buc, E. Fox, and R. Garnett (Curran Associates, Inc., Vancouver, 2019), pp. 8024–8035.
  27. D. Kingma and J. Ba, adam: A method for stochastic optimization, arXiv:1412.6980.
  28. L. Biewald, Experiment tracking with weights and biases (2020), https://www.wandb.com/.
  29. R. L. Workman et al. (Particle Data Group), Review of particle physics, Prog. Theor. Exp. Phys. 2022, 083C01 (2022).
  30. J. Birk, E. Buhmann, C. Ewen, G. Kasieczka, and D. Shih, Flow matching beyond kinematics: Generating jets with particle-ID and trajectory displacement information, Phys. Rev. D 111, 052008 (2025).
  31. J. Chan, X. Ju, A. Kania, B. Nachman, V. Sangli, and A. Siodmok, herwig dataset for hadml particle gan training, 10.5281/zenodo.10246934 (2023).
  32. J. Chan, X. Ju, A. Kania, B. Nachman, V. Sangli, and A. Siodmok, Code for hadml particle gan training, 10.5281/zenodo.10275487 (2023).
  33. https://github.com/hep-lbdl/hadml/releases/tag/2.0.0

Outline

Information

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation