- Open Access
Deep learning for jet modification in the presence of the quark gluon plasma background
Phys. Rev. C 113, 024912 – Published 20 February, 2026
DOI: https://doi.org/10.1103/469v-zbdd
Abstract
Jet interactions with the color-deconfined quantum chromodynamics medium in relativistic heavy-ion collisions are conventionally assessed by measuring the modification of the distributions of jet observables with respect to their baselines in proton-proton collisions. Deep learning methods enable per-jet evaluation of these modifications, enhancing the use of jets as precision probes of the nuclear medium. In this work, we predict the jet-by-jet fractional energy loss for jets evolving through a quark-gluon plasma (QGP) medium using a linear Boltzmann transport model. To approximate realistic experimental conditions, we embed medium-modified jets in a thermal background and apply constituent subtraction for background removal. Two network architectures are studied: convolutional neural networks (CNNs) using jet images, and dynamic graph convolutional neural networks (DGCNNs) using particle clouds. We find that CNNs achieve accurate predictions for background-free jets but degrade in the presence of the QGP background and remain below the background-free baseline even after background subtraction. In contrast, DGCNNs applied to background-subtracted particle clouds maintain high accuracy across the entire range, demonstrating the advantage of point-cloud–based graph neural networks that exploit full jet structure under realistic conditions.
Physics Subject Headings (PhySH)
Article Text
References (93)
- M. Gyulassy and L. McLerran, Nucl. Phys. A 750, 30 (2005).
- P. Jacobs and X.-N. Wang, Prog. Part. Nucl. Phys. 54, 443 (2005).
- W. Busza, K. Rajagopal, and W. van der Schee, Annu. Rev. Nucl. Part. Sci. 68, 339 (2018).
- H. Elfner and B. Müller, J. Phys. G: Nucl. Part. Phys. 50, 103001 (2023).
- X.-N. Wang and M. Gyulassy, Phys. Rev. Lett. 68, 1480 (1992).
- G.-Y. Qin and X.-N. Wang, Int. J. Mod. Phys. E 24, 1530014 (2015).
- S. Cao and X.-N. Wang, Rep. Prog. Phys. 84, 024301 (2021).
- B. Abelev, et al. (The ALICE Collaboration), J. High Energy Phys. 03 (2014) 013.
- J. Adam, et al. (ALICE Collaboration), Phys. Lett. B 746, 1 (2015).
- G. Aad, et al. (ATLAS Collaboration), Phys. Rev. Lett. 114, 072302 (2015).
- M. Aaboud, et al. (The ATLAS Collaboration), Phys. Lett. B 774, 379 (2017).
- M. Aaboud Jr., et al. (The ATLAS Collaboration), Phys. Lett. B 790, 108 (2019).
- S. Acharya, et al. (ALICE Collaboration), Phys. Rev. C 101, 034911 (2020).
- L. Apolinário, Y.-T. Chien, and L. Cunqueiro Mendez, Int. J. Mod. Phys. E 33, 2430003 (2024).
- J. Casalderrey-Solana, D. Gulhan, G. Milhano, D. Pablos, and K. Rajagopal, J. High Energy Phys. 03 (2017) 135.
- Y. Tachibana, N.-B. Chang, and G.-Y. Qin, Phys. Rev. C 95, 044909 (2017).
- R. Kunnawalkam Elayavalli and K. C. Zapp, J. High Energy Phys. 07 (2017) 141.
- G. Milhano, U. A. Wiedemann, and K. C. Zapp, Phys. Lett. B 779, 409 (2018).
- W. Chen, S. Cao, T. Luo, L.-G. Pang, and X.-N. Wang, Phys. Lett. B 810, 135783 (2020).
- C. Park, S. Jeon, and C. Gale, Nucl. Phys. A 982, 643 (2019).
- T. Luo, S. Cao, Y. He, and X.-N. Wang, Phys. Lett. B 782, 707 (2018).
- J. Casalderrey-Solana, G. Milhano, D. Pablos, and K. Rajagopal, J. High Energy Phys. 01 (2020) 044.
- N.-B. Chang, Y. Tachibana, and G.-Y. Qin, Phys. Lett. B 801, 135181 (2020).
- Y. Tachibana, C. Shen, and A. Majumder, Phys. Rev. C 106, L021902 (2022).
- Z. Yang, Y. He, I. Moult, and X.-N. Wang, Phys. Rev. Lett. 132, 011901 (2024).
- W.-J. Xing, S. Cao, G.-Y. Qin, and X.-N. Wang, Phys. Rev. Lett. 134, 052301 (2025).
- J. Brewer, J. G. Milhano, and J. Thaler, Phys. Rev. Lett. 122, 222301 (2019).
- Y.-L. Du, D. Pablos, and K. Tywoniuk, J. High Energy Phys. 03 (2021) 206.
- Y.-L. Du, D. Pablos, and K. Tywoniuk, Phys. Rev. Lett. 128, 012301 (2022).
- A. Takacs and K. Tywoniuk, J. High Energy Phys. 10 (2021) 038.
- J. Brewer, Q. Brodsky, and K. Rajagopal, J. High Energy Phys. 02 (2022) 175.
- C. Andres, J. Holguin, R. Kunnawalkam Elayavalli, and J. Viinikainen, Phys. Rev. Lett. 134, 082303 (2025).
- Y.-T. Chien and R. Kunnawalkam Elayavalli, arXiv:1803.03589.
- Y.-L. Du, D. Pablos, and K. Tywoniuk, PoS PANIC2021, 224 (2022).
- L. Apolinário, N. F. Castro, M. Crispim Romão, J. G. Milhano, R. Pedro, and F. C. R. Peres, J. High Energy Phys. 11 (2021) 219.
- L. Liu, J. Velkovska, Y. Wu, and M. Verweij, J. High Energy Phys. 04 (2023) 140.
- Y. S. Lai, J. Mulligan, M. Płoskoń, and F. Ringer, J. High Energy Phys. 10 (2022) 011.
- M. Crispim Romão, J. G. Milhano, and M. van Leeuwen, SciPost Phys. 16, 015 (2024).
- U. S. Qureshi and R. K. Elayavalli, arXiv:2411.19389.
- Z. Yang, Y. He, W. Chen, W.-Y. Ke, L.-G. Pang, and X.-N. Wang, Eur. Phys. J. C 83, 652 (2023).
- R. Haake and C. Loizides, Phys. Rev. C 99, 064904 (2019).
- S. Acharya, et al. (ALICE Collaboration), Phys. Lett. B 849, 138412 (2024).
- R. Li, Y.-L. Du, and S. Cao, Phys. Lett. B 870, 139940 (2025).
- D. Stewart and J. Putschke, Phys. Rev. C 111, 054902 (2025).
- U. S. Qureshi and R. K. Elayavalli, Phys. Rev. C 112, 064906 (2025).
- P. Baldi, K. Bauer, C. Eng, P. Sadowski, and D. Whiteson, Phys. Rev. D 93, 094034 (2016).
- L. de Oliveira, M. Kagan, L. Mackey, B. Nachman, and A. Schwartzman, J. High Energy Phys. 07 (2016) 069.
- P. T. Komiske, E. M. Metodiev, and M. D. Schwartz, J. High Energy Phys. 01 (2017) 110.
- G. Kasieczka, T. Plehn, M. Russell, and T. Schell, J. High Energy Phys. 05 (2017) 006.
- D. Guest, J. Collado, P. Baldi, S.-C. Hsu, G. Urban, and D. Whiteson, Phys. Rev. D 94, 112002 (2016).
- F. A. Dreyer, G. P. Salam, and G. Soyez, J. High Energy Phys. 12 (2018) 064.
- G. Louppe, K. Cho, C. Becot, and K. Cranmer, J. High Energy Phys. 01 (2019) 057.
- T. Cheng, Comput. Softw. Big Sci. 2, 3 (2018).
- P. T. Komiske, E. M. Metodiev, and J. Thaler, J. High Energy Phys. 01 (2019) 121.
- M. J. Dolan and A. Ore, Phys. Rev. D 103, 074022 (2021).
- H. Qu and L. Gouskos, Phys. Rev. D 101, 056019 (2020).
- J. Guo, J. Li, T. Li, and R. Zhang, Phys. Rev. D 103, 116025 (2021).
- X. Ju and B. Nachman, Phys. Rev. D 102, 075014 (2020).
- S. Gong, Q. Meng, J. Zhang, H. Qu, C. Li, S. Qian, W. Du, Z.-M. Ma, and T.-Y. Liu, J. High Energy Phys. 07 (2022) 030.
- J. Casalderrey-Solana, D. C. Gulhan, J. G. Milhano, D. Pablos, and K. Rajagopal, J. High Energy Phys. 09 (2015) 175.
- S. Cao, T. Luo, G.-Y. Qin, and X.-N. Wang, Phys. Rev. C 94, 014909 (2016).
- T. Luo, Y. He, S. Cao, and X.-N. Wang, Phys. Rev. C 109, 034919 (2024).
- Y. Wang, Y. Sun, Z. Liu, S. E. Sarma, M. M. Bronstein, and J. M. Solomon, ACM Trans. Graphics 38, 1 (2019).
- T. Sjöstrand, S. Mrenna, and P. Z. Skands, J. High Energy Phys. 05 (2006) 026.
- T. Sjöstrand, S. Ask, J. R. Christiansen, R. Corke, N. Desai, P. Ilten, S. Mrenna, S. Prestel, C. O. Rasmussen, and P. Z. Skands, Comput. Phys. Commun. 191, 159 (2015).
- JetToyHI: Framework to study jet substructure in heavy ion environment, https://github.com/mverwe/JetToyHI.
- P. Skands, S. Carrazza, and J. Rojo, Eur. Phys. J. C 74, 3024 (2014).
- M. Cacciari, G. P. Salam, and G. Soyez, Eur. Phys. J. C 72, 1896 (2012).
- Y. He, T. Luo, X.-N. Wang, and Y. Zhu, Phys. Rev. C 91, 054908 (2015); 97, 019902(E) (2018).
- X.-F. Guo and X.-N. Wang, Phys. Rev. Lett. 85, 3591 (2000).
- X.-N. Wang and X.-F. Guo, Nucl. Phys. A 696, 788 (2001).
- B.-W. Zhang, E. Wang, and X.-N. Wang, Phys. Rev. Lett. 93, 072301 (2004).
- A. Majumder, Phys. Rev. D 85, 014023 (2012).
- L.-G. Pang, H. Petersen, and X.-N. Wang, Phys. Rev. C 97, 064918 (2018).
- X.-Y. Wu, G.-Y. Qin, L.-G. Pang, and X.-N. Wang, Phys. Rev. C 105, 034909 (2022).
- M. Zhang, Y. He, S. Cao, and L. Yi, Chin. Phys. C 47, 024106 (2023).
- S. Acharya, et al. (ALICE Collaboration), Phys. Rev. C 101, 044907 (2020).
- M. Cacciari, G. P. Salam, and G. Soyez, J. High Energy Phys. 04 (2008) 063.
- P. Berta, L. Masetti, D. W. Miller, and M. Spousta, J. High Energy Phys. 08 (2019) 175.
- M. Cacciari, G. P. Salam, and G. Soyez, J. High Energy Phys. 04 (2008) 005.
- S. Catani, Y. L. Dokshitzer, M. H. Seymour, and B. R. Webber, Nucl. Phys. B 406, 187 (1993).
- S. D. Ellis and D. E. Soper, Phys. Rev. D 48, 3160 (1993).
- 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).
- K. He, X. Zhang, S. Ren, and J. Sun, in Proceedings of the IEEE International Conference on Computer Vision (IEEE, Santiago, Chile, 2015), pp. 1026–1034.
- A. Y. Ng, in Proceedings of the Twenty-First International Conference on Machine Learning (ACM, New York, 2004), p. 78.
- S. Ioffe and C. Szegedy, in Proceedings of the 32nd International Conference on Machine Learning (PMLR, Lille, 2015), pp. 448–456.
- D. P. Kingma and J. Ba, in 3rd International Conference for Learning Representations (ICLR, San Diego, 2015).
- E. Fix and J. L. Hodges, Jr., Discriminatory analysis-nonparametric discrimination: Small sample performance, Technical Report, USAF, Project No. 21-49-004, Report No. 11 (1952).
- T. Cover and P. Hart, IEEE Trans. Inf. Theory 13, 21 (1967).
- X. Glorot, A. Bordes, and Y. Bengio, in Proceedings of the Fourteenth International Conference on Artificial Intelligence and Statistics, JMLR Workshop and Conference Proceedings (JMLR, Fort Lauderdale, 2011), pp. 315–323.
- I. Loshchilov and F. Hutter, in Procedings of Seventh International Conference on Learning Representations (ICLR, Vancouver, 2019).
- A. J. Larkoski, S. Marzani, G. Soyez, and J. Thaler, J. High Energy Phys. 05 (2014) 146.