To illustrate automatic differentiation with dynamic control flow, consider a contrived function where a while loop scales an input vector $$\mathbf{a}$$ until its norm exceeds a threshold, followed by an if statement that returns different multiples of the vector based on the sum of its elements. Even though the number of loop iterations and the final scaling factor depend entirely on the specific values of the input, the deep learning framework constructs a concrete computational graph during the forward pass. Because the resulting function $$f(\mathbf{a})$$ is linear with a piecewise defined scale, its gradient with respect to $$\mathbf{a}$$ can be logically verified to match the element-wise division $$f(\mathbf{a}) / \mathbf{a}$$.

Claude

One of the major benefits of automatic differentiation is its ability to compute gradients even when the sequence of operations depends on dynamic control flow, such as Python if statements, while loops, and arbitrary function calls. Since the exact path of computation is not known a priori, the computational graph is realized dynamically only during the forward evaluation on a specific input. Once this specific graph is constructed, deep learning frameworks can execute the backward pass to accurately calculate the gradient of the resulting variable.

Automatic Differentiation with Dynamic Control Flow

Dive into Deep Learning

Example: Differentiating a Function with Dynamic Control Flow

Dynamic control flow is pervasive in modern deep learning applications. For instance, in natural language processing, the sequence of operations and the resulting computational graph inherently depend on the varying lengths of input text sequences. Because it is impossible to compute the exact gradient formula for such models a priori, the ability of automatic differentiation to handle dynamically realized computational graphs is vital for training complex statistical models.

Learn Before

Related