English

Photonic systolic array for all-optical matrix-matrix multiplication

Optics 2025-11-27 v1

Abstract

Systolic arrays have proven to be highly efficient for parallelized matrix-matrix multiplication (MMM), utilizing synchronized, heartbeat-like data flows across an array of processing elements. While optical structures such as waveguide crossbar arrays and Mach-Zehnder interferometer-based meshes serve as photonic equivalents to the systolic arrays, the disparity between the two input matrices for multiplication -- one using optical signals and the other with system-defined parameters -- gives rise to a bottleneck in modern machine-learning tasks, such as evaluating attention scores in large language models. Here, we propose a photonic systolic array that performs MMM entirely with optical signals, utilizing homodyne detection at each array cell. Adjoint-based design of compact on-chip freeform optical modules enables precise control of light flow without bulky waveguide coupling schemes. The operation of a 4×44\times4 photonic systolic array is numerically verified, achieving a theoretical computation density of 6.2 PMACs/mm2/s6.2~\mathrm{PMACs}/\mathrm{mm}^2/\mathrm{s}. This design marks a significant step toward practical photonic computing hardware for modern AI workloads.

Keywords

Cite

@article{arxiv.2410.21671,
  title  = {Photonic systolic array for all-optical matrix-matrix multiplication},
  author = {Jungmin Kim and Qingyi Zhou and Zongfu Yu},
  journal= {arXiv preprint arXiv:2410.21671},
  year   = {2025}
}

Comments

Main 8 pages, 4 figures Supplementary Note S1-2, Supplementary Figures S1-9

R2 v1 2026-06-28T19:39:04.847Z