Improving the solver for the Balitsky-Kovchegov evolution equation with Automatic Differentiation

TL;DR

This paper introduces a differentiable programming approach to solving the Balitsky-Kovchegov equation, enabling automatic calculation of derivatives to improve fitting procedures and relate TMD functions more efficiently.

Contribution

It presents a novel BK solver implementation using automatic differentiation, facilitating faster parameter fitting and derivative calculations for TMD analysis.

Findings

Automatic differentiation simplifies and accelerates the fitting process.

Derivatives of the amplitude with respect to parameters are automatically computed.

The solver includes the one-loop running coupling and kinematic constraints.

Abstract

The Balitsky-Kovchegov (BK) evolution equation is an equation derived from perturbative Quantum Chromodynamics that allows one to evolve with collision energy the scattering amplitude of a pair of quark and antiquark off a hadron target, called the dipole amplitude. The initial condition, being a non-perturbative object, usually has to be modeled separately. Typically, the model contains several tunable parameters that are determined by fitting to experimental data. In this contribution, we propose an implementation of the BK solver using differentiable programming. Automatic differentiation offers the possibility that the first and second derivatives of the amplitude with respect to the initial condition parameters are automatically calculated at all stages of the simulation. This fact should considerably facilitate and speed up the fitting step. Moreover, in the context of Transverse Momentum Distributions (TMD), we demonstrate that automatic differentiation can be used to obtain the first and second derivatives of the amplitude with respect to the quark-antiquark separation. These derivatives can be used to relate various TMD functions to the dipole amplitude. Our C++ code for the solver, which is available in a public repository, includes the Balitsky one-loop running coupling prescription and the kinematic constraint. This version of the BK equation is widely used in the small-$x$ evolution framework.