Portability | GHC only |
---|---|

Stability | experimental |

Maintainer | ekmett@gmail.com |

Safe Haskell | None |

This module provides reverse-mode Automatic Differentiation using post-hoc linear time topological sorting.

For reverse mode AD we use `StableName`

to recover sharing information from
the tape to avoid combinatorial explosion, and thus run asymptotically faster
than it could without such sharing information, but the use of side-effects
contained herein is benign.

- data Kahn a s
- grad :: (Traversable f, Num a) => (forall s. f (Kahn a s) -> Kahn a s) -> f a -> f a
- grad' :: (Traversable f, Num a) => (forall s. f (Kahn a s) -> Kahn a s) -> f a -> (a, f a)
- gradWith :: (Traversable f, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> Kahn a s) -> f a -> f b
- gradWith' :: (Traversable f, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> Kahn a s) -> f a -> (a, f b)
- jacobian :: (Traversable f, Functor g, Num a) => (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (f a)
- jacobian' :: (Traversable f, Functor g, Num a) => (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (a, f a)
- jacobianWith :: (Traversable f, Functor g, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (f b)
- jacobianWith' :: (Traversable f, Functor g, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (a, f b)
- hessian :: (Traversable f, Num a) => (forall s s'. f (On (Kahn (Kahn a s') s)) -> On (Kahn (Kahn a s') s)) -> f a -> f (f a)
- hessianF :: (Traversable f, Functor g, Num a) => (forall s s'. f (On (Kahn (Kahn a s') s)) -> g (On (Kahn (Kahn a s') s))) -> f a -> g (f (f a))
- diff :: Num a => (forall s. Kahn a s -> Kahn a s) -> a -> a
- diff' :: Num a => (forall s. Kahn a s -> Kahn a s) -> a -> (a, a)
- diffF :: (Functor f, Num a) => (forall s. Kahn a s -> f (Kahn a s)) -> a -> f a
- diffF' :: (Functor f, Num a) => (forall s. Kahn a s -> f (Kahn a s)) -> a -> f (a, a)
- vgrad :: Grad i o o' a => i -> o
- vgrad' :: Grad i o o' a => i -> o'
- class Num a => Grad i o o' a | i -> a o o', o -> a i o', o' -> a i o

# Documentation

`Kahn`

is a `Mode`

using reverse-mode automatic differentiation that provides fast `diffFU`

, `diff2FU`

, `grad`

, `grad2`

and a fast `jacobian`

when you have a significantly smaller number of outputs than inputs.

Typeable2 Kahn | |

(Num a, Bounded a) => Bounded (Kahn a s) | |

(Num a, Enum a) => Enum (Kahn a s) | |

(Num a, Eq a) => Eq (Kahn a s) | |

Floating a => Floating (Kahn a s) | |

Fractional a => Fractional (Kahn a s) | |

Num a => Num (Kahn a s) | |

(Num a, Ord a) => Ord (Kahn a s) | |

Real a => Real (Kahn a s) | |

RealFloat a => RealFloat (Kahn a s) | |

RealFrac a => RealFrac (Kahn a s) | |

Show a => Show (Kahn a s) | |

MuRef (Kahn a s) | |

Erf a => Erf (Kahn a s) | |

InvErf a => InvErf (Kahn a s) | |

Num a => Mode (Kahn a s) | |

Num a => Jacobian (Kahn a s) | |

Num a => Grad (Kahn a ()) [a] (a, [a]) a | |

Grad i o o' a => Grad (Kahn a () -> i) (a -> o) (a -> o') a |

# Gradient

grad :: (Traversable f, Num a) => (forall s. f (Kahn a s) -> Kahn a s) -> f a -> f aSource

The `grad`

function calculates the gradient of a non-scalar-to-scalar function with kahn-mode AD in a single pass.

`>>>`

[2,1,1]`grad (\[x,y,z] -> x*y+z) [1,2,3]`

grad' :: (Traversable f, Num a) => (forall s. f (Kahn a s) -> Kahn a s) -> f a -> (a, f a)Source

The `grad'`

function calculates the result and gradient of a non-scalar-to-scalar function with kahn-mode AD in a single pass.

`>>>`

(28.566231899122155,[29.5562243957226,29.5562243957226,-0.1411200080598672])`grad' (\[x,y,z] -> 4*x*exp y+cos z) [1,2,3]`

gradWith :: (Traversable f, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> Kahn a s) -> f a -> f bSource

gradWith' :: (Traversable f, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> Kahn a s) -> f a -> (a, f b)Source

# Jacobian

jacobian :: (Traversable f, Functor g, Num a) => (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (f a)Source

The `jacobian`

function calculates the jacobian of a non-scalar-to-non-scalar function with kahn AD lazily in `m`

passes for `m`

outputs.

`>>>`

[[0,1],[1,0],[1,2]]`jacobian (\[x,y] -> [y,x,x*y]) [2,1]`

`>>>`

[[0.0,7.38905609893065],[-0.8414709848078965,0.0],[1.0,1.0]]`jacobian (\[x,y] -> [exp y,cos x,x+y]) [1,2]`

jacobian' :: (Traversable f, Functor g, Num a) => (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (a, f a)Source

The `jacobian'`

function calculates both the result and the Jacobian of a nonscalar-to-nonscalar function, using `m`

invocations of kahn AD,
where `m`

is the output dimensionality. Applying `fmap snd`

to the result will recover the result of `jacobian`

| An alias for `gradF'`

ghci> jacobian' ([x,y] -> [y,x,x*y]) [2,1] [(1,[0,1]),(2,[1,0]),(2,[1,2])]

jacobianWith :: (Traversable f, Functor g, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (f b)Source

'jacobianWith g f' calculates the Jacobian of a non-scalar-to-non-scalar function `f`

with kahn AD lazily in `m`

passes for `m`

outputs.

Instead of returning the Jacobian matrix, the elements of the matrix are combined with the input using the `g`

.

`jacobian`

=`jacobianWith`

(_ dx -> dx)`jacobianWith`

`const`

= (f x ->`const`

x`<$>`

f x)

jacobianWith' :: (Traversable f, Functor g, Num a) => (a -> a -> b) -> (forall s. f (Kahn a s) -> g (Kahn a s)) -> f a -> g (a, f b)Source

`jacobianWith`

g f' calculates both the result and the Jacobian of a nonscalar-to-nonscalar function `f`

, using `m`

invocations of kahn AD,
where `m`

is the output dimensionality. Applying `fmap snd`

to the result will recover the result of `jacobianWith`

Instead of returning the Jacobian matrix, the elements of the matrix are combined with the input using the `g`

.

`jacobian'`

==`jacobianWith'`

(_ dx -> dx)

# Hessian

hessian :: (Traversable f, Num a) => (forall s s'. f (On (Kahn (Kahn a s') s)) -> On (Kahn (Kahn a s') s)) -> f a -> f (f a)Source

Compute the `hessian`

via the `jacobian`

of the gradient. gradient is computed in `Kahn`

mode and then the `jacobian`

is computed in `Kahn`

mode.

However, since the

is square this is not as fast as using the forward-mode `grad`

f :: f a -> f a`jacobian`

of a reverse mode gradient provided by `hessian`

.

`>>>`

[[0,1],[1,0]]`hessian (\[x,y] -> x*y) [1,2]`

hessianF :: (Traversable f, Functor g, Num a) => (forall s s'. f (On (Kahn (Kahn a s') s)) -> g (On (Kahn (Kahn a s') s))) -> f a -> g (f (f a))Source

Compute the order 3 Hessian tensor on a non-scalar-to-non-scalar function via the `Kahn`

-mode Jacobian of the `Kahn`

-mode Jacobian of the function.

Less efficient than `hessianF`

.

`>>>`

[[[0.0,1.0],[1.0,0.0]],[[0.0,0.0],[0.0,0.0]],[[-1.1312043837568135,-2.4717266720048188],[-2.4717266720048188,1.1312043837568135]]]`hessianF (\[x,y] -> [x*y,x+y,exp x*cos y]) [1,2]`

# Derivatives

diff :: Num a => (forall s. Kahn a s -> Kahn a s) -> a -> aSource

Compute the derivative of a function.

`>>>`

1.0`diff sin 0`

`>>>`

1.0`cos 0`

diff' :: Num a => (forall s. Kahn a s -> Kahn a s) -> a -> (a, a)Source

The `diff'`

function calculates the value and derivative, as a
pair, of a scalar-to-scalar function.

`>>>`

(0.0,1.0)`diff' sin 0`

diffF :: (Functor f, Num a) => (forall s. Kahn a s -> f (Kahn a s)) -> a -> f aSource

Compute the derivatives of a function that returns a vector with regards to its single input.

`>>>`

[1.0,0.0]`diffF (\a -> [sin a, cos a]) 0`

diffF' :: (Functor f, Num a) => (forall s. Kahn a s -> f (Kahn a s)) -> a -> f (a, a)Source

Compute the derivatives of a function that returns a vector with regards to its single input as well as the primal answer.

`>>>`

[(0.0,1.0),(1.0,0.0)]`diffF' (\a -> [sin a, cos a]) 0`

# Unsafe Variadic Gradient

Variadic combinators for variadic mixed-mode automatic differentiation.

Unfortunately, variadicity comes at the expense of being able to use
quantification to avoid sensitivity confusion, so be careful when
counting the number of `auto`

calls you use when taking the gradient
of a function that takes gradients!