• Photonics Research
  • Vol. 8, Issue 6, 940 (2020)
Tiankuang Zhou1、2、3、†, Lu Fang2、3、†, Tao Yan1、2, Jiamin Wu1、2, Yipeng Li1、2, Jingtao Fan1、2, Huaqiang Wu4、5, Xing Lin1、2、4、7、*, and Qionghai Dai1、2、6、8、*
Author Affiliations
  • 1Department of Automation, Tsinghua University, Beijing 100084, China
  • 2Institute for Brain and Cognitive Science, Tsinghua University, Beijing 100084, China
  • 3Tsinghua Shenzhen International Graduate School, Tsinghua University, Shenzhen 518055, China
  • 4Beijing Innovation Center for Future Chip, Tsinghua University, Beijing 100084, China
  • 5Institute of Microelectronics, Tsinghua University, Beijing 100084, China
  • 6Beijing National Research Center for Information Science and Technology, Tsinghua University, Beijing 100084, China
  • 7e-mail: lin-x@tsinghua.edu.cn
  • 8e-mail: qhdai@tsinghua.edu.cn
  • show less
    DOI: 10.1364/PRJ.389553 Cite this Article Set citation alerts
    Tiankuang Zhou, Lu Fang, Tao Yan, Jiamin Wu, Yipeng Li, Jingtao Fan, Huaqiang Wu, Xing Lin, Qionghai Dai. In situ optical backpropagation training of diffractive optical neural networks[J]. Photonics Research, 2020, 8(6): 940 Copy Citation Text show less

    Abstract

    Training an artificial neural network with backpropagation algorithms to perform advanced machine learning tasks requires an extensive computational process. This paper proposes to implement the backpropagation algorithm optically for in situ training of both linear and nonlinear diffractive optical neural networks, which enables the acceleration of training speed and improvement in energy efficiency on core computing modules. We demonstrate that the gradient of a loss function with respect to the weights of diffractive layers can be accurately calculated by measuring the forward and backward propagated optical fields based on light reciprocity and phase conjunction principles. The diffractive modulation weights are updated by programming a high-speed spatial light modulator to minimize the error between prediction and target output and perform inference tasks at the speed of light. We numerically validate the effectiveness of our approach on simulated networks for various applications. The proposed in situ optical learning architecture achieves accuracy comparable to in silico training with an electronic computer on the tasks of object classification and matrix-vector multiplication, which further allows the diffractive optical neural network to adapt to system imperfections. Also, the self-adaptive property of our approach facilitates the novel application of the network for all-optical imaging through scattering media. The proposed approach paves the way for robust implementation of large-scale diffractive neural networks to perform distinctive tasks all-optically.
    Uk=MkWkUk1,(1)

    View in Article

    UN+1=WN+1(k=N1MkWk)U0,(2)

    View in Article

    O=|UN+1|2=|WN+1(k=N1MkWk)U0|2.(3)

    View in Article

    Lϕk=LUN+1UN+1ϕk+LUN+1*UN+1*ϕk=2Re{(LOUN+1*)TUN+1ϕk},(4)

    View in Article

    UN+1/ϕk=jWN+1(i=Nk+1MiWi)diag((i=k1MiWi)U0)=(j·diag((i=k1MiWi)U0)T((i=k+1NWiTMi)WN+1T))T,(5)

    View in Article

    Lϕk=2Re{ETUN+1/ϕk}=2Re{j((i=k1MiWi)U0)((i=k+1NWiTMi)WN+1TE)}T=2Re{(jPkfPkb)}T,(6)

    View in Article

    {Pkf=(i=k1MiWi)U0,Pkb=(i=k+1NWiTMi)WN+1TE.(7)

    View in Article

    {θk=arctan((I3I1)/(I0I2)),Ak=α((I1I0)/2(sinθkcosθk)),(8)

    View in Article

    t=max{8/Fs,1/Fm},(A1)

    View in Article

    E=R/Pt=16NM2(2log2M+3)/Pt.(A2)

    View in Article

    O=|UN+1|2=|WN+1(k=N1fkWkMkWk12)U0|2,(A3)

    View in Article

    UN+1/ϕk=(jdiag(MkWk12(i=k11fiWiMiWi12)U0)T(WkTdiag(gk)(i=k+1NWi12TMiWiTdiag(gi))WN+1T))T,(A4)

    View in Article

    {Pkf=MkWk12(i=k11fiWiMiWi12)U0,Pkb=WkTdiag(gk)(i=k+1NWi12TMiWiTdiag(gi))WN+1TE.(A5)

    View in Article

    {PkfNL=WkMkWk12(i=k11fiWiMiWi12)U0,PkbNL=(i=k+1NWi12TMiWiTdiag(gi))WN+1TE.(A6)

    View in Article

    f(Ein)=Einejπ|Ein|21+|Ein|2.(A7)

    View in Article

    Tiankuang Zhou, Lu Fang, Tao Yan, Jiamin Wu, Yipeng Li, Jingtao Fan, Huaqiang Wu, Xing Lin, Qionghai Dai. In situ optical backpropagation training of diffractive optical neural networks[J]. Photonics Research, 2020, 8(6): 940
    Download Citation