Abstract:Employing deep learning models to design high-performance metasurfaces has garnered significant attention due to its potential benefits in terms of accuracy and efficiency. A deep learning-based metasurface design framework typically comprises a forward prediction path for predicting optical responses and a backward retrieval path for generating geometrical configurations. In the forward design path, a specific geometrical configuration corresponds to a unique optical response. However, in the inverse design path, a single performance metric can correspond to multiple potential designs. This one-to-many mapping poses a significant challenge for deep learning models and can potentially impede their performance. Although representing the inverse path as a probabilistic distribution is a widely adopted method for tackling this problem, accurately capturing the posterior distribution to encompass all potential solutions remains an ongoing challenge. Furthermore, in most pioneering works, the forward and backward paths are captured using separate models. However, the knowledge acquired from the forward path does not contribute to the training of the backward model. This separation of models adds complexity to the system and can hinder the overall efficiency and effectiveness of the design framework. Here, we utilized an invertible neural network (INN) to simultaneously model both the forward and inverse process. Unlike other frameworks, INN focuses on the forward process and implicitly captures a probabilistic model for the inverse process. Given a specific optical response, the INN enables the recovery of the complete posterior over the parameter space. This capability allows for the generation of novel designs that are not present in the training data. Through the integration of the INN with the angular spectrum method, we have developed an efficient and automated end-to-end metasurface design and evaluation framework. This novel approach eliminates the need for human intervention and significantly speeds up the design process. Utilizing this advanced framework, we have effectively designed high-efficiency metalenses and dual-polarization metasurface holograms. This approach extends beyond dielectric metasurface design, serving as a general method for modeling optical inverse design problems in diverse optical fields.

End-to-end metasurface inverse design for single-shot multi-channel imaging

Neural Network-Based Surrogate Model for Inverse Design of Metasurfaces

Bi-channel Compressive Hyperspectral Imager Based on Polarization Conversion Metasurface

High-performance full-color imaging system based on end-to-end joint optimization of computer-generated holography and metalens.

End-to-end optimization of single-shot monocular metasurface camera for RGBD imaging

Spectro-Polarimetric-Depth Imaging by Inverse-Designed Single-Cell Metasurface

End-to-End Optimization of Metasurfaces for Imaging with Compressed Sensing

Broadband Thermal Imaging using Meta-Optics

End-to-end joint optimization of metasurface and image processing for compact snapshot hyperspectral imaging

3D-patterned inverse-designed mid-infrared metaoptics

Polarization Multi-Image Synthesis with Birefringent Metasurfaces

End-to-End Diverse Metasurface Design and Evaluation Using an Invertible Neural Network

Single-shot three-dimensional imaging with a metasurface depth camera

Deep Learning Design for Multiwavelength Infrared Image Sensors Based on Dielectric Freeform Metasurface

Monocular metasurface camera for passive single-shot 4D imaging

Computational metaoptics for imaging

Neural-Optic Co-Designed Polarization-Multiplexed Metalens for Compact Computational Spectral Imaging

Deep-learning-assisted inverse design of polarization-multiplexed structural color filters with ultrahigh saturation based on all-dielectric metasurface

Real-time machine learning-enhanced hyperspectro-polarimetric imaging via an encoding metasurface

Single-Shot Simultaneous Intensity, Phase, and Polarization Imaging with Metasurface

Inverse Design and Experimental Verification of a Bianisotropic Metasurface Using Optimization and Machine Learning