Optimal Dirichlet Boundary Control by Fourier Neural Operators Applied to Nonlinear Optics

Abstract

We present an approach for solving optimal Dirichlet boundary control problems of nonlinear optics by using deep learning. For computing high resolution approximations of the solution to the nonlinear wave model, we propose higher order space-time finite element methods in combination with collocation techniques. Thereby, $C^l$-regularity in time of the global discrete is ensured. The resulting simulation data is used to train solution operators that effectively leverage the higher regularity of the training data. The solution operator is represented by Fourier Neural Operators and Gated Recurrent Units and can be used as the forward solver in the optimal Dirichlet boundary control problem. The proposed algorithm is implemented and tested on modern high-performance computing platforms, with a focus on efficiency and scalability. The effectiveness of the approach is demonstrated on the problem of generating Terahertz radiation in periodically poled Lithium Niobate, where the neural network is used as the solver in the optimal control setting to optimize the parametrization of the optical input pulse and maximize the yield of $0.3$THz-frequency radiation. We exploit the periodic layering of the crystal to design the neural networks. The networks are trained to learn the propagation through one period of the layers. The recursive application of the network onto itself yields an approximation to the full problem. Our results indicate that the proposed method can achieve a significant speedup in computation time compared to classical methods. A comparison of our results to experimental data shows the potential to revolutionize the way we approach optimization problems in nonlinear optics.

arXiv comment43 pages, 11 figures, 7 tables. Submitted to Journal of Comptutational Physics

References in corpus (15)

• DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
• Artificial Neural Networks for Solving Ordinary and Partial Differential Equations
• DGM: A deep learning algorithm for solving partial differential equations
• Stable Architectures for Deep Neural Networks
• A Machine Learning Framework for Solving High-Dimensional Mean Field Game and Mean Field Control Problems
• Physics-Informed Probabilistic Learning of Linear Embeddings of Non-linear Dynamics With Guaranteed Stability
• Structure-preserving neural networks
• A Mean-Field Optimal Control Formulation of Deep Learning
• BCR-Net: a neural network based on the nonstandard wavelet form
• Highly efficient narrowband terahertz generation driven by a two-spectral-line laser in PPLN
• Mapped tent pitching schemes for hyperbolic systems
• Derivative-Informed Projected Neural Networks for High-Dimensional Parametric Maps Governed by PDEs
• Flexible goal-oriented adaptivity for higher-order space-time discretizations of transport problems with coupled flow
• Variational Time Discretizations of Higher Order and Higher Regularity
• Accurate simulation of THz generation with Finite-Element Time Domain methods

Cited by in corpus (2)

• Neural Operators for PDE Backstepping Control of First-Order Hyperbolic PIDE with Recycle and Delay
• DNN-MG: A Hybrid Neural Network/Finite Element Method with Applications to 3D Simulations of the Navier-Stokes Equations