Optical Neural Ordinary Differential Equations

Yun Zhao,Hang Chen,Min Lin,Haiou Zhang,Tao Yan,Ruqi Huang,Xing Lin,Qionghai Dai
DOI: https://doi.org/10.1364/ol.477713
IF: 3.6
2023-01-01
Optics Letters
Abstract:Increasing the layer number of on-chip photonic neural networks (PNNs) is essential to improve its model perfor-mance. However, the successive cascading of network hidden layers results in larger integrated photonic chip areas. To address this issue, we propose the optical neural ordinary differential equations (ON-ODEs) architecture that parame-terizes the continuous dynamics of hidden layers with optical ODE solvers. The ON-ODE comprises the PNNs followed by the photonic integrator and optical feedback loop, which can be configured to represent residual neural networks (ResNets) and implement the function of recurrent neu-ral networks with effectively reduced chip area occupancy. For the interference-based optoelectronic nonlinear hidden layer, the numerical experiments demonstrate that the single hidden layer ON-ODE can achieve approximately the same accuracy as the two-layer optical ResNets in image classifi-cation tasks. In addition, the ON-ODE improves the model classification accuracy for the diffraction-based all-optical linear hidden layer. The time-dependent dynamics property of ON-ODE is further applied for trajectory prediction with high accuracy. (c) 2023 Optica Publishing Group
What problem does this paper attempt to address?