用新模型提升复杂光子器件仿真精度,误差降73%且更快
PACE: Pacing Operator Learning to Accurate Optical Field Simulation for Complicated Photonic Devices
- 设计跨轴分解的PACE算子,长距离建模连接局部结构与全局场
- 单模型误差比顶尖方法低73%,参数少50%,两阶段更适复杂场景
- 适合光子器件设计、仿真加速的科研与工程人员使用
电磁场仿真在光子器件与电路的设计、优化和验证中至关重要,但数值仿真计算成本高,严重制约可扩展性和设计效率。神经算子提供了替代方案,但现有最先进方法NeurOLight在真实复杂光子器件上仍难以实现高保真场预测,最佳报告误差为0.38(归一化均绝对误差)。复杂的光-物质相互作用(如散射、共振)、对局部结构的敏感性、全域模拟中学习复杂度非均匀分布以及丰富的频率信息,导致现有神经偏微分方程求解器失效。本文针对上述挑战,提出新型交叉轴分解的PACE算子,具备强大的长距离建模能力,能将全局场模式与局部器件结构有效关联。受人类学习启发,将极难任务拆分为两阶段渐进式求解:第一阶段模型学习初始解,第二阶段模型进行精细化修正。在多个复杂光子器件基准测试中,单一PACE模型实现误差降低73%、参数减少50%的效果;两阶段设置进一步提升对更复杂情况的高保真模拟能力。运行时方面,相较scipy数值求解器,速度提升154–577倍;相比高度优化的pardiso求解器,提速11.8–12倍。代码与数据集已开源。
原文摘要 · Abstract (English)
Electromagnetic field simulation is central to designing, optimizing, and validating photonic devices and circuits. However, costly computation associated with numerical simulation poses a significant bottleneck, hindering scalability and turnaround time in the photonic circuit design process. Neural operators offer a promising alternative, but existing SOTA approaches, NeurOLight, struggle with predicting high-fidelity fields for real-world complicated photonic devices, with the best reported 0.38 normalized mean absolute error in NeurOLight. The inter-plays of highly complex light-matter interaction, e.g., scattering and resonance, sensitivity to local structure details, non-uniform learning complexity for full-domain simulation, and rich frequency information, contribute to the failure of existing neural PDE solvers. In this work, we boost the prediction fidelity to an unprecedented level for simulating complex photonic devices with a novel operator design driven by the above challenges. We propose a novel cross-axis factorized PACE operator with a strong long-distance modeling capacity to connect the full-domain complex field pattern with local device structures. Inspired by human learning, we further divide and conquer the simulation task for extremely hard cases into two progressively easy tasks, with a first-stage model learning an initial solution refined by a second model. On various complicated photonic device benchmarks, we demonstrate one sole PACE model is capable of achieving 73% lower error with 50% fewer parameters compared with various recent ML for PDE solvers. The two-stage setup further advances high-fidelity simulation for even more intricate cases. In terms of runtime, PACE demonstrates 154-577x and 11.8-12x simulation speedup over numerical solver using scipy or highly-optimized pardiso solver, respectively. We open sourced the code and dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。