[发明专利]一种利用级联深度神经网络的介质超表面逆向设计算法

说明书

本发明公开了一种利用级联深度神经网络的介质超表面逆向设计算法。该方法包括以下几个步骤：步骤1，选择工作波段和超表面响应单元介质材料，随机生成若干个介质超表面响应单元，用时域有限差分的方法计算每个响应单元的电磁响应，形成一一对应的数据库。步骤2，搭建级联的深度神经网络。步骤3，利用包含结构信息的图片作为输入，电磁响应为输出训练级联深度神经网络的正向网络。步骤4，训练自编码器用于包含结构信息图片的降维和特征提取。步骤5，测试级联深度神经网络的性能和泛化能力。本发明可应用于介质超表面响应单元透射谱的快速预测和针对性的逆向设计，大大节省设计的时间成本和计算力。

技术领域

本发明涉及微纳光学和机器学习技术领域，特别涉及一种利用级联深度神经网络的介质超表面逆向设计算法。

背景技术

超表面是一种由亚波长结构构成的二维器件，能够实现对入射光振幅、相位和偏振的任意调制。近年来，超表面已经广泛应用于全息、超透镜和特殊光束生成器等。目前超表面的设计方法依赖于时域有限差分或有限元等的方法。虽然一些商业化的软件已经建立用于超表面响应单元电磁响应的计算，但是响应单元的参数扫描需要耗费大量的时间成本和计算力。此外，超表面的设计和大程度上需要特定的振幅和相位分布，因此超表面响应单元的逆向设计方法亟待解决。目前的一些传统方法如粒子群算法、进化算法和模拟退火算法的逆向设计并不能提供一种有效的解决方案。

深度学习作为一种近年来兴起的机器学习方式，由于计算能力的大幅提升和大数据时代的到来被广泛应用于图像分类、目标检测和语义分割等领域。深度神经网络强大的模型拟合和表达能力有望应用于介质超表面的自动设计。

发明内容

本发明的目的是提供一种利用级联深度神经网络的介质超表面逆向设计算法，有益效果是：提供了一种介质超表面的自动设计方法。

为了达到上述目的，本发明采用的技术方案其特征在于，包括以下几个步骤：

步骤1，选择工作波段和超表面响应单元介质材料，随机生成若干个介质超表面响应单元，用时域有限差分的方法计算每个响应单元的电磁响应，形成一一对应的数据库。

步骤2，搭建级联的深度神经网络。

步骤3，利用包含结构信息的图片作为输入，电磁响应为输出训练级联深度神经网络的正向网络。

步骤4，训练自编码器用于包含结构信息图片的降维和特征提取。

步骤5，测试级联的深度神经网络的性能和泛化能力。

步骤1的具体过程如下：

选择可见光为工作波长范围，根据波长范围设计合理的响应单元的衬底和纳米柱材料，应用奎因斯特采样定律和加工条件的约束确定响应单元的周期。在半边周期内随机生成若干个长方体，并以镜像对称的方式映射到另一侧。用时域有限差分的方法计算该结构的电磁响应。重复生成随机结构和计算电磁响应的过程建立数据库，该过程重复次数越多越好，应不小于20000(经验值)次。

步骤1具体过程中所述的电磁响应由于相位会产生2π的突变，所以采用实部和虚部的方式记录。

步骤2中提到的级联深度神经网络应该包括4个组分：模拟器、编码器、解码器和映射器。其中模拟器用于预测超表面响应单元的电磁响应，编码器和解码器用于对超表面响应单元结构的编码和特征向量的提取，映射器和解码器的组合实现对超表面响应单元结构的预测。

步骤3中提到的正向网络包括6个卷积层和2个全连接层，每个卷积层包含一个卷积操作、一个激活函数(Leaky ReLU)和一个Batch-Norm层。其中第1、3和5个卷积层的步长为1，第2、4和6个卷积层的步长为2。全连接层的激活函数为双曲正切函数。训练过程的损失函数定义为：