Home | Resources | Generative Neural Reparameterization for Differentiable PDE-Constrained Optimization

Generative Neural Reparameterization for Differentiable PDE-Constrained Optimization

Tags:
Artificial Intelligence Computational Physics FES Funded Machine Learning Numerical Analysis Plasma Physics
Funded
Archis S. Joglekar

Abstract:

Partial-differential-equation (PDE)-constrained optimization is a well-worn technique for acquiring optimal parameters of systems governed by PDEs. However, this approach is limited to providing a single set of optimal parameters per optimization. Given a differentiable PDE solver, if the free parameters are reparametrized as the output of a neural network, that neural network can be trained to learn a map from a probability distribution to the distribution of optimal parameters. This proves useful in the case where there are many well performing local minima for the PDE. We apply this technique to train a neural network that generates optimal parameters that minimize laser-plasma instabilities relevant to laser fusion and show that the neural network generates many well performing and diverse minima.

full publication

Resources

Core Resources for the Fusion Energy Community