Space-efficient optical computing with an integrated chip diffractive neural network.

Journal: Nature communications
Published Date:

Abstract

Large-scale, highly integrated and low-power-consuming hardware is becoming progressively more important for realizing optical neural networks (ONNs) capable of advanced optical computing. Traditional experimental implementations need N units such as Mach-Zehnder interferometers (MZIs) for an input dimension N to realize typical computing operations (convolutions and matrix multiplication), resulting in limited scalability and consuming excessive power. Here, we propose the integrated diffractive optical network for implementing parallel Fourier transforms, convolution operations and application-specific optical computing using two ultracompact diffractive cells (Fourier transform operation) and only N MZIs. The footprint and energy consumption scales linearly with the input data dimension, instead of the quadratic scaling in the traditional ONN framework. A ~10-fold reduction in both footprint and energy consumption, as well as equal high accuracy with previous MZI-based ONNs was experimentally achieved for computations performed on the MNIST and Fashion-MNIST datasets. The integrated diffractive optical network (IDNN) chip demonstrates a promising avenue towards scalable and low-power-consumption optical computational chips for optical-artificial-intelligence.

Authors

  • H H Zhu
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore.
  • J Zou
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore.
  • H Zhang
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore.
  • Y Z Shi
    National Key Laboratory of Science and Technology on Micro/Nano Fabrication, Department of Micro/Nano Electronics, Shanghai Jiao Tong University, Shanghai, 200240, China.
  • S B Luo
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore.
  • N Wang
  • H Cai
    Institute of Microelectronics, A*STAR (Agency for Science, Technology and Research), Singapore, 138634, Singapore.
  • L X Wan
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore.
  • B Wang
    Shandong Weigao Surgical Robot Company, Weihai, China.
  • X D Jiang
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore. exdjiang@ntu.edu.sg.
  • J Thompson
    Centre for Quantum Technologies, National University of Singapore, Singapore, 117543, Singapore.
  • X S Luo
    Advanced Micro Foundry, 11 Science Park Road, 117685, Singapore, Singapore.
  • X H Zhou
    State Key Joint Laboratory of ESPC, Center for Sensor Technology of Environment and Health, School of Environment, Tsinghua University, Beijing, 100084, China. xhzhou@mail.tsinghua.edu.cn.
  • L M Xiao
    Shanghai Engineering Research Center of Ultra-Precision Optical Manufacturing, School of Information Science and Technology, Fudan University, Shanghai, 200433, China. liminxiao@fudan.edu.cn.
  • W Huang
    Mobile Health Ministry of Education - China Mobile Joint Laboratory, Xiangya Hospital, Central South University, Changsha, China.
  • L Patrick
    Advanced Micro Foundry, 11 Science Park Road, 117685, Singapore, Singapore.
  • M Gu
    Quantum Hub, School of Physical and Mathematical Science, Nanyang Technological University, 50 Nanyang Ave, 639798, Singapore, Singapore. gumile@ntu.edu.sg.
  • L C Kwek
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore. cqtklc@nus.edu.sg.
  • A Q Liu
    Quantum Science and Engineering Centre (QSec), Nanyang Technological University, Singapore, 639798, Singapore. eaqliu@ntu.edu.sg.