Control-free and efficient integrated photonic neural networks via hardware-aware training and pruning

Tengji Xu, Weipeng Zhang, Jiawei Zhang, Zeyu Luo, Qiarong Xiao, Benshan Wang, Mingcheng Luo, Xingyuan Xu, Bhavin J. Shastri, Paul R. Prucnal, Chaoran Huang

Research output: Contribution to journalArticlepeer-review

10 Scopus citations

Abstract

Integrated photonic neural networks (PNNs) are at the forefront of AI computing, leveraging light's unique properties, such as large bandwidth, lowlatency, and potentially lowpower consumption.Nevertheless, the integrated optical components are inherently sensitive to external disturbances, thermal interference, and various device imperfections, which detrimentally affect computing accuracy and reliability. Conventional solutions use complicated control methods to stabilize optical devices and chip, which result in high hardware complexity and are impractical for large-scale PNNs. To address this, we propose a training approach to enable control-free, accurate, and energy-efficient photonic computing without adding hardware complexity. The core idea is to train the parameters of a physical neural network towards its noise-robust and energy-efficient region. Our method is validated on different integrated PNN architectures and is applicable to solve various device imperfections in thermally tuned PNNs and PNNs based on phase change materials. A notable 4-bit improvement is achieved in micro-ring resonator-based PNNs without needing complex device control or power-hungry temperature stabilization circuits. Additionally, our approach reduces the energy consumption by tenfold. This advancement represents a significant step towards the practical, energy-efficient, and noise-resilient implementation of large-scale integrated PNNs.

Original languageEnglish (US)
Pages (from-to)1039-1049
Number of pages11
JournalOptica
Volume11
Issue number8
DOIs
StatePublished - Aug 2024

All Science Journal Classification (ASJC) codes

  • Electronic, Optical and Magnetic Materials
  • Atomic and Molecular Physics, and Optics

Fingerprint

Dive into the research topics of 'Control-free and efficient integrated photonic neural networks via hardware-aware training and pruning'. Together they form a unique fingerprint.

Cite this