• Photonics Research
  • Vol. 11, Issue 6, 1125 (2023)
Yuyao Huang, Tingzhao Fu, Honghao Huang, Sigang Yang, and Hongwei Chen*
Author Affiliations
  • Beijing National Research Center for Information Science and Technology, Department of Electronic Engineering, Tsinghua University, Beijing 100084, China
  • show less
    DOI: 10.1364/PRJ.484662 Cite this Article Set citation alerts
    Yuyao Huang, Tingzhao Fu, Honghao Huang, Sigang Yang, Hongwei Chen. Sophisticated deep learning with on-chip optical diffractive tensor processing[J]. Photonics Research, 2023, 11(6): 1125 Copy Citation Text show less

    Abstract

    Ever-growing deep-learning technologies are making revolutionary changes for modern life. However, conventional computing architectures are designed to process sequential and digital programs but are burdened with performing massive parallel and adaptive deep-learning applications. Photonic integrated circuits provide an efficient approach to mitigate bandwidth limitations and the power-wall brought on by its electronic counterparts, showing great potential in ultrafast and energy-free high-performance computation. Here, we propose an optical computing architecture enabled by on-chip diffraction to implement convolutional acceleration, termed “optical convolution unit” (OCU). We demonstrate that any real-valued convolution kernels can be exploited by the OCU with a prominent computational throughput boosting via the concept of structral reparameterization. With the OCU as the fundamental unit, we build an optical convolutional neural network (oCNN) to implement two popular deep learning tasks: classification and regression. For classification, Fashion Modified National Institute of Standards and Technology (Fashion-MNIST) and Canadian Institute for Advanced Research (CIFAR-4) data sets are tested with accuracies of 91.63% and 86.25%, respectively. For regression, we build an optical denoising convolutional neural network to handle Gaussian noise in gray-scale images with noise level σ=10, 15, and 20, resulting in clean images with an average peak signal-to-noise ratio (PSNR) of 31.70, 29.39, and 27.72 dB, respectively. The proposed OCU presents remarkable performance of low energy consumption and high information density due to its fully passive nature and compact footprint, providing a parallel while lightweight solution for future compute-in-memory architecture to handle high dimensional tensors in deep learning.
    O[i,j]=m=n=I[im,jn]·K[m,n],

    View in Article

    O^=K^·I^=[k1k2kH2]T·[i11i12i1,G2i21i22i2,G2iH2,1iG2,2iH2,G2],

    View in Article

    w2,v(l)=λ2π(n1n2)·Δϕv(l),

    View in Article

    fu,v(l)=1jλ·(1+cosθu,v2ru,v)·exp(j2πru,vn1λ)·ηexp(jΔψ).

    View in Article

    T(l)=[exp(jΔϕ1(l))000exp(jΔϕ2(l))000exp(jΔϕV(l))].

    View in Article

    ROCU={F(M)T(M1)[l=1M2(F(l+1)T(l))]F(1)}·I^.

    View in Article

    RBPD=κ{R1R1*R2R2*}.

    View in Article

    J=12·i=1G2RBPD(ΔΦ)[i]R^[i]2,

    View in Article

    ΔΦ*=argminΔΦJ(ΔΦ).

    View in Article

    Oconv=2·H21OPs,Okernel=C·OconvOPs.

    View in Article

    Socu=Oconv·rOPS,Sock=C·SocuOPS.

    View in Article

    Emod=H2·(B·D·C·Eb),Edet=Pd·(B/r),Eocu=Emod+Edet.

    View in Article

    Operation density=Operation number  (OPs)Total area of photonic chip  (mm2),

    View in Article

    Power efficiency=Computation throughput  (TOPS)Power consumption  (W).

    View in Article

    Yuyao Huang, Tingzhao Fu, Honghao Huang, Sigang Yang, Hongwei Chen. Sophisticated deep learning with on-chip optical diffractive tensor processing[J]. Photonics Research, 2023, 11(6): 1125
    Download Citation