Efficient inverse design and spectrum prediction for nanophotonic devices based on deep recurrent neural networks

Abstract

Abstract The development of nanophotonic devices has presented a revolutionary means to manipulate light at nanoscale. How to efficiently design these devices is an active area of research. Recently, artificial neural networks (ANNs) have displayed powerful ability in the inverse design of nanophotonic devices. However, there is limited research on the inverse design for modeling and learning the sequence characteristics of a spectrum. In this work, we propose a deep learning method based on an improved recurrent neural network to extract the sequence characteristics of a spectrum and achieve inverse design and spectrum prediction. A key feature of the network is that the memory or feedback loops it comprises allow it to effectively recognize time series data. In the context of nanorods hyperbolic metamaterials, we demonstrated the high consistency between the target spectrum and the predicted spectrum, and the network learned the deep physical relationship concerning the structural parameter changes reflected on the spectrum. The effectiveness of our approach is also tested by user-drawn spectra. Moreover, the proposed model is capable of predicting an unknown spectrum based on a known spectrum with only 0.32% mean relative error. The prediction model may be helpful to predict data beyond the detection limit. We propose this versatile method as an effective and accurate alternative to the application of ANNs in nanophotonics, paving way for fast and accurate design of desired devices.