[发明专利]一种基于Transformer的电磁超材料复杂光谱高精度预测方法

说明书

本发明公开了一种基于Transformer的电磁超材料复杂光谱高精度预测方法，预测方法的构建基于Transformer神经网络模型，利用严格耦合波分析法RCWA在给定的电磁超材料物理模型结构参数范围内随机组合计算得到样本数据，将样本数据按4：1分为训练集和验证集，用于神经网络模型的训练和验证，输入训练集至Transformer神经网络模型进行训练，验证集用于验证神经网络模型性能。能根据输入的电磁超材料结构参数准确快速的预测光学响应，克服传统数值模拟方法求解麦克斯韦方程组复杂耗时的缺点，可以做到对光谱实时准确预测，降低时间和硬件成本。能极大加快电磁超材料设计周期，易于推广其它电磁超材料模型中。

技术领域

本发明涉及电磁超材料与人工智能技术领域，具体为一种基于Transformer的电磁超材料复杂光谱高精度预测方法。

背景技术

电磁超材料超是具有微结构的人造复合材料，它们具有天然材料所不具备的非凡物理特性，能够在纳米尺度实现对电磁波相位、振幅和偏振灵活有效的调控。基于电磁超材料的器件具备体积小、灵敏度高、灵活度高等优势，广泛应用于传感、检测、能量存储和热辐射等。

电磁超材料器件尺寸在纳米级别，其制备依赖精密的加工手段，需要在加工前对光学响应进行精确模拟以优化器件结构。时域有限差分法(FDTD)或者有限元法(FEM)是两种模拟电磁超材料光学响应的通用数值计算方法。这些方法往往需要较大的计算时间成本和硬件成本，且随着器件结构复杂度增加，计算过程将变得越来越困难。

随着机器学习的发展，深度学习在语音识别、图像识别、自然语言处理等领域已经展现出其强大的能力，深度学习是一种数据驱动算法，通过大量数据的训练，可以表示数据之间复杂的非线性关系，与数值计算方法相比可以在毫秒级时间内给出计算结果，且计算时间成本和硬件不随器件结构复杂度增加而增加。

深度学习辅助电磁超材料设计引起了不少研究者关注，并且已经成功被应用于电磁学多个领域。目前基于深度学习的电磁超材料设计方案主要有两类网络：1.根据电磁超材料结构参数预测光学响应，即正向网络；2.根据目标光学响应预测电磁超材料结构参数，即逆向网络。正向网络不仅可以替代数值计算方法高效快速的预测光学响应而且对逆向网络的训练至关重要，例如通过训练好的正向网络协助逆向网络来训练以解决光谱和结构参数之间存在的非唯一映射问题。目前常用于正向网络的深度学习模型包括多层感知机(MLP)、卷积神经网络(CNN)、自编码器(AE)等。这些模型在训练样本量充足或光谱复杂度不高的情况下通常具有较好的预测精度。

特别的，电磁超材料训练数据的收集不可避免的会使用到数值计算方法，可获得的数据量有限，这要求神经网络模型要在低样本量的情况下，尽可能多的提升学习能力，充分利用已有训练集。针对复杂的光谱曲线，传统的神经网络模型在光谱变化较大的位置往往具有较大的预测误差，而这些位置往往蕴含着重要的物理信息，提升正向网络对复杂光谱的预测精度是至关重要的。

发明内容

针对现有技术存在的不足，本发明提出了一种基于Transformer的电磁超材料复杂光谱高精度预测方法，旨在解决现有技术对复杂光谱预测一致性不高的问题。

本发明提供了如下的技术方案：一种基于Transformer的电磁超材料复杂光谱高精度预测方法，所述预测方法的构建基于Transformer神经网络模型，利用严格耦合波分析法RCWA在给定的电磁超材料物理模型结构参数范围内随机组合计算得到样本数据，将样本数据按照4：1分为训练集和验证集，用于神经网络模型的训练和验证，输入训练集至Transformer神经网络模型进行训练，验证集用于验证神经网络模型性能，根据训练完成的Transformer神经网络模型预测得到电磁超材料结构参数对应的光学响应。

优选的，使用严格耦合波分析法RCWA在给定的层状金属/电介质结构参数范围内随机组合计算得到对应的反射率，用于Transformer神经网络模型的训练与验证，使Transformer神经网络模型能得到层状金属/电介质结构参数与反射率之间复杂的非线性映射关系。