Efficient training for the hybrid optical diffractive deep neural network

被引:0
|
作者
Fang, Tao [1 ]
Lia, Jingwei [1 ]
Wu, Tongyu [1 ]
Cheng, Ming [1 ]
Dong, Xiaowen [1 ]
机构
[1] Huawei Co LTD, Cent Res Inst, Shenzhen 518129, Peoples R China
来源
关键词
Optical diffractive deep neural network; direct feedback alignment; error-sign-based DFA; direct random target projection;
D O I
10.1117/12.2607567
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
As a new emerging machine learning mechanism, optical diffractive deep neural network ((ODNN)-N-2) has been intensively studied recently due to its incomparable advantages on speed and power efficiency. However, the training process of the (ODNN)-N-2 with traditional back-propagation (BP) method is always time-consuming. Here, we introduce the biologically plausible training methods without feedback to accelerate the training process of the hybrid (ODNN)-N-2. Direct feedback alignment (DFA), error-sign-based DFA (sDFA) and direct random target projection (DRTP) are utilized and evaluated in the training process of the hybrid (ODNN)-N-2 respectively. For the hybrid (ODNN)-N-2 with 20 diffractive layers, about 160x (DFA; CPU), 30x (DFA; GPU), 170x (sDFA; CPU), 32x (sDFA; GPU), 158x (DRTP; CPU) and 32x (DRTP; GPU) accelerations are achieved respectively without significant loss of accuracy, compared with the training process using BP method on CPU or GPU.
引用
收藏
页数:6
相关论文
共 50 条
  • [1] Resampling the optical transmission matrix to accelerate the training of the diffractive deep neural network
    Wu, Lin
    Xia, Rui
    Zhao, Ming
    Xiao, Xi
    He, Zhixue
    Zeng, Yongquan
    Yang, Zhenyu
    Tao, Jin
    [J]. APPLIED OPTICS, 2024, 63 (20) : 5330 - 5337
  • [2] Magneto- optical diffractive deep neural network
    Fujita, Takumi
    Sakaguchi, Hotaka
    Zhang, Jian
    Nonaka, Hirofumi
    Sumi, Satoshi
    Awano, Hiroyuki
    Ishibashi, Takayuki
    [J]. OPTICS EXPRESS, 2022, 30 (20): : 36889 - 36899
  • [3] Hybrid Neural Network for Efficient Training
    Hossain, Md. Billal
    Islam, Sayeed
    Zhumur, Noor-e-Hafsa
    Khanam, Najmoon Nahar
    Khan, Md. Imran
    Kabir, Md. Ahasan
    [J]. 2017 INTERNATIONAL CONFERENCE ON ELECTRICAL, COMPUTER AND COMMUNICATION ENGINEERING (ECCE), 2017, : 528 - 532
  • [4] Optical random phase dropout in a diffractive deep neural network
    Xiao, Yong-Liang
    Li, Sikun
    Situ, Guohai
    You, Zhisheng
    [J]. OPTICS LETTERS, 2021, 46 (20) : 5260 - 5263
  • [5] An Optical Satellite Controller Based on Diffractive Deep Neural Network
    Liu, Shaohua
    Dou, Hongkun
    Li, Hongjue
    Deng, Yue
    [J]. ARTIFICIAL INTELLIGENCE, CICAI 2022, PT II, 2022, 13605 : 46 - 58
  • [6] Diffractive deep neural network adjoint assist or (DNA)2: a fast and efficient nonlinear diffractive neural network implementation
    Idehenre, Ighodalo U.
    Harper, Eric S.
    Mills, Matthew S.
    [J]. OPTICS EXPRESS, 2022, 30 (05) : 7441 - 7456
  • [7] A Memory-Efficient Hybrid Parallel Framework for Deep Neural Network Training
    Li, Dongsheng
    Li, Shengwei
    Lai, Zhiquan
    Fu, Yongquan
    Ye, Xiangyu
    Cai, Lei
    Qiao, Linbo
    [J]. IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 2024, 35 (04) : 577 - 591
  • [8] Memory Efficient Deep Neural Network Training
    Shilova, Alena
    [J]. EURO-PAR 2021: PARALLEL PROCESSING WORKSHOPS, 2022, 13098 : 515 - 519
  • [9] Orthogonality of diffractive deep neural network
    Zheng, Shuiqin
    Xu, Shixiang
    Fan, Dianyuan
    [J]. OPTICS LETTERS, 2022, 47 (07) : 1798 - 1801
  • [10] A Training-Efficient Hybrid-Structured Deep Neural Network With Reconfigurable Memristive Synapses
    Bai, Kangjun
    An, Qiyuan
    Liu, Lingjia
    Yi, Yang
    [J]. IEEE TRANSACTIONS ON VERY LARGE SCALE INTEGRATION (VLSI) SYSTEMS, 2020, 28 (01) : 62 - 75