- Open Access
Machine learning driven identification of heavy flavor decay leptons in proton-proton collisions at the Large Hadron Collider
Phys. Rev. D 113, 094025 – Published 20 May, 2026
DOI: https://doi.org/10.1103/qpmc-gtmt
Abstract
The study of heavy-flavor hadrons is topical in the era of precision measurements, which is useful to test theories based on pQCD. The heavy-flavor hadrons are produced initially during heavy-ion or hadronic collisions and are one of the best probes to understand the initial stages of the collisions as well as the system evolution. In experiments, the heavy-flavor sectors are studied directly via their decay to different hadrons or di-leptons or via their semileptonic decay, which is accompanied by additional neutrinos. However, their measurement in experiments is resource intensive and requires input from different Monte Carlo event generators. In this study, we provide an independent method based on machine learning algorithms to separate such leptons coming from heavy-flavor semileptonic decays. We use pythia8 to generate events for this study, which gives a good qualitative and quantitative description of heavy-flavor production in collisions. We use the xgboost model for this study, which is trained with collisions at . We use , , and pseudorapidity as the input features to the machine. The ML model provides an accuracy of 98% for heavy-flavor decay electrons and almost 100% for heavy-flavor decay muons.
Physics Subject Headings (PhySH)
Article Text
References (46)
- S. Acharya et al. (ALICE Collaboration), Phys. Lett. B 810, 135758 (2020).
- J. Adam et al. (STAR Collaboration), Phys. Lett. B 786 (2018).
- D. Adamová et al. (ALICE Collaboration), Phys. Lett. B 776, 91 (2018).
- S. Acharya et al. (ALICE Collaboration), J. High Energy Phys. 09 (2020) 162.
- J. Adam et al. (ALICE Collaboration), J. High Energy Phys. 09 (2015) 148.
- S. Chatrchyan et al. (CMS Collaboration), J. High Energy Phys. 04 (2014) 103.
- E. Norrbin and T. Sjostrand, Eur. Phys. J. C 17, 137 (2000).
- S. Acharya et al. (ALICE Collaboration), Eur. Phys. J. C 82, 335 (2022).
- S. Acharya et al. (ALICE Collaboration), Eur. Phys. J. C 80, 979 (2020).
- S. Navas et al. (Particle Data Group), Phys. Rev. D 110, 030001 (2024).
- S. Acharya et al. (ALICE Collaboration), Phys. Rev. Lett. 126, 162001 (2021).
- L. Adamczyk et al. (STAR Collaboration), Phys. Rev. C 95, 034907 (2017).
- J. Adam et al. (ALICE Collaboration), J. High Energy Phys. 09 (2016) 028.
- S. Acharya et al. (ALICE Collaboration), J. High Energy Phys. 08 (2023) 006.
- S. Acharya et al. (ALICE Collaboration), J. High Energy Phys. 09 (2019) 008.
- S. Acharya et al. (ALICE Collaboration), Phys. Lett. B 804, 135377 (2020).
- J. Adam et al. (ALICE Collaboration), Phys. Lett. B 771, 467 (2017).
- D. Bowser-Chao and D. L. Dzialo, Phys. Rev. D 47 (1993).
- P. Chiappetta, P. Colangelo, P. De Felice, G. Nardulli, and G. Pasquariello, Phys. Lett. B 322 (1994).
- S. Acharya et al. (ALICE Collaboration), Phys. Lett. B 849, 138412 (2024).
- G. Aad et al. (ATLAS Collaboration), Phys. Rev. Lett. 125, 131801 (2020).
- H. Hassan, N. Mallick, and D. J. Kim, arXiv:2504.18291.
- M. Draguet (ATLAS Collaboration), Proc. Sci. ICHEP2024 (2025) 1002.
- A. M. Sirunyan et al. (CMS Collaboration), J. Instrum. 15, P06005 (2020).
- Ł. K. Graczykowski, M. Jakubowska, K. Rafał Deja, and M. Kabus (ALICE Collaboration), J. Instrum. 17, C07016 (2022).
- N. Tamir, I. Bessudo, B. Chen, H. Raiko, and L. Barak, J. Instrum. 19, P07004 (2024).
- D. Basak, H. Hushnud, and K. Dey, J. Phys. G 53, 015102 (2026).
- A. Ortiz, A. Paz, J. D. Romo, S. Tripathy, E. A. Zepeda, and I. Bautista, Phys. Rev. D 102, 076014 (2020).
- S. Prasad, S. Tripathy, B. Sahoo, and R. Sahoo, Phys. Rept. 1181, 1 (2026).
- N. Mallick, S. Tripathy, A. N. Mishra, S. Deb, and R. Sahoo, Phys. Rev. D 103, 094031 (2021).
- S. A. Bass, A. Bischoff, J. A. Maruhn, H. Stoecker, and W. Greiner, Phys. Rev. C 53 (1996).
- X. Zhang, Y. Huang, W. Lin, X. Liu, H. Zheng, R. Wada, A. Bonasera, Z. Chen, L. Chen, J. Han et al., Phys. Rev. C 105, 034611 (2022).
- N. Mallick, S. Prasad, A. N. Mishra, R. Sahoo, and G. G. Barnaföldi, Phys. Rev. D 105, 114022 (2022).
- N. Mallick, S. Prasad, A. N. Mishra, R. Sahoo, and G. G. Barnaföldi, Phys. Rev. D 107, 094001 (2023).
- H. Hirvonen, K. J. Eskola, and H. Niemi, Phys. Rev. C 108, 034905 (2023).
- K. Singh, K. Goswami, R. Sahoo, and S. Samal, Phys. Rev. Accel. Beams 28, 043101 (2025).
- S. Prasad, N. Mallick, and R. Sahoo, Phys. Rev. D 109, 014005 (2024).
- K. Goswami, S. Prasad, N. Mallick, R. Sahoo, and G. B. Mohanty, Phys. Rev. D 110, 034017 (2024).
- N. Mallick, H. Hassan, and D. J. Kim, arXiv:2504.18279.
- T. Sjostrand, S. Mrenna, and P. Z. Skands, Comput. Phys. Commun. 178, 852 (2008).
- C. Bierlich, S. Chakraborty, N. Desai, L. Gellersen et al., SciPost Phys. Codebases 2022, 8 (2022).
- T. Chen et al., in 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (Association for Computing Machinery, New York, 2016), p. 785.
- Available online: https://xgboost.readthedocs.io/en/stable/.
- J. Adam et al. (STAR Collaboration), Phys. Rev. C 99, 034908 (2019).
- S. Acharya et al. (ALICE Collaboration), J. High Energy Phys. 03 (2022) 190.
- S. Acharya et al. (ALICE Collaboration), Eur. Phys. J. C 83, 741 (2023).