Backpropagation in the Simply Typed Lambda-calculus with Linear Negation
Backpropagation is a classic automatic differentiation algorithm computing the gradient of functions specified by a certain class of simple, first-order programs, called computational graphs. It is a fundamental tool in several fields, most notably machine learning, where it is the key for efficiently training (deep) neural networks. Recent years have witnessed the quick growth of a research field called differentiable programming, the aim of which is to express computational graphs more synthetically and modularly by resorting to actual programming languages endowed with control flow operators and higher-order combinators, such as map and fold. In this paper, we extend the backpropagation algorithm to a paradigmatic example of such a programming language: we define a compositional program transformation from the simply-typed lambda-calculus to itself augmented with a notion of linear negation, and prove that this computes the gradient of the source program with the same efficiency as first-order backpropagation. The transformation is completely effect-free and thus provides a purely logical understanding of the dynamics of backpropagation.
Slides (Backprop.pdf) | 184KiB |
Wed 22 JanDisplayed time zone: Saskatchewan, Central America change
15:35 - 16:40 | Automatic Differentiation / Kleene AlgebraResearch Papers at Ile de France II (IDF II) Chair(s): Lars Birkedal Aarhus University | ||
15:35 21mTalk | A Simple Differentiable Programming Language Research Papers Link to publication DOI Media Attached | ||
15:56 21mTalk | Backpropagation in the Simply Typed Lambda-calculus with Linear Negation Research Papers Link to publication DOI Media Attached File Attached | ||
16:18 21mTalk | Guarded Kleene Algebra with Tests: Verification of Uninterpreted Programs in Nearly Linear TimeDistinguished Paper Research Papers Steffen Smolka Cornell University, Nate Foster Cornell University, Justin Hsu University of Wisconsin-Madison, USA, Tobias Kappé University College London, Dexter Kozen Cornell University, Alexandra Silva University College London Link to publication DOI Media Attached |