[Top][All Lists]

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

[Swarm-modeling] Automatic Differentiation of Programs with Discrete Ran

From: glen e ropella
Subject: [Swarm-modeling] Automatic Differentiation of Programs with Discrete Randomness
Date: Thu, 20 Oct 2022 11:11:50 -0700
User-agent: Godzilla/8.0 (X11; Plan9 raytheon; rv:0.7) Salamander/19100101, Eudora/1.7.0


Automatic differentiation (AD), a technique for constructing new programs which compute the derivative of an original program, has become ubiquitous throughout scientific computing and deep learning due to the improved performance afforded by gradient-based optimization. However, AD systems have been restricted to the subset of programs that have a continuous dependence on parameters. Programs that have discrete stochastic behaviors governed by distribution parameters, such as flipping a coin with probability of being heads, pose a challenge to these systems because the connection between the result (heads vs tails) and the parameters (p) is fundamentally discrete. In this paper we develop a new reparameterization-based methodology that allows for generating programs whose expectation is the derivative of the expectation of the original program. We showcase how this method gives an unbiased and low-variance estimator which is as automated as traditional AD mechanisms. We demonstrate unbiased forward-mode AD of discrete-time Markov chains, agent-based models such as Conway's Game of Life, and unbiased reverse-mode AD of a particle filter. Our code is available at this https URL.

reply via email to

[Prev in Thread] Current Thread [Next in Thread]