Automated multi-layer optical design via deep reinforcement learning. Issue 2 (3rd February 2021)

• Record Type: Journal Article
• Authors: Wang, Haozhu; Zheng, Zeyu; Ji, Chengang; Jay Guo, L
• Is Part Of: Machine learning: science and technology. Volume 2:Issue 2(2021)
• Publication Date: 2021-02-03
• Subjects: reinforcement learning -- optical design -- optimization; 006.31
• DOI: 10.1088/2632-2153/abc327
• Languages: English
• ISSNs: 2632-2153

Abstract

Abstract: Optical multi-layer thin films are widely used in optical and energy applications requiring photonic designs. Engineers often design such structures based on their physical intuition. However, solely relying on human experts can be time-consuming and may lead to sub-optimal designs, especially when the design space is large. In this work, we frame the multi-layer optical design task as a sequence generation problem. A deep sequence generation network is proposed for efficiently generating optical layer sequences. We train the deep sequence generation network with proximal policy optimization to generate multi-layer structures with desired properties. The proposed method is applied to two energy applications. Our algorithm successfully discovered high-performance designs, outperforming structures designed by human experts in task 1, and a state-of-the-art memetic algorithm in task 2.