Analytical derivatives of neural networks
Abstract
We propose a simple recursive algorithm that allows the computation of the first- and second-order derivatives with respect to the inputs of an arbitrary deep feed forward neural network (DFNN). The algorithm naturally incorporates the derivatives with respect to the network parameters. To test the algorithm, we apply it to the study of the quantum mechanical variational problem for few cases of simple potentials, modeling the ground-state wave function in terms of a DFNN.
- Publication:
-
Computer Physics Communications
- Pub Date:
- January 2022
- DOI:
- 10.1016/j.cpc.2021.108169
- arXiv:
- arXiv:2105.02737
- Bibcode:
- 2022CoPhC.27008169R
- Keywords:
-
- Deep neural networks;
- PINN;
- Variational neural network;
- Physics - Computational Physics;
- High Energy Physics - Phenomenology
- E-Print:
- 12 pages, 7 figures