Achieving efficient inverse design of low-dimensional heterostructures based on a vigorous scalable multi-task learning network

作     者：Du, Shiyin You, Jie Tang, Yuhua Ouyang, Hao Tao, Zilong Jiang, Tian 

作者机构：Natl Univ Def Technol Coll Comp State Key Lab High Performance Comp Changsha 410073 Peoples R China Acad Mil Sci PLA China Def Innovat Inst Beijing 100071 Peoples R China Natl Univ Def Technol Coll Adv Interdisciplinary Studies Changsha 410073 Peoples R China Natl Univ Def Technol Beijing Inst Adv Study Beijing 100020 Peoples R China 

出 版 物：《OPTICS EXPRESS》 (Opt. Express)

年 卷 期：2021年第29卷第13期

页      面：19727-19742页

核心收录：

学科分类：070207[理学-光学] 07[理学] 08[工学] 0803[工学-光学工程] 0702[理学-物理学] 

基　　金：National Natural Science Foundation of China [62075240, 11902358, 41904167] Distinguished Young Scholar Foundation of Hunan Province [2020JJ2036] 

主　　题：Computation methods Light scattering Nonlinear optical devices Optical absorption Plasmon waveguides Surface plasmon polaritons 

摘      要：A scalable multi-task learning (SMTL) model is proposed for the efficient inverse design of low-dimensional heterostructures and the prediction of their optical response. Specifically, several types of nanostructures, including single and periodic graphene-Si heterostructures consisting of nxn graphene squares (n=1 similar to 9), 1D periodic graphene ribbons, 2D arrays of graphene squares, pure Si cubes and their periodic array counterparts, are investigated using both traditional finite element method and SMTL network, with the former providing training data (optical absorption) for the latter. There are two important algorithms implemented in SMTL model: one is the normalization mechanism that makes different parameters of different structures on the same scale, ensuring that SMTL network can deal with tasks with different dataset impartially and without bias;the other one is used to capture the impact of nanostructures dimensions on their optical absorption and thus improve the generalization ability of SMTL. Utilizing SMTL model, we first study the absorption property of the multiple shaped nanostructures and look deeper into the impacts of nxn graphene squares and Si cuboid on the optical absorption of their heterostructures. Equally important, the multi-structure inverse design functionality of SMTL is confirmed in this context, which not only owns high accuracy, fast computational speed, and excellent generalizable ability, but also can be applied to contrive new structures with desired optical response. This work adds to the rapidly expanding field of inverse design in nanophotonics and establishes a multi-task learning framework for heterostructures and more complicated nanoparticles. (C) 2021 Optical Society of America under the terms of the OSA Open Access Publishing Agreement