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

摘要

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