Automated multi-layer optical design via deep reinforcement learning

被引:53
作者
Wang, Haozhu [1 ]
Zheng, Zeyu [1 ]
Ji, Chengang [2 ]
Jay Guo, L. [1 ]
机构
[1] Univ Michigan, Dept Elect Engn & Comp Sci, Ann Arbor, MI 48109 USA
[2] Ningbo Inlight Technol Co Ltd, Ningbo, Zhejiang, Peoples R China
来源
MACHINE LEARNING-SCIENCE AND TECHNOLOGY | 2021年 / 2卷 / 02期
关键词
reinforcement learning; optical design; optimization; OPTIMIZATION; TEMPERATURE;
D O I
10.1088/2632-2153/abc327
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:11
相关论文
共 43 条
[1]  
Achiam Joshua, 2018, Spinning up in deep reinforcement learning
[2]   Broadband optical absorption enhancement through coherent light trapping in thin-film photovoltaic cells [J].
Agrawal, Mukul ;
Peumans, Peter .
OPTICS EXPRESS, 2008, 16 (08) :5385-5396
[3]  
Angermueller C, 2020, INT C LEARN REPR
[4]  
[Anonymous], 2013, GENERATING SEQUENCES
[5]  
Bello I., 2017, 5 INT C LEARN REPR I
[6]  
Byrnes S. J, 2016, arXiv preprint arXiv:1603.02720
[7]  
Chen XY, 2019, ADV NEUR IN, V32
[8]  
Chung J., 2014, ARXIV
[9]  
Dai HJ, 2017, ADV NEUR IN, V30
[10]  
Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1