Data-driven acceleration of photonic simulations

• Published in: Scientific reports Vol. 9; no. 1; p. 19728
• Main Authors: Trivedi, Rahul, Su, Logan, Lu, Jesse, Schubert, Martin F., Vuckovic, Jelena
• Format: Journal Article
• Language: English
• Published: London Nature Publishing Group UK 23.12.2019; Nature Publishing Group
• Subjects: 639/624/400/1021; 639/705/1042; Humanities and Social Sciences; Learning algorithms; multidisciplinary; Neural networks; Science; Science (multidisciplinary); Simulation
• ISSN: 2045-2322; 2045-2322
• DOI: 10.1038/s41598-019-56212-5

Designing modern photonic devices often involves traversing a large parameter space via an optimization procedure, gradient based or otherwise, and typically results in the designer performing electromagnetic simulations of a large number of correlated devices. In this paper, we investigate the possibility of accelerating electromagnetic simulations using the data collected from such correlated simulations. In particular, we present an approach to accelerate the Generalized Minimal Residual (GMRES) algorithm for the solution of frequency-domain Maxwell’s equations using two machine learning models (principal component analysis and a convolutional neural network). These data-driven models are trained to predict a subspace within which the solution of the frequency-domain Maxwell’s equations approximately lies. This subspace is then used for augmenting the Krylov subspace generated during the GMRES iterations, thus effectively reducing the size of the Krylov subspace and hence the number of iterations needed for solving Maxwell’s equations. By training the proposed models on a dataset of wavelength-splitting gratings, we show an order of magnitude reduction (~10–50) in the number of GMRES iterations required for solving frequency-domain Maxwell’s equations.