Scalable Photonic Neural Networks via Surrogate Scattering-Matrix Inverse Design

TL;DR

This paper presents a scalable method for designing optical neural networks using surrogate inverse design, significantly reducing simulation costs and enabling efficient training of compact photonic processors.

Contribution

The authors introduce a two-stage surrogate workflow and a banded-router architecture that decouple task learning from electromagnetic realization, improving efficiency and scalability.

Findings

Achieved near-accurate all-optical classification on MedMNIST after 20 epochs.

Improved test accuracy by over 15 percentage points on RSSCN7 with the new architecture.

Validated the framework on nonlinear decision tasks like Yin-Yang.

Abstract

Inverse-designed nanophotonic media are a promising platform for compact optical neural networks, but training them end to end is expensive because each adjoint iteration couples the full-wave solver to the dataset minibatch, so the number of electromagnetic simulations scales with both the network depth and the batch size. We introduce a two-stage surrogate workflow that decouples task learning from electromagnetic realization. In the first stage, the trainable optical block is represented as a passive complex matrix with bounded singular values and the classification task is solved directly in matrix space at negligible cost. In the second stage, the selected target operator is transferred to a fabrication-aware freeform device through an adjoint problem driven by a Frobenius-norm transmission residual and a reflection penalty, which removes the minibatch dependence from the full-wave loop and yields a smoother loss landscape than intensity-domain cross-entropy. We further introduce a banded-router architecture composed with a fixed evanescent-coupling region, which exploits the bandwidth-additive property of matrix products to realize dense effective operators within a design region roughly half as long as a fully local router would require. The framework is validated on three tasks. On MedMNIST, the realized all-optical classifier reproduces the surrogate accuracy within $0.6$ percentage points after only 20 adjoint epochs. On RSSCN7, the banded router plus evanescent stage improves test accuracy by more than 15 percentage points over a linear readout baseline. A Yin-Yang task confirms that the same framework supports nonlinear decision boundaries. These results indicate that surrogate-guided inverse design is a practical route to training compact photonic processors with simulation budgets orders of magnitude smaller than direct geometry-to-task pipelines.