- Open Access
Ultrafast, event-by-event heavy-ion simulations for next-generation experiments
Phys. Rev. C 112, 054907 – Published 24 November, 2025
DOI: https://doi.org/10.1103/wyq5-hlp5
Abstract
We present a novel deep generative framework that uses probabilistic diffusion models for ultrafast, event-by-event simulations of heavy-ion collision output. This new framework is trained on ultrarelativistic quantum molecular dynamics (UrQMD) cascade data to generate a full collision event output containing 26 distinct hadron species. The output is represented as a point cloud, where each point is defined by a particle's momentum vector and its corresponding species information. Our architecture integrates a normalizing flow-based condition generator that encodes global event features into a latent vector, and a diffusion model that synthesizes a point cloud of particles based on this condition. A detailed description of the model and an in-depth analysis of its performance is provided. The conditional point-cloud diffusion model learns to generate realistic output particles of collision events which successfully reproduce the UrQMD distributions for multiplicity, momentum, and rapidity of each hadron type. The flexible point-cloud representation of the event output preserves full event-level granularity, enabling direct application to inverse problems and parameter estimation tasks while also making it easily adaptable for accelerating any event-by-event model calculation or detector simulation.
Physics Subject Headings (PhySH)
See Also
Toward a foundation model for heavy-ion collision experiments based on point-cloud diffusion
Article Text
References (93)
- H. Stöcker and W. Greiner, Phys. Rep. 137, 277 (1986).
- K. Fukushima and T. Hatsuda, Rep. Prog. Phys. 74, 014001 (2011).
- Y. Aoki, G. Endrodi, Z. Fodor, S. D. Katz, and K. K. Szabo, Nature (London) 443, 675 (2006).
- S. Borsanyi, G. Endrodi, Z. Fodor, A. Jakovac, S. D. Katz, S. Krieg, C. Ratti, and K. K. Szabo, J. High Energy Phys. 11 (2010) 077.
- A. Bazavov et al. (HotQCD Collaboration), Phys. Rev. D 90, 094503 (2014).
- M. Cheng et al., Phys. Rev. D 77, 014511 (2008).
- A. Bazavov et al., Phys. Rev. D 85, 054503 (2012).
- S. Aoki et al. (PACS-CS Collaboration), Phys. Rev. D 79, 034503 (2009).
- G. Agakichiev et al. (HADES Collaboration), Eur. Phys. J. A 41, 243 (2009).
- C. Yang et al. (STAR Collaboration), Nucl. Phys. A 967, 800 (2017).
- M. M. Aggarwal et al. (STAR Collaboration), arXiv:1007.2613.
- L. Adamczyk et al. (STAR Collaboration), Phys. Rev. C 96, 044904 (2017).
- K. C. Meehan (STAR Collaboration), Nucl. Phys. A 956, 878 (2016).
- T. Ablyazimov et al. (CBM Collaboration), Eur. Phys. J. A 53, 60 (2017).
- J. C. Yang et al., Nucl. Instrum. Meth. B 317, 263 (2013).
- V. Golovatyuk, V. Kekelidze, V. Kolesnikov, O. Rogachevsky, and A. Sorin, Eur. Phys. J. A 52, 212 (2016).
- S. A. Bass et al., Prog. Part. Nucl. Phys. 41, 255 (1998).
- M. Bleicher et al., J. Phys. G 25, 1859 (1999).
- W. Cassing and E. L. Bratkovskaya, Nucl. Phys. A 831, 215 (2009).
- Y. Nara, N. Otuka, A. Ohnishi, K. Niita, and S. Chiba, Phys. Rev. C 61, 024901 (1999).
- H. Petersen, J. Steinheimer, G. Burau, M. Bleicher, and H. Stöcker, Phys. Rev. C 78, 044901 (2008).
- M. O. Kuttan, A. Motornenko, J. Steinheimer, H. Stoecker, Y. Nara, and M. Bleicher, Eur. Phys. J. C 82, 427 (2022).
- J. Steinheimer, M. O. Kuttan, A. Motornenko, A. Sorensen, Y. Nara, V. Koch, M. Bleicher, and H. Stoecker, EPJ Web Conf. 276, 01021 (2023).
- J. Steinheimer, T. Reichert, Y. Nara, and M. Bleicher, J. Phys. G 52, 035103 (2025).
- D. Oliinychenko, A. Sorensen, V. Koch, and L. McLerran, Phys. Rev. C 108, 034908 (2023).
- J. Mohs, S. Spies, and H. Elfner, arXiv:2409.16927.
- Y. Nara and A. Ohnishi, Phys. Rev. C 105, 014911 (2022).
- J. Hofmann, H. Stoecker, U. W. Heinz, W. Scheid, and W. Greiner, Phys. Rev. Lett. 36, 88 (1976).
- H. Stöcker, Nucl. Phys. A 750, 121 (2005).
- M. A. Stephanov, K. Rajagopal, and E. V. Shuryak, Phys. Rev. Lett. 81, 4816 (1998).
- Y. Hatta and M. A. Stephanov, Phys. Rev. Lett. 91, 102003 (2003); [Erratum: 91, 129901 (2003)].
- T. D. Lee, A Possible New Form of Matter at High Density, Report of the Workshop on GeV/nucleon Collisions of Heavy Ions How and Why (Brookhaven National Lab., Upton, N.Y., USA, 1973), pp. 1–12.
- J. Hofmann, H. Stöcker, W. Scheid, and W. Greiner, On the Possibility of Nuclear Shock Waves in Relativistic Heavy-Ion Collisions, Report of the Workshop on GeV/nucleon Collisions of Heavy Ions How and Why (Brookhaven National Lab., Upton, N.Y., USA, 1973), pp. 39–47.
- J. E. Bernhard, J. S. Moreland, S. A. Bass, J. Liu, and U. Heinz, Phys. Rev. C 94, 024907 (2016).
- D. Everett et al. (JETSCAPE Collaboration), Phys. Rev. C 103, 054904 (2021).
- S. Pratt, E. Sangaline, P. Sorensen, and H. Wang, Phys. Rev. Lett. 114, 202301 (2015).
- G. Nijs, W. van der Schee, U. Gürsoy, and R. Snellings, Phys. Rev. C 103, 054909 (2021).
- M. O. Kuttan, J. Steinheimer, K. Zhou, and H. Stoecker, Phys. Rev. Lett. 131, 202303 (2023).
- S. Huth et al., Nature (London) 606, 276 (2022).
- Y.-L. Cheng, S. Shi, Y.-G. Ma, H. Stöcker, and K. Zhou, Phys. Rev. C 107, 064909 (2023).
- K. Zhou, L. Wang, L.-G. Pang, and S. Shi, Prog. Part. Nucl. Phys. 135, 104084 (2024).
- M. O. Kuttan, J. Steinheimer, K. Zhou, A. Redelbach, and H. Stoecker, Phys. Lett. B 811, 135872 (2020).
- M. O. Kuttan, K. Zhou, J. Steinheimer, A. Redelbach, and H. Stoecker, J. High Energy Phys. 10 (2020) 184.
- M. O. Kuttan, J. Steinheimer, K. Zhou, A. Redelbach, and H. Stoecker, Proc. Sci. FAIRness2022, 040 (2023).
- L.-G. Pang, K. Zhou, N. Su, H. Petersen, H. Stöcker, and X.-N. Wang, Nat. Commun. 9, 210 (2018).
- Y.-L. Du, K. Zhou, J. Steinheimer, L.-G. Pang, A. Motornenko, H.-S. Zong, X.-N. Wang, and H. Stöcker, Eur. Phys. J. C 80, 516 (2020).
- Y.-L. Du, D. Pablos, and K. Tywoniuk, J. High Energy Phys. 03 (2020) 206.
- W.-B. He, Y.-G. Ma, L.-G. Pang, H.-C. Song, and K. Zhou, Nucl. Sci. Tech. 34, 88 (2023).
- F. Li, Y. Wang, H. Lü, P. Li, Q. Li, and F. Liu, J. Phys. G 47, 115104 (2020).
- N. Mallick, S. Prasad, A. N. Mishra, R. Sahoo, and G. G. Barnaföldi, Phys. Rev. D 105, 114022 (2022).
- Y.-D. Song, R. Wang, Y.-G. Ma, X.-G. Deng, and H.-L. Liu, Phys. Lett. B 814, 136084 (2021).
- H. Hirvonen, K. J. Eskola, and H. Niemi, Phys. Rev. C 108, 034905 (2023).
- F. Sergeev, E. Bratkovskaya, I. Kisel, and I. Vassiliev, Int. J. Mod. Phys. A 35, 2043002 (2020).
- L. Jiang, L. Wang, and K. Zhou, Phys. Rev. D 103, 116023 (2021).
- P. Murali, S. Dash, and B. K. Nandi, arXiv:2411.11001.
- G. Aarts, K. Fukushima, T. Hatsuda, A. Ipp, S. Shi, L. Wang, and K. Zhou, Nat. Rev. Phys. 7, 154 (2025) .
- K. Zhou, G. Endrődi, L.-G. Pang, and H. Stöcker, Phys. Rev. D 100, 011501(R) (2019).
- K. Cranmer, G. Kanwar, S. Racanière, D. J. Rezende, and P. E. Shanahan, Nat. Rev. Phys. 5, 526 (2023).
- J.-A. Sun, L. Yan, C. Gale, and S. Jeon, arXiv:2410.13069.
- H. Huang, B. Xiao, Z. Liu, Z. Wu, Y. Mu, and H. Song, Phys. Rev. Res. 3, 023256 (2021).
- D. Torbunov, Y. Huang, M. Lin, Y. Ren, Y. Go, T. Rinn, H. Yu, B. Viren, and J. Huang, Phys. Rev. C 110, 034912 (2024).
- M. Omana Kuttan, K. Zhou, J. Steinheimer, and H. Stöcker, companion paper, Phys. Rev. C 112, L051902 (2025).
- L. V. Bravina et al., Phys. Rev. C 78, 014907 (2008).
- P. Achlioptas, O. Diamanti, I. Mitliagkas, and L. Guibas, in Proceedings of the 35th International Conference on Machine Learning, edited by J. Dy and A. Krause, Proceedings of Machine Learning Research (PMLR, 2018), Vol. 80, pp. 40–49.
- G. Yang, X. Huang, Z. Hao, M.-Y. Liu, S. Belongie, and B. Hariharan, in Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV) (IEEE, 2019).
- Y. Guo, H. Wang, Q. Hu, H. Liu, L. Liu, and M. Bennamoun, IEEE Trans. Pattern Anal. Mach. Intell. 43, 4338 (2020).
- S. Luo and W. Hu, in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (IEEE, New York, 2021), pp. 2837–2845.
- Y. Sun, Y. Wang, Z. Liu, J. Siegel, and S. Sarma, in Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (IEEE, New York, 2020), pp. 61–70.
- V. Mikuni and B. Nachman, Phys. Rev. D 106, 092009 (2022).
- O. Amram and K. Pedro, Phys. Rev. D 108, 072014 (2023).
- F. T. Acosta, V. Mikuni, B. Nachman, M. Arratia, B. Karki, R. Milton, P. Karande, and A. Angerami, J. Instrum. 19, P05003 (2024).
- V. Mikuni and B. Nachman, J. Instrum. 19, P02001 (2024).
- E. Buhmann, S. Diefenbacher, E. Eren, F. Gaede, G. Kasieczka, A. Korol, W. Korcari, K. Krüger, and P. McKeown, J. Instrum. 18, P11025 (2023).
- A. Butter, N. Huetsch, S. Palacios Schweitzer, T. Plehn, P. Sorrenson, and J. Spinner, SciPost Phys. Core 8, 026 (2025).
- M. Leigh, D. Sengupta, J. A. Raine, G. Quétant, and T. Golling, Phys. Rev. D 109, 012010 (2024).
- E. Buhmann, C. Ewen, D. A. Faroughy, T. Golling, G. Kasieczka, M. Leigh, G. Quétant, J. A. Raine, D. Sengupta, and D. Shih, arXiv:2310.00049.
- E. Buhmann, G. Kasieczka, and J. Thaler, SciPost Phys. 15, 130 (2023).
- V. Mikuni, B. Nachman, and M. Pettee, Phys. Rev. D 108, 036025 (2023).
- M. Leigh, D. Sengupta, G. Quétant, J. A. Raine, K. Zoch, and T. Golling, SciPost Phys. 16, 018 (2024).
- J. Y. Araz, V. Mikuni, F. Ringer, N. Sato, F. T. Acosta, and R. Whitehill, arXiv:2410.22421.
- P. Devlin, J.-W. Qiu, F. Ringer, and N. Sato, Phys. Rev. D 110, 016030 (2024).
- C. R. Qi, H. Su, K. Mo, and L. J. Guibas, in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (IEEE, New York, 2017), pp. 652–660.
- L. Dinh, J. Sohl-Dickstein, and S. Bengio, arXiv:1605.08803.
- J. Ho, A. Jain, and P. Abbeel, Advances in neural information processing systems 33, 6840 (2020).
- M. Omana Kuttan, J. Steinheimer, K. Zhou, A. Redelbach, and H. Stoecker, Particles 4, 47 (2021).
- S. Guo, H.-S. Wang, K. Zhou, and G.-L. Ma, Phys. Rev. C 110, 024910 (2024).
- D. P. Kingma and M. Welling, arXiv:1312.6114.
- G. Papamakarios, E. Nalisnick, D. J. Rezende, S. Mohamed, and B. Lakshminarayanan, J. Mach. Learn. Res. 22, 1 (2021).
- F.-A. Croitoru, V. Hondru, R. T. Ionescu, and M. Shah, IEEE Trans. Pattern Anal. Mach. Intell. 45, 10850 (2023).
- L. Yang, Z. Zhang, Y. Song, S. Hong, R. Xu, Y. Zhao, W. Zhang, B. Cui, and M.-H. Yang, ACM Comput. Surv. 55, 1 (2023).
- L. Wang, G. Aarts, and K. Zhou, arXiv:2311.03578.
- L. Wang, G. Aarts, and K. Zhou, J. High Energy Phys. 05 (2024) 060.
- C. Li, A. Kies, K. Zhou, M. Schlott, O. E. Sayed, M. Bilousova, and H. Stöcker, Appl. Energy 359, 122779 (2024).