The finite-difference time-domain (FDTD) method, which is important in
photonic hardware design flow, is widely adopted to solve time-domain Maxwell
equations. However, FDTD is known for its prohibitive runtime cost, taking
minutes to hours to simulate a single device. Recently, AI has been applied to
realize orders-of-magnitude speedup in partial differential equation (PDE)
solving. However, AI-based FDTD solvers for photonic devices have not been
clearly formulated. Directly applying off-the-shelf models to predict the
optical field dynamics shows unsatisfying fidelity and efficiency since the
model primitives are agnostic to the unique physical properties of Maxwell
equations and lack algorithmic customization. In this work, we thoroughly
investigate the synergy between neural operator designs and the physical
property of Maxwell equations and introduce a physics-inspired AI-based FDTD
prediction framework PIC2O-Sim which features a causality-aware dynamic
convolutional neural operator as its backbone model that honors the space-time
causality constraints via careful receptive field configuration and explicitly
captures the permittivity-dependent light propagation behavior via an efficient
dynamic convolution operator. Meanwhile, we explore the trade-offs among
prediction scalability, fidelity, and efficiency via a multi-stage partitioned
time-bundling technique in autoregressive prediction. Multiple key techniques
have been introduced to mitigate iterative error accumulation while maintaining
efficiency advantages during autoregressive field prediction. Extensive
evaluations on three challenging photonic device simulation tasks have shown
the superiority of our PIC2O-Sim method, showing 51.2% lower roll-out
prediction error, 23.5 times fewer parameters than state-of-the-art neural
operators, providing 300-600x higher simulation speed than an open-source FDTD
numerical solver.

本研究提出了一种基于物理启发的人工智能有限差分时域（FDTD）预测框架 PIC2O-Sim，通过神经算子设计与麦克斯韦方程的物理特性相结合，并采用多阶段分割时间捆绑技术，提高模拟速度并降低预测误差。对三个具有挑战性的光子器件模拟任务进行了广泛评估，结果显示我们的 PIC2O-Sim 方法具有更低的预测误差、比现有神经算子更少的参数以及比开源 FDTD 数值求解器更高的模拟速度。