Fourier Stochastic Backpropagation

Amine Echraibi (1, 2) , Joachim Flocon Cholet , Stéphane Gosselin , Sandrine Vaton (1)

1 IMT Atlantique - INFO - Département Informatique
2 Orange Labs [Lannion]

Backpropagating gradients through random variables is at the heart of numerous machine learning applications. In this paper, we present a general framework for deriving stochastic backpropagation rules for any distribution, discrete or continuous. Our approach exploits the link between the characteristic function and the Fourier transform, to transport the derivatives from the parameters of the distribution to the random variable. Our method generalizes previously known estimators, and results in new estimators for the gamma, beta, Dirichlet and Laplace distributions. Furthermore, we show that the classical deterministic backproapagation rule and the discrete random variable case, can also be interpreted through stochastic backpropagation.

Domaines: Machine Learning [stat.ML]

Amine Echraibi, Joachim Flocon Cholet, Stéphane Gosselin, Sandrine Vaton. Fourier Stochastic Backpropagation. 2020. ⟨hal-02968975v2⟩