An end-to-end generative diffusion model for heavy-ion collisions

Jing-An Sun,Li Yan,Charles Gale,Sangyong Jeon
2024-10-17
Abstract:We train a generative diffusion model (DM) to simulate ultra-relativistic heavy-ion collisions from end to end. The model takes initial entropy density profiles as input and produces two-dimensional final particle spectra, successfully reproducing integrated and differential observables. It also captures higher-order fluctuations and correlations. These findings suggest that the generative model has successfully learned the complex relationship between initial conditions and final particle spectra for various shear viscosities, as well as the fluctuations introduced during initial entropy production and hadronization stages, providing an efficient framework for resource-intensive physical goals. The code and trained model are available at <a class="link-external link-https" href="https://huggingface.co/Jing-An/DiffHIC/tree/main" rel="external noopener nofollow">this https URL</a>.
Nuclear Theory,High Energy Physics - Phenomenology,Nuclear Experiment
What problem does this paper attempt to address?