May, 2021

反向模式自动微分的分解

TL;DR本文提出了将反向模式自动微分分解为前向模式线性化和转置的方法,它既可以将二者的本质区别隔离出来,又简化了它们的联合实现。具体而言,一旦在源语言中定义了每个基本操作的前向模式自动微分规则,只有线性基元需要额外的转置规则才能得到完整的反向模式自动微分实现。这就是反向模式自动微分在 JAX 和 Dex 中的编写方式。