An on-chip photonic deep neural network for image classification

被引:423
作者
Ashtiani, Farshid [1 ]
Geers, Alexander J. [1 ]
Aflatouni, Firooz [1 ]
机构
[1] Univ Penn, Dept Elect & Syst Engn, Philadelphia, PA 19104 USA
关键词
ACCELERATOR;
D O I
10.1038/s41586-022-04714-0
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Deep neural networks with applications from computer vision to medical diagnosis(1-5) are commonly implemented using clock-based processors(6-14), in which computation speed is mainly limited by the clock frequency and the memory access time. In the optical domain, despite advances in photonic computation(15-17), the lack of scalable on-chip optical non-linearity and the loss of photonic devices limit the scalability of optical deep networks. Here we report an integrated end-to-end photonic deep neural network (PDNN) that performs sub-nanosecond image classification through direct processing of the optical waves impinging on the on-chip pixel array as they propagate through layers of neurons. In each neuron, linear computation is performed optically and the non-linear activation function is realized opto-electronically, allowing a classification time of under 570 ps, which is comparable with a single clock cycle of state-of-the-art digital platforms. A uniformly distributed supply light provides the same per-neuron optical output range, allowing scalability to large-scale PDNNs. Two-class and four-class classification of handwritten letters with accuracies higher than 93.8% and 89.8%, respectively, is demonstrated. Direct, clock-less processing of optical data eliminates analogue-to-digital conversion and the requirement for a large memory module, allowing faster and more energy efficient neural networks for the next generations of deep learning systems.
引用
收藏
页码:501 / +
页数:20
相关论文
共 50 条
[1]  
amd, AMD RAD TM RX 6700 X
[2]   Single-chip nanophotonic near-field imager [J].
Ashtiani, Farshid ;
Risi, Angelina ;
Aflatouni, Firooz .
OPTICA, 2019, 6 (10) :1255-1260
[3]   On the use of deep learning for computational imaging [J].
Barbastathis, George ;
Ozcan, Aydogan ;
Situ, Guohai .
OPTICA, 2019, 6 (08) :921-943
[4]   Programmable photonic circuits [J].
Bogaerts, Wim ;
Perez, Daniel ;
Capmany, Jose ;
Miller, David A. B. ;
Poon, Joyce ;
Englund, Dirk ;
Morichetti, Francesco ;
Melloni, Andrea .
NATURE, 2020, 586 (7828) :207-216
[5]   Salient Object Detection: A Benchmark [J].
Borji, Ali ;
Sihite, Dicky N. ;
Itti, Laurent .
COMPUTER VISION - ECCV 2012, PT II, 2012, 7573 :414-429
[6]   Reinforcement learning in a large-scale photonic recurrent neural network [J].
Bueno, J. ;
Maktoobi, S. ;
Froehly, L. ;
Fischer, I. ;
Jacquot, M. ;
Larger, L. ;
Brunner, D. .
OPTICA, 2018, 5 (06) :756-760
[7]   Hybrid optical-electronic convolutional neural networks with optimized diffractive optics for image classification [J].
Chang, Julie ;
Sitzmann, Vincent ;
Dun, Xiong ;
Heidrich, Wolfgang ;
Wetzstein, Gordon .
SCIENTIFIC REPORTS, 2018, 8
[8]   Global Contrast based Salient Region Detection [J].
Cheng, Ming-Ming ;
Zhang, Guo-Xin ;
Mitra, Niloy J. ;
Huang, Xiaolei ;
Hu, Shi-Min .
2011 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2011, :409-416
[9]   On-chip photonic synapse [J].
Cheng, Zengguang ;
Rios, Carlos ;
Pernice, Wolfram H. P. ;
Wright, C. David ;
Bhaskaran, Harish .
SCIENCE ADVANCES, 2017, 3 (09)
[10]   A 3.7-43.7-GHz Low-Power Consumption Variable Gain Distributed Amplifier in 90-nm CMOS [J].
Chiu, Tzu-Yang ;
Wang, Yunshan ;
Wang, Huei .
IEEE MICROWAVE AND WIRELESS COMPONENTS LETTERS, 2021, 31 (02) :169-172