• Advanced Photonics Nexus
  • Vol. 2, Issue 3, 036014 (2023)
Li Pei1, Zeya Xi1, Bing Bai1、2、*, Jianshuai Wang1, Jingjing Zheng1, Jing Li1, and Tigang Ning1
Author Affiliations
  • 1Beijing Jiaotong University, Institute of Lightwave Technology, Key Lab of All Optical Network & Advanced Telecommunication Network of EMC, Beijing, China
  • 2Photoncounts (Beijing) Technology Co. Ltd., Beijing, China
  • show less
    DOI: 10.1117/1.APN.2.3.036014 Cite this Article Set citation alerts
    Li Pei, Zeya Xi, Bing Bai, Jianshuai Wang, Jingjing Zheng, Jing Li, Tigang Ning. Joint device architecture algorithm codesign of the photonic neural processing unit[J]. Advanced Photonics Nexus, 2023, 2(3): 036014 Copy Citation Text show less

    Abstract

    The photonic neural processing unit (PNPU) demonstrates ultrahigh inference speed with low energy consumption, and it has become a promising hardware artificial intelligence (AI) accelerator. However, the nonidealities of the photonic device and the peripheral circuit make the practical application much more complex. Rather than optimizing the photonic device, the architecture, and the algorithm individually, a joint device-architecture-algorithm codesign method is proposed to improve the accuracy, efficiency and robustness of the PNPU. First, a full-flow simulator for the PNPU is developed from the back end simulator to the high-level training framework; Second, the full system architecture and the complete photonic chip design enable the simulator to closely model the real system; Third, the nonidealities of the photonic chip are evaluated for the PNPU design. The average test accuracy exceeds 98%, and the computing power exceeds 100TOPS.
    M=UΣV*,

    View in Article

    yij;k=ij,lKij,klx(sxi+i)(syj+j);l.

    View in Article

    {O1=x1*k1+x2*k2+x3*k11O2=x2*k1+x3*k2+x4*k11O3=x3*k1+x4*k2+x5*k11.

    View in Article

    Li Pei, Zeya Xi, Bing Bai, Jianshuai Wang, Jingjing Zheng, Jing Li, Tigang Ning. Joint device architecture algorithm codesign of the photonic neural processing unit[J]. Advanced Photonics Nexus, 2023, 2(3): 036014
    Download Citation