Search papers, labs, and topics across Lattice.
This paper presents a novel approach to training neural network surrogates for wave-scattering inverse problems by dynamically generating salient training examples, overcoming the limitations of traditional single-step surrogates that can only handle a few tens of variables. The proposed method employs gradient ascent to optimize configurations where the surrogate diverges from a full-wave simulator, significantly enhancing the surrogate's capacity to manage up to 41,772 controllable variables and generalizing to over 3 million without retraining. The resulting surrogates demonstrate robust accuracy and substantial speedups in simulations compared to finite-difference time-domain methods, making them a viable alternative for complex photonic designs.
Surrogates can now handle over 3 million controllable variables, achieving up to 26.5x speedups over traditional simulation methods.
Neural network surrogates are an emerging alternative to traditional electromagnetic wave simulators like finite-difference time-domain (FDTD); their goal is to replace rigorous physical simulations with pre-trained neural networks that solve wave-scattering forward and inverse problems orders of magnitude faster. However, nonrecurrent, single-step surrogates have scaled only to a few tens of simulation variables. Here, we show that this barrier can be overcome by dynamically generating salient training examples during training, rather than randomly sampling the large space of possible examples. We introduce an algorithm that runs in parallel with surrogate training, using gradient ascent to search refractive-index and source configurations for cases where the surrogate disagrees with a full-wave ground-truth simulator. We also use source and ground-truth normalization with an evolving replay dataset to stabilize and accelerate learning. Using this approach, we train a fast, single-step surrogate for two-dimensional wave scattering with up to 41,772 controllable variables, including dense, freely configurable grids of refractive indices and complex-valued sources. The resulting neural surrogate is robustly accurate across diverse structured and unstructured examples and generalizes inductively to larger domains, reaching over 3 million controllable variables without retraining, a $73.8\times$ increase. We demonstrate the surrogate on large-scale forward simulations and inverse design of freeform beam splitters and gradient-index (GRIN) lenses up to 98 wavelengths wide, showing comparable or better performance than FDTD-based designs, with speedups from $1.29\times$ to $26.5\times$. These results demonstrate a practical path toward fast, robustly accurate, inductively scalable neural simulators for photonic inverse design and other wave-scattering inverse problems.
