论文标题

POVIT:视觉变压器,用于多目标设计和纳米光设备的表征

POViT: Vision Transformer for Multi-objective Design and Characterization of Nanophotonic Devices

论文作者

Chen, Xinyu, Li, Renjie, Yu, Yueyao, Shen, Yuanwen, Li, Wenye, Zhang, Zhaoyu, Zhang, Yin

论文摘要

我们解决了半导体IC设计中的基本挑战:纳米级光子设备的快速和准确表征。就像AI和EDA之间的融合一样,已经做出了许多努力,将DNN(例如卷积神经网络(CNN))应用于原型,并表征了在光子综合电路(PIC)和激光雷达中常见的下一代光电设备。这些先前的作品通常努力预测具有超高准确性和速度的光子晶体的质量因子(q)和模态体积(v)。但是,最新的模型仍然远非直接适用于现实世界中:例如V($ v_ {coeff} $)的相关系数仅约80%,远低于生成可靠且可重现的纳米光设计所需的。最近,基于注意力的变压器模型吸引了广泛的兴趣,并广泛用于简历和NLP中。在这项工作中,我们提出了有史以来第一个变压器模型(POVIT),以有效地设计和模拟具有多个目标的半导体光子设备。与标准视觉变压器(VIT)不同,我们提供了光子晶体作为数据输入,并将激活层从GELU更改为绝对值函数(ABS)。我们的实验表明,POVIT超过了先前模型报告的结果。相关系数$ v_ {coeff} $增加了12%以上(即,达到92.0%),Q的预测错误通过数量级降低,包括其他几项关键指标改进。我们的工作有可能推动EDA扩展到完全自动化的光子设计。完整的数据集和代码将被发布,以帮助研究人员在物理和计算机科学的跨学科领域努力。

We solve a fundamental challenge in semiconductor IC design: the fast and accurate characterization of nanoscale photonic devices. Much like the fusion between AI and EDA, many efforts have been made to apply DNNs such as convolutional neural networks (CNN) to prototype and characterize next-gen optoelectronic devices commonly found in photonic integrated circuits (PIC) and LiDAR. These prior works generally strive to predict the quality factor (Q) and modal volume (V) of for instance, photonic crystals, with ultra-high accuracy and speed. However, state-of-the-art models are still far from being directly applicable in the real-world: e.g. the correlation coefficient of V ($V_{coeff}$ ) is only about 80%, which is much lower than what it takes to generate reliable and reproducible nanophotonic designs. Recently, attention-based transformer models have attracted extensive interests and been widely used in CV and NLP. In this work, we propose the first-ever Transformer model (POViT) to efficiently design and simulate semiconductor photonic devices with multiple objectives. Unlike the standard Vision Transformer (ViT), we supplied photonic crystals as data input and changed the activation layer from GELU to an absolute-value function (ABS). Our experiments show that POViT exceeds results reported by previous models significantly. The correlation coefficient $V_{coeff}$ increases by over 12% (i.e., to 92.0%) and the prediction errors of Q is reduced by an order of magnitude, among several other key metric improvements. Our work has the potential to drive the expansion of EDA to fully automated photonic design. The complete dataset and code will be released to aid researchers endeavoring in the interdisciplinary field of physics and computer science.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源