Machine learning approach to analyze heavy quark diffusion coefficient in relativistic heavy-ion collisions

The diffusion coefficient of heavy quarks in the deconfined medium is examined in this research using a deep convolutional neural network (CNN) trained with data from relativistic heavy ion collisions involving heavy flavor hadrons. The CNN is trained using observables such as the nuclear modification factor $R_{AA}$ and the elliptic flow $v_2$ of non-prompt $J/\psi$ from 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 characterizing the temperature and momentum dependence of the heavy quark diffusion coefficient. By inputting the experimental data of non-prompt $J/\psi$ $(R_{AA}, v_2)$ from various collision centralities into multiple channels of the 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 $(R_{AA}, v_2)$, which serve as inputs to the deep neural network.