paperbot · PL 论文追踪

RSS

Perturbation confusion in forward automatic differentiation of higher-order functions

JFP vol.292019
OLEKSANDR MANZYUK, BARAK A. PEARLMUTTER, ALEXEY ANDREYEVICH RADUL, DAVID R. RUSH, JEFFREY MARK SISKIND

尚未生成 AI 速览(可能缺少 API key 或等待下次运行补跑)。

原文摘要(Abstract)

Abstract Automatic differentiation (AD) is a technique for augmenting computer programs to compute derivatives. The essence of AD in its forward accumulation mode is to attach perturbations to each number, and propagate these through the computation by overloading the arithmetic operators. When derivatives are nested, the distinct derivative calculations, and their associated perturbations, must be distinguished. This is typically accomplished by creating a unique tag for each derivative calculation and tagging the perturbations. We exhibit a subtle bug, present in fielded implementations which support derivatives of higher-order functions, in which perturbations are confused despite the tagging machinery, leading to incorrect results. The essence of the bug is as follows: a unique tag is needed for each derivative calculation, but in existing implementations unique tags are created when taking the derivative of a function at a point. When taking derivatives of higher-order functions, these need not correspond! We exhibit a simple example: a higher-order function f whose derivative at a point x , namely f ′( x ), is itself a function which calculates a derivative. This situation arises naturally when taking derivatives of curried functions. Two potential solutions are presented, and their deficiencies discussed. One uses eta expansion to delay the creation of fresh tags in order to put them into one-to-one correspondence with derivative calculations. The other wraps outputs of derivative operators with tag substitution machinery. Both solutions seem very difficult to implement without violating the desirable complexity guarantees of forward AD.

链接与引用

DOI 原文 ·

BibTeX
@article{paperbot720,
  title = {Perturbation confusion in forward automatic differentiation of higher-order functions},
  author = {OLEKSANDR MANZYUK and BARAK A. PEARLMUTTER and ALEXEY ANDREYEVICH RADUL and DAVID R. RUSH and JEFFREY MARK SISKIND},
  journal = {Journal of Functional Programming},
  volume = {29},
  year = {2019},
  doi = {10.1017/s095679681900008x}
}