Ensemble-Based Global Search Framework for the Design Optimization of Fabrication-Constrained Freeform Devices

TL;DR

This paper introduces a Gaussian ensemble gradient descent framework that enables efficient, fully differentiable optimization of freeform devices with fabrication constraints by combining ensemble sampling with variance reduction techniques.

Contribution

It presents a novel ensemble-based optimization method that achieves full differentiability and feasibility in design space for complex freeform devices.

Findings

Enhanced sampling efficiency in high-dimensional spaces

Effective hybrid of exploration and exploitation strategies

Improved optimization of fabrication-constrained designs

Abstract

Although freeform devices with complex internal structures promise drastic increases in performance, the discreteness of the set of available materials presents challenges for gradient-based optimization necessary for the efficient exploration of the high-dimensional freeform parameter space. Several schemes have been devised to utilize a continuous latent parameter space that maps to actual discrete designs, but none thus far simultaneously achieves full differentiability and strictly feasible material choices during optimization. Here, we propose the Gaussian ensemble gradient descent framework, which transforms the piecewise-constant fabrication-constrained cost function by convolution with a Gaussian kernel to render it differentiable. The transformed cost and gradient are estimated through ensemble sampling, which is combined with variance reduction methods that greatly improve the sampling efficiency in high-dimensional parameter spaces. Furthermore, the use of ensemble sampling within a gradient descent framework leads to the effective hybridization of the exploration and exploitation strengths of population- and gradient-based methods, respectively.