arXiv · 2101.06757
Higher Order Automatic Differentiation of Higher Order Functions
Abstract
We present semantic correctness proofs of automatic differentiation (AD). We consider a forward-mode AD method on a higher-order language with algebraic data types and we characterise it as the unique structure-preserving macro given a choice of derivatives for basic operations. We describe a rich semantics for differentiable programming based on diffeological spaces. We show that it interprets our language and we phrase what it means for the AD method to be correct with respect to this semantics. We show that our characterisation of AD gives rise to an elegant semantic proof of its correctness based on a gluing construction on diffeological spaces. We explain how this is in essence a logical relations argument. Throughout we show how the analysis extends to AD methods for computing higher-order derivatives using a Taylor approximation.
Explore related subjects
Keep this discovery
Mathieu Huot, Sam Staton, Matthijs Vákár. 2021-01-17. Higher Order Automatic Differentiation of Higher Order Functions. https://doi.org/10.46298/lmcs-18(1:41)2022
Cite the original work for its findings. Save a collection to share your selection of sources.