(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211546437.6 (22)申请日 2022.12.05 (71)申请人 香港中文大 学 （深圳） 地址 518172 广东省深圳市龙岗区龙翔大 道2001号 (72)发明人 张昭宇　李文烨　李任杰　俞跃耀　 (74)专利代理 机构 深圳市君胜知识产权代理事 务所(普通 合伙) 44268 专利代理师 谢松 (51)Int.Cl. G02B 1/00(2006.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 基于自注意力的深度神经网络编码光子晶 体的方法 (57)摘要 本发明公开了一种基于自注意力的深度神 经网络编码光子晶体的方法， 提出了POViT模型， 并将其应用到编码光子晶体； 方法包括步骤： 获 取光子晶体的几何结构参数图像； 光子晶体具有 若干个空气孔， 其几何结构参数图像的每个像素 包括： 空气孔的位置和半径； 对几何结构参数图 像进行维度重塑， 得到若干个补丁图像； 将补丁 图像输入嵌入模块和位置编码模块， 得到符号序 列； 将符号序列输入tran sformer编码 模块， 得到 编码特征； 将编码特征输入全连接层模块中， 得 到品质因子Q和模式体积V。 POViT应用自注意力 Transformer模型到光电设计领域， 提高了预测 光子晶体的Q因子和模式体积V的速度和准确度。 权利要求书2页 说明书9页 附图6页 CN 115542433 A 2022.12.30 CN 115542433 A 1.一种基于自注意力的深度神经网络编码光子晶体的方法， 其特征在于， 将POViT模型 应用到编码光子晶体， 所述POViT模型包括： 嵌入模块、 位置编码模块、 transformer编码模 块以及全连接层模块； 所述方法包括 步骤： 获取光子晶体的几何结构参数图像； 其中， 所述光子晶体具有若干个空气孔， 若干个空 气孔排列形成周期性空气孔阵列， 所述几何结构参数图像的每个像素包括： 空气孔的位置 和半径； 对所述几何结构参数图像进行维度重塑， 得到若干个补丁图像； 将所述补丁图像输入所述嵌入模块和位置编码模块， 得到符号序列； 其中， 所述符号序 列按照补丁图像在几何结构参数图像中对应的位置排序； 将所述符号序列输入所述t ransformer编码模块， 得到编码特 征； 将所述编码特 征输入所述全连接层模块， 得到品质因子Q和模式体积V。 2.根据权利要求1所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 所述transformer编码模块包括： 若干个编码Block， 每个编码Block包括： 第一标准化 层、 多头自注 意力层、 第一Dr opout层、 第二标准化层、 MLP模块以及第二Dr opout层； 其中， 所 述MLP模块包括： 第一线性层、 ABS激活层、 第三Dropout层、 第二线性层以及第四Dropout层。 3.根据权利要求2所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 采用Adam优化器进行训练， 学习率 为0.0001~0.01。 4.根据权利要求1所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 对所述几何结构参数图像进行维度重塑， 得到若干个补丁图像， 包括： 根据空气孔的位置和半径， 以及空气孔的初始位置和初始半径， 确定空气孔的 x轴偏移 量图像、y轴偏移量图像以及半径偏移量图像； 当所述空气孔的 x轴偏移量图像、 y轴偏移量图像以及 半径偏移量图像的大小无法被补 丁大小整除时， 对所述空气孔的 x轴偏移量图像、 y轴偏移量图像以及半径偏移量图像补充 维度； 当所述空气孔的 x轴偏移量图像、 y轴偏移量图像以及 半径偏移量图像的大小被补丁大 小整除时， 将所述空气孔的 x轴偏移量图像、y轴偏移量图像以及半径偏移 量图像， 按照预设 高度和预设宽度分割成若干个补丁图像。 5.根据权利要求4所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 所述几何结构参数图像中有54个空气孔， 且分成5排； 所述初始半径 为89.6nm， 所述初 始位置为相邻孔中心 距为320nm的孔 位置； 所述补丁图像的数量 为18。 6.根据权利要求1所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 在收集数据时， 按照高斯分布随机改变空气孔的初始位置和初始半径， 得到训练数 据。 7.根据权利要求1所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 所述POViT模型基于 MSE损失、 预测误差以及相关系数进行评估。 8.根据权利要求7所述的基于自注意力的深度神经网络编码光子晶体的方法， 其特征 在于， 所述MSE损失为：权　利　要　求　书 1/2 页 2 CN 115542433 A 2其中， MSE表示MSE损失， ti表示第i个目标输出， pi表示第i个预测输出， n表示训练样本 的数量， ∑表示 求和符号； 所述预测误差为： 其中， 表示预测误差； 所述相关系数为： 其中， 表示相关系数， 表示数学期望， 表示目标输出的平均值， 表示预 测输出的平均值。 9.一种计算机设备， 包括存储器和处理器， 所述存储器存储有计算机程序， 其特征在 于， 所述处 理器执行所述计算机程序时实现权利要求1至8中任一项所述方法的步骤。 10.一种计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 所述计算机程序 被处理器执行时实现权利要求1至8中任一项所述的方法的步骤。权　利　要　求　书 2/2 页 3 CN 115542433 A 3