论文标题
调查具有条件可逆神经网络的多层薄膜的反设计
Investigation of inverse design of multilayer thin-films with conditional invertible Neural Networks
论文作者
论文摘要
目前,使用基于梯度的优化以及可以引入其他薄膜层的方法来解决有关给定目标的光学多层薄膜的任务。最近,已经介绍了深入学习和强化学习的任务,即设计薄膜取得了巨大的成功,但是训练有素的网络通常只能熟练到一个目标,并且如果光学目标变化,则必须进行重新培训。在这项工作中,我们将条件可逆的神经网络(CINN)应用于具有光学目标的多层薄膜。由于cinn了解了训练数据集中所有薄膜配置的能量格局,因此我们表明cinn可以生成薄膜配置的随机组合,该组合仅根据随机变量而相当接近所需的目标。通过通过局部优化进一步完善所提出的配置,我们表明,生成的薄膜以比可比的最新技术方法更高的精度到达目标。此外,我们测试了在训练数据分布之外的样品上的生成能力,发现CINN也能够预测分布式靶标的薄膜。结果表明,为了改善薄膜的生成设计,将建立和新的机器学习方法结合起来是有益的,以获得最有利的结果。
The task of designing optical multilayer thin-films regarding a given target is currently solved using gradient-based optimization in conjunction with methods that can introduce additional thin-film layers. Recently, Deep Learning and Reinforcement Learning have been been introduced to the task of designing thin-films with great success, however a trained network is usually only able to become proficient for a single target and must be retrained if the optical targets are varied. In this work, we apply conditional Invertible Neural Networks (cINN) to inversely designing multilayer thin-films given an optical target. Since the cINN learns the energy landscape of all thin-film configurations within the training dataset, we show that cINNs can generate a stochastic ensemble of proposals for thin-film configurations that that are reasonably close to the desired target depending only on random variables. By refining the proposed configurations further by a local optimization, we show that the generated thin-films reach the target with significantly greater precision than comparable state-of-the art approaches. Furthermore, we tested the generative capabilities on samples which are outside the training data distribution and found that the cINN was able to predict thin-films for out-of-distribution targets, too. The results suggest that in order to improve the generative design of thin-films, it is instructive to use established and new machine learning methods in conjunction in order to obtain the most favorable results.