# Automatic differentiation

Jump to: navigation, search

In mathematics and computer algebra, automatic differentiation (AD), also called algorithmic differentiation or computational differentiation,[1][2] is a set of techniques to numerically evaluate the derivative of a function specified by a computer program. AD exploits the fact that every computer program, no matter how complicated, executes a sequence of elementary arithmetic operations (addition, subtraction, multiplication, division, etc.) and elementary functions (exp, log, sin, cos, etc.). By applying the chain rule repeatedly to these operations, derivatives of arbitrary order can be computed automatically, accurately to working precision, and using at most a small constant factor more arithmetic operations than the original program.

Automatic differentiation is not:

Figure 1: How automatic differentiation relates to symbolic differentiation

These classical methods run into problems: symbolic differentiation leads to inefficient code (unless carefully done) and faces the difficulty of converting a computer program into a single expression, while numerical differentiation can introduce round-off errors in the discretization process and cancellation. Both classical methods have problems with calculating higher derivatives, where the complexity and errors increase. Finally, both classical methods are slow at computing the partial derivatives of a function with respect to many inputs, as is needed for gradient-based optimization algorithms. Automatic differentiation solves all of these problems, at the expense of introducing more software dependencies.

## The chain rule, forward and reverse accumulation

Fundamental to AD is the decomposition of differentials provided by the chain rule. For the simple composition y = g(h(x)) = g(w) the chain rule gives

${\displaystyle {\frac {dy}{dx}}={\frac {dy}{dw}}{\frac {dw}{dx}}}$

