Machine Learning Approach to Analyze the Heavy Quark Diffusion Coefficient in Relativistic Heavy Ion Collisions

The diffusion coefficient of heavy quarks in a deconfined medium is examined in this research using a deep convolutional neural network (CNN) that is trained with data from relativistic heavy ion collisions involving heavy flavor hadrons. The CNN is trained using observables such as the nuclear modification factor RAA and the elliptic flow v2 of non-prompt J/ψ from the B-hadron decay in different centralities, where B meson evolutions are calculated using the Langevin equation and the instantaneous coalescence model. The CNN outputs the parameters, thereby characterizing the temperature and momentum dependence of the heavy quark diffusion coefficient. By inputting the experimental data of the non-prompt J/ψ(RAA,v2) from various collision centralities into multiple channels of a well-trained network, we derive the values of the diffusion coefficient parameters. Additionally, we evaluate the uncertainty in determining the diffusion coefficient by taking into account the uncertainties present in the experimental data (RAA,v2), which serve as inputs to the deep neural network.