Prediction of metasurface spectral response based on a deep neural network

Abstract

The two-dimensional optical metasurface can realize the free regulation of light waves through the free design of structure, which is highly appreciated by researchers. As there are high requirements for computer hardware, long time for simulation calculations, and data waste in the process of using the time-domain finite-difference method to solve the optical properties of the metasurface, the deep neural network (DNN) is proposed to predict the spectral response of an optical metasurface. The structural parameters of the metasurface are taken as inputs and the metasurface transmission spectrum is used as the output. To achieve better prediction results, different gradient descent algorithms were selected and the parameters of the DNN model were optimized. After 5 × 104 times of epoch training, the loss function mean squared error (MSE) reaches 2.665 × 10−3, the sum error of 98% test data is less than 3.23, and the relative error is less than 2%. The results show that the DNN model has an excellent prediction effect. Compared with the traditional simulation method, the efficiency of this model is improved by 104 times, which can improve the efficiency of optical micro-nano structure design.