DNNsolver: Accurate and Efficient Deep Learning Modeling of Polarization-Sensitive Diffractive Neural Networks

Abstract

Diffractive neural networks (DNNs), composed of cascaded diffractive layers, offer a promising platform for high-speed optical computing. However, accurate and efficient modeling of high-density, polarization-sensitive DNNs remains challenging. Conventional model based on thin element approximation (TEA) fails to capture electromagnetic coupling within layers, while full-wave simulations such as finite-difference time-domain (FDTD) methods are computationally expensive. Here, we propose DNNsolver, a deep learning-based electromagnetic surrogate model that learns the scattering response of diffractive layers. By predicting the scattering matrix rather than source-dependent output fields, DNNsolver decouples the incident wavefront from the model and could be applied to arbitrary input fields. A local scattering kernel and sliding-window strategy further allow efficient modeling of DNNs with arbitrary spatial sizes. For single diffractive layer under random speckle illumination, DNNsolver achieves an average mean squared error (MSE) of 0.0095 and about 10^6-fold speedup compared with FDTD simulations. DNNsolver is further applied to the optimization of a trilayer polarization-multiplexed image classifier, which exhibits better agreement with FDTD simulations than its TEA-based counterpart. Our work provides an accurate and efficient framework for the design and optimization of high-density, polarization-sensitive DNNs.