Usually, two distinct modes of AD are presented, forward accumulation (or forward mode) and reverse accumulation (or reverse mode). Forward accumulation specifies that one traverses the chain rule from inside to outside (that is, first one computes dw/dx and then dy/dw, while reverse accumulation has the traversal from outside to inside.

Generally, both forward and reverse accumulation are specific manifestations of applying the operator of program composition, with the appropriate one of the two mappings ${\displaystyle (w,y)}$ being fixed.

### Forward accumulation

Figure 2: Example of forward accumulation with computational graph

In forward accumulation AD, one first fixes the independent variable to which differentiation is performed and computes the derivative of each sub-expression recursively. In a pen-and-paper calculation, one can do so by repeatedly substituting the derivative of the inner functions in the chain rule:

${\displaystyle {\frac {\partial y}{\partial x}}={\frac {\partial y}{\partial w_{1}}}{\frac {\partial w_{1}}{\partial x}}={\frac {\partial y}{\partial w_{1}}}\left({\frac {\partial w_{1}}{\partial w_{2}}}{\frac {\partial w_{2}}{\partial x}}\right)={\frac {\partial y}{\partial w_{1}}}\left({\frac {\partial w_{1}}{\partial w_{2}}}\left({\frac {\partial w_{2}}{\partial w_{3}}}{\frac {\partial w_{3}}{\partial x}}\right)\right)=\cdots }$

This can be generalized to multiple variables as a matrix product of Jacobians.

Compared to reverse accumulation, forward accumulation is very natural and easy to implement as the flow of derivative information coincides with the order of evaluation. One simply augments each variable w with its derivative (stored as a numerical value, not a symbolic expression),

${\displaystyle {\dot {w}}={\frac {\partial w}{\partial x}}}$

as denoted by the dot. The derivatives are then computed in sync with the evaluation steps and combined with other derivatives via the chain rule.

As an example, consider the function:

{\displaystyle {\begin{aligned}z&=f(x_{1},x_{2})\\&=x_{1}x_{2}+\sin x_{1}\\&=w_{1}w_{2}+\sin w_{1}\\&=w_{3}+w_{4}\\&=w_{5}\end{aligned}}}

For clarity, the individual sub-expressions have been labeled with the variables wi.

The choice of the independent variable to which differentiation is performed affects the seed values 1 and 2. Suppose one is interested in the derivative of this function with respect to x1. In this case, the seed values should be set to:

{\displaystyle {\begin{aligned}{\dot {w}}_{1}={\frac {\partial x_{1}}{\partial x_{1}}}=1\\{\dot {w}}_{2}={\frac {\partial x_{2}}{\partial x_{1}}}=0\end{aligned}}}

With the seed values set, one may then propagate the values using the chain rule as shown in both the table below. Figure 2 shows a pictorial depiction of this process as a computational graph.

${\displaystyle {\begin{array}{l|l}{\text{Operations to compute value}}&{\text{Operations to compute derivative}}\\\hline w_{1}=x_{1}&{\dot {w}}_{1}=1{\text{ (seed)}}\\w_{2}=x_{2}&{\dot {w}}_{2}=0{\text{ (seed)}}\\w_{3}=w_{1}\cdot w_{2}&{\dot {w}}_{3}=w_{2}\cdot {\dot {w}}_{1}+w_{1}\cdot {\dot {w}}_{2}\\w_{4}=\sin w_{1}&{\dot {w}}_{4}=\cos w_{1}\cdot {\dot {w}}_{1}\\w_{5}=w_{3}+w_{4}&{\dot {w}}_{5}={\dot {w}}_{3}+{\dot {w}}_{4}\end{array}}}$

To compute the gradient of this example function, which requires the derivatives of f with respect to not only x1 but also x2, one must perform an additional sweep over the computational graph using the seed values ${\displaystyle {\dot {w}}_{1}=0;{\dot {w}}_{2}=1}$.

The computational complexity of one sweep of forward accumulation is proportional to the complexity of the original code.

Forward accumulation is more efficient than reverse accumulation for functions f : ℝn → ℝm with mn as only n sweeps are necessary, compared to m sweeps for reverse accumulation.

### Reverse accumulation

Figure 3: Example of reverse accumulation with computational graph

In reverse accumulation AD, one first fixes the dependent variable to be differentiated and computes the derivative with respect to each sub-expression recursively. In a pen-and-paper calculation, one can perform the equivalent by repeatedly substituting the derivative of the outer functions in the chain rule:

${\displaystyle {\frac {\partial y}{\partial x}}={\frac {\partial y}{\partial w_{1}}}{\frac {\partial w_{1}}{\partial x}}=\left({\frac {\partial y}{\partial w_{2}}}{\frac {\partial w_{2}}{\partial w_{1}}}\right){\frac {\partial w_{1}}{\partial x}}=\left(\left({\frac {\partial y}{\partial w_{3}}}{\frac {\partial w_{3}}{\partial w_{2}}}\right){\frac {\partial w_{2}}{\partial w_{1}}}\right){\frac {\partial w_{1}}{\partial x}}=\cdots }$

In reverse accumulation, the quantity of interest is the adjoint, denoted with a bar (); it is a derivative of a chosen dependent variable with respect to a subexpression w:

${\displaystyle {\bar {w}}={\frac {\partial y}{\partial w}}}$

Reverse accumulation traverses the chain rule from outside to inside, or in the case of the computational graph in Figure 3, from top to bottom. The example function is real-valued, and thus there is only one seed for the derivative computation, and only one sweep of the computational graph is needed in order to calculate the (two-component) gradient. This is only half the work when compared to forward accumulation, but reverse accumulation requires the storage of the intermediate variables wi as well as the instructions that produced them in a data structure known as a Wengert list (or "tape"),[3][4] which may represent a significant memory issue if the computational graph is large. This can be mitigated to some extent by storing only a subset of the intermediate variables and then reconstructing the necessary work variables by repeating the evaluations, a technique known as checkpointing.

The operations to compute the derivative using reverse accumulation are shown in the table below (note the reversed order):

${\displaystyle {\begin{array}{l}{\text{Operations to compute derivative}}\\\hline {\bar {w}}_{5}=1{\text{ (seed)}}\\{\bar {w}}_{4}={\bar {w}}_{5}\\{\bar {w}}_{3}={\bar {w}}_{5}\\{\bar {w}}_{2}={\bar {w}}_{3}\cdot w_{1}\\{\bar {w}}_{1}={\bar {w}}_{3}\cdot w_{2}+{\bar {w}}_{4}\cdot \cos w_{1}\end{array}}}$

The data flow graph of a computation can be manipulated to calculate the gradient of its original calculation. This is done by adding an adjoint node for each primal node, connected by adjoint edges which parallel the primal edges but flow in the opposite direction. The nodes in the adjoint graph represent multiplication by the derivatives of the functions calculated by the nodes in the primal. For instance, addition in the primal causes fanout in the adjoint; fanout in the primal causes addition in the adjoint; a unary function y = f(x) in the primal causes = ȳ f′(x) in the adjoint; etc.

Reverse accumulation is more efficient than forward accumulation for functions f : ℝn → ℝm with mn as only m sweeps are necessary, compared to n sweeps for forward accumulation.

Reverse mode AD was first published in 1970 by Seppo Linnainmaa in his master thesis.[5][6][7]

Backpropagation of errors in multilayer perceptrons, a technique used in machine learning, is a special case of reverse mode AD.

### Beyond forward and reverse accumulation

Forward and reverse accumulation are just two (extreme) ways of traversing the chain rule. The problem of computing a full Jacobian of f : ℝn → ℝm with a minimum number of arithmetic operations is known as the optimal Jacobian accumulation (OJA) problem, which is NP-complete.[8] Central to this proof is the idea that there may exist algebraic dependencies between the local partials that label the edges of the graph. In particular, two or more edge labels may be recognized as equal. The complexity of the problem is still open if it is assumed that all edge labels are unique and algebraically independent.

## Automatic differentiation using dual numbers

Forward mode automatic differentiation is accomplished by augmenting the algebra of real numbers and obtaining a new arithmetic. An additional component is added to every number which will represent the derivative of a function at the number, and all arithmetic operators are extended for the augmented algebra. The augmented algebra is the algebra of dual numbers. This approach was generalized by the theory of operational calculus on programming spaces (see Analytic programming space), through tensor algebra of the dual space.

Replace every number ${\displaystyle \,x}$ with the number ${\displaystyle x+x'\varepsilon }$, where ${\displaystyle x'}$ is a real number, but ${\displaystyle \varepsilon }$ is an abstract number with the property ${\displaystyle \varepsilon ^{2}=0}$ (an infinitesimal; see Smooth infinitesimal analysis). Using only this, we get for the regular arithmetic

{\displaystyle {\begin{aligned}(x+x'\varepsilon )+(y+y'\varepsilon )&=x+y+(x'+y')\varepsilon \\(x+x'\varepsilon )\cdot (y+y'\varepsilon )&=xy+xy'\varepsilon +yx'\varepsilon +x'y'\varepsilon ^{2}=xy+(xy'+yx')\varepsilon \end{aligned}}}

and likewise for subtraction and division.

Now, we may calculate polynomials in this augmented arithmetic. If ${\displaystyle P(x)=p_{0}+p_{1}x+p_{2}x^{2}+\cdots +p_{n}x^{n}}$, then

{\displaystyle {\begin{aligned}P(x+x'\varepsilon )&=p_{0}+p_{1}(x+x'\varepsilon )+\cdots +p_{n}(x+x'\varepsilon )^{n}\\&=p_{0}+p_{1}x+\cdots +p_{n}x^{n}+p_{1}x'\varepsilon +2p_{2}xx'\varepsilon +\cdots +np_{n}x^{n-1}x'\varepsilon \\&=P(x)+P^{(1)}(x)x'\varepsilon \end{aligned}}}

where ${\displaystyle P^{(1)}}$ denotes the derivative of ${\displaystyle P}$ with respect to its first argument, and ${\displaystyle x'}$, called a seed, can be chosen arbitrarily.

The new arithmetic consists of ordered pairs, elements written ${\displaystyle \langle x,x'\rangle }$, with ordinary arithmetics on the first component, and first order differentiation arithmetic on the second component, as described above. Extending the above results on polynomials to analytic functions we obtain a list of the basic arithmetic and some standard functions for the new arithmetic:

{\displaystyle {\begin{aligned}\left\langle u,u'\right\rangle +\left\langle v,v'\right\rangle &=\left\langle u+v,u'+v'\right\rangle \\\left\langle u,u'\right\rangle -\left\langle v,v'\right\rangle &=\left\langle u-v,u'-v'\right\rangle \\\left\langle u,u'\right\rangle *\left\langle v,v'\right\rangle &=\left\langle uv,u'v+uv'\right\rangle \\\left\langle u,u'\right\rangle /\left\langle v,v'\right\rangle &=\left\langle {\frac {u}{v}},{\frac {u'v-uv'}{v^{2}}}\right\rangle \quad (v\neq 0)\\\sin \left\langle u,u'\right\rangle &=\left\langle \sin(u),u'\cos(u)\right\rangle \\\cos \left\langle u,u'\right\rangle &=\left\langle \cos(u),-u'\sin(u)\right\rangle \\\exp \left\langle u,u'\right\rangle &=\left\langle \exp u,u'\exp u\right\rangle \\\log \left\langle u,u'\right\rangle &=\left\langle \log(u),u'/u\right\rangle \quad (u>0)\\\left\langle u,u'\right\rangle ^{k}&=\left\langle u^{k},ku^{k-1}u'\right\rangle \quad (u\neq 0)\\\left|\left\langle u,u'\right\rangle \right|&=\left\langle \left|u\right|,u'{\mbox{sign}}u\right\rangle \quad (u\neq 0)\end{aligned}}}

and in general for the primitive function ${\displaystyle g}$,

${\displaystyle g(\langle u,u'\rangle ,\langle v,v'\rangle )=\langle g(u,v),g_{u}(u,v)u'+g_{v}(u,v)v'\rangle }$

where ${\displaystyle g_{u}}$ and ${\displaystyle g_{v}}$ are the derivatives of ${\displaystyle g}$ with respect to its first and second arguments, respectively.

When a binary basic arithmetic operation is applied to mixed arguments—the pair ${\displaystyle \langle u,u'\rangle }$ and the real number ${\displaystyle c}$—the real number is first lifted to ${\displaystyle \langle c,0\rangle }$. The derivative of a function ${\displaystyle f:\mathbb {R} \rightarrow \mathbb {R} }$ at the point ${\displaystyle x_{0}}$ is now found by calculating ${\displaystyle f(\langle x_{0},1\rangle )}$ using the above arithmetic, which gives ${\displaystyle \langle f(x_{0}),f'(x_{0})\rangle }$ as the result.

### Vector arguments and functions

Multivariate functions can be handled with the same efficiency and mechanisms as univariate functions by adopting a directional derivative operator. That is, if it is sufficient to compute ${\displaystyle y'=\nabla f(x)\cdot x'}$, the directional derivative ${\displaystyle y'\in \mathbb {R} ^{m}}$ of ${\displaystyle f:\mathbb {R} ^{n}\rightarrow \mathbb {R} ^{m}}$ at ${\displaystyle x\in \mathbb {R} ^{n}}$ in the direction ${\displaystyle x'\in \mathbb {R} ^{n}}$, this may be calculated as ${\displaystyle (\langle y_{1},y'_{1}\rangle ,\ldots ,\langle y_{m},y'_{m}\rangle )=f(\langle x_{1},x'_{1}\rangle ,\ldots ,\langle x_{n},x'_{n}\rangle )}$ using the same arithmetic as above. If all the elements of ${\displaystyle \nabla f}$ are desired, then ${\displaystyle n}$ function evaluations are required. Note that in many optimization applications, the directional derivative is indeed sufficient.

### High order and many variables

The above arithmetic can be generalized to calculate second order and higher derivatives of multivariate functions. However, the arithmetic rules quickly grow very complicated: complexity will be quadratic in the highest derivative degree. Instead, truncated Taylor polynomial algebra can be used. The resulting arithmetic, defined on generalized dual numbers, allows to efficiently compute using functions as if they were a new data type. Once the Taylor polynomial of a function is known, the derivatives are easily extracted. A rigorous, general formulation is achieved through the tensor series expansion using operational calculus on programming spaces. Currently there are a few software projects that implement the Taylor polynomial truncated algebra. AuDi, CTaylor and COSY infinity. Only the first two are open source.

## Operational calculus on programming spaces

Operational calculus on programming spaces [9] generalizes concepts of automatic differentiation and provides a framework that enables analytic conclusions about programs through algebraic means. This formulation using Tensor algebra is a broad generalization of the dual numbers approach. Theory allows computations on the operator level, before the operator is applied to a particular program, favoring efficient implementation.

Operational calculus on programming spaces [9] was first proposed by Žiga Sajovic, mentored by Martin Vuk, to unify existing approaches under a single theory and provide a rigorous framework for program analysis through calculus.

### Analytic programming space

A differentiable programming space ${\displaystyle {\mathcal {P}}_{0}}$ is any subspace of ${\displaystyle {\mathcal {F}}_{0}:{\mathcal {V}}\to {\mathcal {V}}}$ such that

${\displaystyle \partial {\mathcal {P}}_{0}\subset {\mathcal {P}}_{0}\otimes T({\mathcal {V}}^{*})}$

When all elements of ${\displaystyle {\mathcal {P}}_{0}}$ are analytic, we call ${\displaystyle {\mathcal {P}}_{0}}$ an analytic programming space.

Theorem.
Any differentiable programming space ${\displaystyle {\mathcal {P}}_{0}}$ is an infinitely differentiable programming space, meaning that
${\displaystyle \partial ^{k}{\mathcal {P}}_{0}\subset {\mathcal {P}}_{0}\otimes T({\mathcal {V}}^{*})}$
for any ${\displaystyle k\in \mathbb {N} }$. If all elements of ${\displaystyle {\mathcal {P}}_{0}}$ are analytic, than so are the elements of ${\displaystyle {\mathcal {P}}_{n}}$.
Definition.
Let ${\displaystyle {\mathcal {P}}_{0}}$ be a differentiable programming space. The space ${\displaystyle {\mathcal {P}}_{n}<{\mathcal {F}}_{n}:{\mathcal {V}}\to {\mathcal {V}}\otimes T({\mathcal {V}}^{*})}$ spanned by ${\displaystyle {\mathcal {D}}^{n}{\mathcal {P}}_{0}}$ over ${\displaystyle K}$, where ${\displaystyle {\mathcal {D}}^{n}=\{\partial ^{k};\quad 0\leq k\leq n\}}$, is called a differentiable programming space of order ${\displaystyle n}$.
Corollary.
A differentiable programming space of order ${\displaystyle n}$, ${\displaystyle {\mathcal {P}}_{n}:{\mathcal {V}}\to {\mathcal {V}}\otimes T({\mathcal {V}}^{*})}$, can be embedded into the tensor product of the function space ${\displaystyle {\mathcal {P}}_{0}:{\mathcal {V}}\to {\mathcal {V}}}$ and the subspace ${\displaystyle T_{n}({\mathcal {V}}^{*})}$ of the tensor algebra of the dual of the virtual space ${\displaystyle {\mathcal {P}}}$.
By taking the limit as ${\displaystyle n\to \infty }$, we consider
${\displaystyle {\mathcal {P}}_{\infty }<{\mathcal {P}}_{0}\otimes {\mathcal {T}}({\mathcal {V}}^{*}),}$
where ${\displaystyle {\mathcal {T}}({\mathcal {V}}^{*})=\prod _{k=0}^{\infty }({\mathcal {V}}^{*})^{\otimes k}}$ is the tensor series algebra, the algebra of the infinite formal tensor series, which is a completion of the tensor algebra ${\displaystyle T({\mathcal {V}}^{*})}$ in suitable topology.

Proofs can be found in.[9]

This means that we can represent calculation of derivatives of the map ${\displaystyle P:{\mathcal {V}}\to {\mathcal {V}}}$, with only one mapping ${\displaystyle \tau }$. We define the operator ${\displaystyle \tau _{n}}$ as a direct sum of operators

${\displaystyle \tau _{n}=1+\partial +\partial ^{2}+\ldots +\partial ^{n}}$

The image ${\displaystyle \tau _{k}P(\mathbf {x} )}$ is a multitensor of order ${\displaystyle k}$, which is a direct sum of the maps value and all derivatives of order ${\displaystyle n\leq k}$, all evaluated at the point ${\displaystyle \mathbf {x} }$:

${\displaystyle \tau _{k}P(\mathbf {x} )=P(\mathbf {x} )+\partial _{\mathbf {x} }P(\mathbf {x} )+\partial _{\mathbf {x} }^{2}P(\mathbf {x} )+\ldots +\partial _{\mathbf {x} }^{k}P(\mathbf {x} ).}$

The operator ${\displaystyle \tau _{n}}$ satisfies the recursive relation.

${\displaystyle \tau _{k+1}=1+\partial \tau _{k},}$

that can be used to recursively construct programming spaces of arbitrary order. Only explicit knowledge of ${\displaystyle \tau :{\mathcal {P}}_{0}\to {\mathcal {P}}_{1}}$ is required for the construction of ${\displaystyle {\mathcal {P}}_{n}}$ from ${\displaystyle {\mathcal {P}}_{1}}$, which is evident from the above theorem.

### Analytic virtual machine

The paper [9] proposed an abstract computational model, a virtual machine capable of constructing and implementing the theory. Such a machine provides a framework for analytic study of algorithmic procedures through algebraic means.

Claim
The tuple ${\displaystyle ({\mathcal {V}},{\mathcal {P}}_{0})}$ and the belonging tensor series algebra are sufficient conditions for the existence and construction of infinitely differentiable programing spaces ${\displaystyle {\mathcal {P}}_{\infty }}$, through linear combinations of elements of ${\displaystyle {\mathcal {P}}_{0}\otimes {\mathcal {T}}({\mathcal {V}}^{*})}$.

This claim allows a simple definition of such a machine.

Definition(Analytic virtual machine)
The tuple ${\displaystyle M=\langle {\mathcal {V}},{\mathcal {P}}_{0}\rangle }$ is an analytic, infinitely differentiable virtual machine, where
• ${\displaystyle {\mathcal {V}}}$ is a finite dimensional vector space
• ${\displaystyle {\mathcal {V}}\otimes {\mathcal {T}}({\mathcal {V}}^{*})}$ is the virtual memory space
• ${\displaystyle {\mathcal {P}}_{0}}$ is an analytic programming space over ${\displaystyle {\mathcal {V}}}$
When ${\displaystyle {\mathcal {P}}_{0}}$ is a differentiable programming space, this defines an infinitely differentiable virtual machine.

An illustrative open source implementation dCpp is available, closely following the theorems, intended as an educational guide.

### Tensor series expansion

Expansion into a series offers valuable insights into programs through methods of analysis.

There exists a space spanned by the set ${\displaystyle {\mathcal {D}}^{n}=\{\partial ^{k};\quad 0\leq k\leq n\}}$ over a field ${\displaystyle K}$. Thus, the expression

${\displaystyle e^{h\partial }=\sum \limits _{n=0}^{\infty }{\frac {(h\partial )^{n}}{n!}}}$

is well defined. The operator ${\displaystyle e^{h\partial }}$ is a mapping between function spaces

${\displaystyle e^{h\partial }:{\mathcal {P}}\to {\mathcal {P}}_{\infty }.}$

It also defines a map

${\displaystyle e^{h\partial }:{\mathcal {P}}\times {\mathcal {V}}\to {\mathcal {V}}\otimes {\mathcal {T}}({\mathcal {V}}^{*}),}$

by taking the image of the map ${\displaystyle e^{h\partial }(P)}$ at a certain point ${\displaystyle \mathbf {v} \in {\mathcal {V}}}$.

We may construct a map from the space of programs, to the space of polynomials. Note that the space of multivariate polynomials ${\displaystyle {\mathcal {V}}\to K}$ is isomorphic to symmetric algebra ${\displaystyle S({\mathcal {V}}^{*})}$, which is in turn a quotient of tensor algebra ${\displaystyle T({\mathcal {V}}^{*})}$. To any element of ${\displaystyle {\mathcal {V}}\otimes T({\mathcal {V}}^{*})}$ one can attach corresponding element of ${\displaystyle {\mathcal {V}}\otimes S({\mathcal {V}}^{i*})}$ namely a polynomial map ${\displaystyle {\mathcal {V}}\to {\mathcal {V}}}$. Thus, we consider the completion of the symmetric algebra ${\displaystyle S({\mathcal {V}}^{*})}$ as the Formal power series ${\displaystyle {\mathcal {S}}({\mathcal {V}}^{*})}$, which is in turn isomorphic to a quotient of tensor series algebra ${\displaystyle {\mathcal {T}}({\mathcal {V}}^{*})}$, arriving at

${\displaystyle e^{h\partial }:{\mathcal {P}}\times {\mathcal {V}}\to {\mathcal {V}}\otimes {\mathcal {S}}({\mathcal {V}}^{i*})}$

For any element ${\displaystyle \mathbf {v} _{0}\in {\mathcal {V}}}$, the expression ${\displaystyle e^{h\partial }(\cdot ,\mathbf {v} _{0})}$ is a map ${\displaystyle {\mathcal {P}}\to {\mathcal {V}}\otimes {\mathcal {S}}({\mathcal {V}}^{*})}$, mapping a program to a Formal power series. We can express the correspondence between multi-tensors in ${\displaystyle {\mathcal {V}}\otimes T({\mathcal {V}}^{*})}$ and polynomial maps ${\displaystyle {\mathcal {V}}\to {\mathcal {V}}}$ given by multiple contractions for all possible indices.

Theorem.
For a program ${\displaystyle P\in {\mathcal {P}}}$ the expansion into an infinite tensor series at the point ${\displaystyle \mathbf {v} _{0}\in {\mathcal {V}}}$ is expressed by multiple contractions
${\displaystyle P(\mathbf {v} _{0}+h\mathbf {v} )={\Big (}(e^{h\partial }P)(\mathbf {v} _{0}){\Big )}\cdot \mathbf {v} =\sum _{n=0}^{\infty }{\frac {h^{n}}{n!}}\partial ^{n}P(\mathbf {v} _{0})\cdot (\mathbf {v} ^{\otimes n})}$

Proof can be found in.[9] Evaluated at ${\displaystyle h=1}$, the operator is a generalization of the Shift operator widely used in physics. For a specific ${\displaystyle v_{0}\in {\mathcal {V}}}$ it is heron denoted by

${\displaystyle e^{\partial }\vert _{v_{0}}:{\mathcal {P}}\to {\mathcal {V}}\otimes {\mathcal {T}}({\mathcal {V}}^{*}).}$

When the choice of ${\displaystyle v_{0}\in {\mathcal {V}}}$ is arbitrary, we omit it from expressions for brevity.

### Operator of program composition

Theory offers a generalization of both forward and reverse mode of automatic differentiation to arbitrary order, under a single invariant operator in the theory. This condenses complex notions to simple expressions allowing meaningful manipulations before being applied to a particular programming space.

Theorem.
Composition of maps ${\displaystyle {\mathcal {P}}}$ is expressed as
${\displaystyle e^{h\partial }(f\circ g)=\exp(\partial _{f}e^{h\partial _{g}})(g,f)}$
where ${\displaystyle \exp(\partial _{f}e^{h\partial _{g}}):{\mathcal {P}}\times {\mathcal {P}}\to {\mathcal {P}}_{\infty }}$ is an operator on pairs of maps ${\displaystyle (g,f)}$, where ${\displaystyle \partial _{g}}$ is applied to ${\displaystyle g}$ and ${\displaystyle \partial _{f}}$ to ${\displaystyle f}$.

Proof can be found in.[9]

Both forward and reverse mode (generalized to arbitrary order) are obtainable using this operator, by fixing the appropriate one of the two maps. This generalizes both concepts under a single operator in the theory. For example, by considering projections of the operator onto the space spanned by ${\displaystyle {\mathcal {D}}=\{1,\partial \}}$, and fixing the second map ${\displaystyle g}$, we retrieve the basic first order forward mode of automatic differentiation, or reverse mode, by fixing ${\displaystyle f}$.

Thus the operator alleviates the need for explicit implementation of the higher order chain rule (see Faà di Bruno's formula), as it is encoded in the structure of the operator itself, which can be efficiently implemented by manipulating its generating map (see [9]).

### Order reduction for nested applications

It is useful to be able to use the ${\displaystyle k}$-th derivative of a program ${\displaystyle P\in {\mathcal {P}}}$ as part of a different differentiable program ${\displaystyle P_{1}}$. As such, we must be able to treat the derivative itself as a differentiable program ${\displaystyle P^{\prime k}\in {\mathcal {P}}}$, while only coding the original program ${\displaystyle P}$.

Theorem
There exists a reduction of order map ${\displaystyle \phi :{\mathcal {P}}_{n}\to {\mathcal {P}}_{n-1}}$ satisfying
${\displaystyle \forall _{P_{1}\in {\mathcal {P}}_{0}}\exists _{P_{2}\in {\mathcal {P}}_{0}}{\Big (}\phi ^{k}\circ e_{n}^{\partial }(P_{1})=e_{n-k}^{\partial }(P_{2}){\Big )}}$
for each ${\displaystyle n\geq 1}$, where ${\displaystyle e_{n}^{\partial }}$ is the projection of the operator ${\displaystyle e^{\partial }}$ onto the set ${\displaystyle {\mathcal {D}}^{n}=\{\partial ^{k};\quad 0\leq k\leq n\}}$.

By the above Theorem, ${\displaystyle n}$-differentiable ${\displaystyle k}$-th derivatives of a program ${\displaystyle P\in {\mathcal {P}}_{0}}$ can be extracted by

${\displaystyle ^{n}P^{k\prime }=\phi ^{k}\circ e_{n+k}^{\partial }(P)\in {\mathcal {P}}_{n}.}$

Thus, we gained the ability of writing a differentiable program acting on derivatives of another program, stressed as crucial by other authors.[10]

## Implementation

Forward-mode AD is implemented by a nonstandard interpretation of the program in which real numbers are replaced by dual numbers, constants are lifted to dual numbers with a zero epsilon coefficient, and the numeric primitives are lifted to operate on dual numbers. This nonstandard interpretation is generally implemented using one of two strategies: source code transformation or operator overloading.

### Source code transformation (SCT)

Figure 4: Example of how source code transformation could work

The source code for a function is replaced by an automatically generated source code that includes statements for calculating the derivatives interleaved with the original instructions.

Source code transformation can be implemented for all programming languages, and it is also easier for the compiler to do compile time optimizations. However, the implementation of the AD tool itself is more difficult.

### Operator overloading (OO)

Figure 5: Example of how operator overloading could work

Operator overloading is a possibility for source code written in a language supporting it. Objects for real numbers and elementary mathematical operations must be overloaded to cater for the augmented arithmetic depicted above. This requires no change in the form or sequence of operations in the original source code for the function to be differentiated, but often requires changes in basic data types for numbers and vectors to support overloading and often also involves the insertion of special flagging operations.

Operator overloading for forward accumulation is easy to implement, and also possible for reverse accumulation. However, current compilers lag behind in optimizing the code when compared to forward accumulation.

Operator overloading, for both forward and reverse accumulation, can be well-suited to applications where the objects are vectors of real numbers rather than scalars. This is because the tape then comprises vector operations; this can facilitate computationally efficient implementations where each vector operation performs many scalar operations. Vector adjoint algorithmic differentiation (vector AAD) techniques may be used, for example, to differentiate values calculated by Monte-Carlo simulation.

## Software

• C/C++
Package License Approach Brief info
ADC Version 4.0 nonfree OO
ADEL MIT OO Forward and reverse modes. Template-based with CUDA support.
Adept Apache 2.0 OO First-order forward and reverse modes. Very fast due to its use of expression templates and an efficient tape structure.
ADIC free for noncommercial SCT forward mode
ADMB BSD SCT+OO Uses a template approach. See http://admb-project.org/
ADNumber dual license OO arbitrary order forward/reverse
ADOL-C CPL 1.0 or GPL 2.0 OO arbitrary order forward/reverse, part of COIN-OR
AuDi GNU GPL 2.0 OO AuDi is an open source, header only, C++ library that allows for AUtomated DIfferentiation implementing the Taylor truncated polynomial algebra (forward mode automated differentiation). It was created with the aim to offer a generic solution to the user in need of a generic automated differentiation system. AuDi is thread-safe and, when possible, uses fine-grained parallelization of the truncated polynomial multiplication. The benefits of this fine grained parallelization are well visible for many variables and high orders. AuDi implements vectorized polynomial multiplication offering the most efficient implementation of a differential algebra for vectorized computations (i.e. computing the derivatives of the same expression on multiple points).
AMPL free for students SCT
CasADi GNU LGPL SCT Forward/reverse modes, matrix-valued atomic operations, code generation.
ceres-solver BSD OO A portable C++ library that allows for modeling and solving large complicated nonlinear least squares problems
CppAD EPL 1.0 or GPL 3.0 OO arbitrary order forward/reverse, AD<Base> for arbitrary Base including AD<Other_Base>, part of COIN-OR; can also be used to produce C source code using the CppADCodeGen library.
CTaylor free OO truncated taylor series, multi variable, high performance, calculating and storing only potentially nonzero derivatives, calculates higher order derivatives, order of derivatives increases when using matching operations until maximum order (parameter) is reached, example source code and executable available for testing performance
Eigen Auto Diff MPL2 OO
FADBAD++ free for
noncommercial
OO uses operator new
OpenAD depends on components SCT
ReverseAD free OO High order reverse mode which evaluates the high order derivative tensor directly instead of a forward/reverse modes hierarchy.
Sacado GNU GPL OO A part of the Trilinos collection, forward/reverse modes.
Stan (software) BSD OO forward- and reverse-mode automatic differentiation with library of special functions, probability functions, matrix operators, and linear algebra solvers; interfaces to MATLAB, R and Python.
TAPENADE Free for noncommercial SCT
Tensorflow Apache 2.0 OO TensorFlow is a Google-developed Python and C++ library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays both on CPU and GPU efficiently.
• Fortran
Package License Approach Brief info
ADF Version 4.0 nonfree OO
ADIFOR >>>
(free for non-commercial)
SCT
AUTO_DERIV free for non-commercial OO
OpenAD depends on components SCT
TAF nonfree SCT
TAPENADE Free for noncommercial SCT
GADfit Free (GPL 3) OO First (forward, reverse) and second (forward) order, principal use is nonlinear curve fitting, includes the differentiation of integrals
• Matlab
Package License Approach Brief info
AD for MATLAB GNU GPL OO Forward (1st & 2nd derivative, Uses MEX files & Windows DLLs)
Adiff BSD OO Forward (1st derivative)
MAD Proprietary OO Forward (1st derivative) full/sparse storage of derivatives
ADiMat Proprietary SCT Forward (1st & 2nd derivative) & Reverse (1st), proprietary server side transform
MADiff GNU GPL OO Reverse
AutoDiff BSD OO
CasADi GNU LGPL SCT Forward/reverse modes, matrix-valued atomic operations, code generation.
• Python
Package License Approach Brief info
ad BSD OO first and second-order, reverse accumulation, transparent on-the-fly calculations, basic NumPy support, written in pure python
FuncDesigner BSD OO uses NumPy arrays and SciPy sparse matrices,
also allows to solve linear/non-linear/ODE systems and
to perform numerical optimizations by OpenOpt
ScientificPython CeCILL OO see modules Scientific.Functions.FirstDerivatives and
Scientific.Functions.Derivatives
pycppad BSD OO arbitrary order forward/reverse, implemented as wrapper for CppAD including AD<double> and AD< AD<double> >.
pyadolc BSD OO wrapper for ADOL-C, hence arbitrary order derivatives in the (combined) forward/reverse mode of AD, supports sparsity pattern propagation and sparse derivative computations
uncertainties BSD OO first-order derivatives, reverse mode, transparent calculations
algopy BSD OO same approach as pyadolc and thus compatible, support to differentiate through numerical linear algebra functions like the matrix-matrix product, solution of linear systems, QR and Cholesky decomposition, etc.
pyderiv GNU GPL OO automatic differentiation and (co)variance calculation
CasADi GNU LGPL SCT Python front-end to CasADi. Forward/reverse modes, matrix-valued atomic operations, code generation.
Tensorflow Apache 2.0 OO TensorFlow is a Google-developed Python and C++ library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays both on CPU and GPU efficiently.
CNTK Apache 2.0 OO CNTK is a Microsoft-developed C++ library which supports Production-quality, Open Source, Multi-machine, Multi-GPU, Highly efficient RNN training, Speech, Image, Text
Theano BSD OO Theano is a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays both on CPU and GPU efficiently.
Autograd MIT OO Autograd can reverse-mode differentiate native Python and Numpy code. It can handle a large subset of Python's features, including loops, ifs, recursion and closures. It is closed under its own operation and hence can compute derivatives of any order.
Chainer MIT OO
pyaudi GNU GPL 2.0 OO Python front-end to AuDi. A fully functional and most efficient implementation of Taylor truncated polynomial algebra (a differential algebra) with the possibility of vectorized computations reaching great efficiency.
• Lua
Package License Approach Brief info
Torch BSD OO Torch is a LuaJIT library used for Deep Learning. Its nn package is divided into modular objects that share a common Module interface. Modules have a forward and backward function that allow them to feedforward and backpropagate (first-order derivatives). Modules can be joined together using module composites to create complex task-tailored graphs.
Torch-autograd Apache 2.0 OO
SciLua MIT OO SciLua, a framework for general purpose scientific computing in LuaJIT, features complete and transparent support for forward-mode automatic differentiation.
• .NET
Package License Approach Brief info
AutoDiff GNU LGPL OO Automatic differentiation with C# operator overloading.
FuncLib MIT OO Automatic differentiation and numerical optimization, operator overloading, unlimited order of differentiation, compilation to IL code for very fast evaluation.
DiffSharp GNU LGPL OO An automatic differentiation library implemented in the F# language. It supports C# and the other CLI languages. The library provides gradients, Hessians, Jacobians, directional derivatives, and matrix-free Hessian- and Jacobian-vector products, which can be incorporated with minimal change into existing algorithms. Operations can be nested to any level, meaning that you can compute exact higher-order derivatives and differentiate functions that are internally making use of differentiation.
Vector Accelerator MIT OO C# library for performing vector adjoint (reverse mode) algorithmic differentiation (also known as 'vector AAD'). Vector Accelerator uses operator overloading of vector operations, accelerated using processor-optimised libraries (e.g. Intel MKL).
• Haskell
Package License Approach Brief info
ad BSD OO Forward Mode (1st derivative or arbitrary order derivatives via lazy lists and sparse tries)
Reverse Mode
Combined forward-on-reverse Hessians.
Uses Quantification to allow the implementation automatically choose appropriate modes.
Quantification prevents perturbation/sensitivity confusion at compile time.
fad BSD OO Forward Mode (lazy list). Quantification prevents perturbation confusion at compile time.
rad BSD OO Reverse Mode. (Subsumed by 'ad').
Quantification prevents sensitivity confusion at compile time.
• Java
Package License Approach Brief info
JAutoDiff MIT OO Provides a framework to compute derivatives of functions on arbitrary types of field using generics. Coded in 100% pure Java.
Apache Commons Math Apache License v2 OO This class is an implementation of the extension to Rall's numbers described in Dan Kalman's paper[11]
Deriva Eclipse Public License v1.0 DSL+Code Generation Deriva automates algorithmic differentiation in Java and Clojure projects. It defines DSL for building extended arithmetic expressions (the extension being support for conditionals, allowing to express non analytic functions). The DSL is used to generate flat byte-code at runtime, providing implementation without overhead of function calls.
Jap Public OO/SCT Jap is a tools using Virtual Operator Overloading for java class. Jap was developed in the thesis of Phuong PHAM-QUANG 2008-2011.
Package License Approach Brief info
AutoDiffSource.jl MIT SCT Fast reverse mode differentiation for scalar and tensor functions using source code transformation.
ForwardDiff.jl MIT OO A unified package for forward-mode automatic differentiation, combining both DualNumbers and vector-based gradient accumulations.
DualNumbers.jl MIT OO Implements a Dual number type which can be used for forward-mode automatic differentiation of first derivatives via operator overloading.
HyperDualNumers.jl MIT OO Implements a Hyper number type which can be used for forward-mode automatic differentiation of first and second derivatives via operator overloading.
ReverseDiffSource.jl MIT SCT Implements reverse-mode automatic differentiation for gradients and high-order derivatives given user-supplied expressions or generic functions. Accepts a subset of valid Julia syntax, including intermediate assignments.
TaylorSeries.jl MIT OO Implements truncated multivariate power series for high-order integration of ODEs and forward-mode automatic differentiation of arbitrary order derivatives via operator overloading.
• Clojure
Package License Approach Brief info
Deriva Eclipse Public License v1.0 DSL+Code Generation Deriva automates algorithmic differentiation in Java and Clojure projects. It defines DSL for building extended arithmetic expressions (the extension being support for conditionals, allowing to express non analytic functions). The DSL is used to generate flat byte-code at runtime, providing implementation without overhead of function calls.
Package License Approach Brief info
ad Apache 2.0 DSL+Code Generation Creates source code corresponding to algebraic expressions. See https://autodiff.info for a demo.
Package License Approach Brief info
PBSadmb GNU GPL SCT+OO Uses a template approach. See http://admb-project.org/ .

## References

1. ^ Neidinger, Richard D. (2010). "Introduction to Automatic Differentiation and MATLAB Object-Oriented Programming" (PDF). SIAM Review. 52 (3): 545–563. doi:10.1137/080743627.
2. ^ http://www.ec-securehost.com/SIAM/SE24.html
3. ^ R.E. Wengert (1964). "A simple automatic derivative evaluation program". Comm. ACM. 7: 463–464. doi:10.1145/355586.364791.
4. ^ Bartholomew-Biggs, Michael; Brown, Steven; Christianson, Bruce; Dixon, Laurence (2000). "Automatic differentiation of algorithms" (PDF). Journal of Computational and Applied Mathematics. 124 (1-2): 171–190. doi:10.1016/S0377-0427(00)00422-2.
5. ^ Linnainmaa, Seppo (1970). The representation of the cumulative rounding error of an algorithm as a Taylor expansion of the local rounding errors. Master's Thesis (in Finnish), Univ. Helsinki, 6-7.
6. ^ Linnainmaa, Seppo (1976). Taylor expansion of the accumulated rounding error. BIT Numerical Mathematics, 16(2), 146-160.
7. ^ Griewank, Andreas (2012). Who Invented the Reverse Mode of Differentiation?. Optimization Stories, Documenta Matematica, Extra Volume ISMP (2012), 389-400.
8. ^ Naumann, Uwe (April 2008). "Optimal Jacobian accumulation is NP-complete". Mathematical Programming. 112 (2): 427–441. doi:10.1007/s10107-006-0042-z. |contribution= ignored (help)
9. Sajovic, Žiga; Vuk, Martin (2016). "Operational calculus on programming spaces and generalized tensor networks". arXiv: [math.FA].
10. ^ Pearlmutter, Barak A.; Siskind, Jeffrey M (May 2008). "PPutting the Automatic Back into AD: Part II". ECE Technical Reports.
11. ^ Kalman, Dan (June 2002). "Doubly Recursive Multivariate Automatic Differentiation" (PDF). Mathematics Magazine. 75 (3): 187–202. doi:10.2307/3219241.