Multi-Level Encoding and Decoding in a Scalable Photonic Tensor Processor With a Photonic General Matrix Multiply (GeMM) Compiler

The resurgence of artificial intelligence enabled by deep learning and high performance computing has seen a dramatic increase of demand in the accuracy of deep learning model which has come at the cost of computational complexity. The fundamental operations in deep learning models are matrix multiplications, and large scale matrix operations and data-centric tasks have experienced bottlenecks from current digital electronic hardware in terms of performance and scalability. Recent research on photonic processors have found solutions to enable applications in machine learning, neuromorphic computing and high performance computing using basic photonic processing elements on integrated silicon photonic platform. However, efficient and scalable photonic computing requires an information encoding/decoding scheme. Here, we propose a multi-level encoding and decoding scheme, and experimentally demonstrate it with a wavelength-multiplexed silicon photonic processor. We also discuss the scalability of our proposed scheme by introducing a photonic general matrix multiply compiler, and consider the effects of speed, bit precision, and noise. Our proposed scheme could be adapted to a variety of photonic information processing architectures for photonic neural networks, photonics tensor cores, and programmable photonic.

[1]  A. Tait Quantifying power use in silicon photonic neural networks , 2021, Physical Review Applied.

[2]  P. Prucnal,et al.  Silicon microring synapses enable photonic deep learning beyond 9-bit precision , 2021, Optica.

[3]  L. Chrostowski,et al.  Scaling Up Silicon Photonic-based Accelerators: Challenges and Opportunities, and Roadmapping with Silicon Photonics 2.0 , 2021, ArXiv.

[4]  Sudip Shekhar,et al.  Photonic pattern reconstruction enabled by on-chip online learning and inference , 2021 .

[5]  Xuan Li,et al.  Parallel convolutional processing using an integrated photonic tensor core , 2021, Nature.

[6]  Bhavin J. Shastri,et al.  Photonics for artificial intelligence and neuromorphic computing , 2020, Nature Photonics.

[7]  Binhao Wang,et al.  Fully-Integrated Heterogeneous DML Transmitters for High-Performance Computing , 2020, Journal of Lightwave Technology.

[8]  Bhavin J. Shastri,et al.  Demonstration of scalable microring weight bank control for large-scale photonic integrated circuits , 2020 .

[9]  Ling Shao,et al.  Editorial: Special Issue on Machine Vision with Deep Learning , 2020, International Journal of Computer Vision.

[10]  Mario Miscuglio,et al.  Photonic tensor cores for machine learning , 2020, Applied Physics Reviews.

[11]  Paul R. Prucnal,et al.  Digital Electronics and Analog Photonics for Convolutional Neural Networks (DEAP-CNNs) , 2019, IEEE Journal of Selected Topics in Quantum Electronics.

[12]  J. Feldmann,et al.  All-optical spiking neurosynaptic networks with self-learning capabilities , 2019, Nature.

[13]  Paul R. Prucnal,et al.  Silicon Photonic Modulator Neuron , 2018, Physical Review Applied.

[14]  Philip Y. Ma,et al.  Feedback control for microring weight banks. , 2018, Optics express.

[15]  Dong Pan,et al.  56GHz Waveguide Ge/Si Avalanche Photodiode , 2018, 2018 Optical Fiber Communications Conference and Exposition (OFC).

[16]  Jeffrey S. Vetter,et al.  NVIDIA Tensor Core Programmability, Performance & Precision , 2018, 2018 IEEE International Parallel and Distributed Processing Symposium Workshops (IPDPSW).

[17]  Somaia Awad Hassan,et al.  Effective Implementation of Matrix-Vector Multiplication on Intel's AVX multicore Processor , 2018, Comput. Lang. Syst. Struct..

[18]  P. Prucnal,et al.  Neuromorphic photonic networks using silicon photonic weight banks , 2017, Scientific Reports.

[19]  Aref Meddeb,et al.  Driver information system: a combination of augmented reality, deep learning and vehicular Ad-hoc networks , 2017, Multimedia Tools and Applications.

[20]  David A. Patterson,et al.  In-datacenter performance analysis of a tensor processing unit , 2017, 2017 ACM/IEEE 44th Annual International Symposium on Computer Architecture (ISCA).

[21]  G. Roelkens,et al.  100-Gbps RZ Data Reception in 67-GHz Si-Contacted Germanium Waveguide p-i-n Photodetectors , 2017, Journal of Lightwave Technology.

[22]  Dirk Englund,et al.  Deep learning with coherent nanophotonic circuits , 2017, 2017 Fifth Berkeley Symposium on Energy Efficient Electronic Systems & Steep Transistors Workshop (E3S).

[23]  P. Prucnal,et al.  NEUROMORPHIC PHOTONICS , 2017 .

[24]  Yi Tay,et al.  Deep Learning based Recommender System: A Survey and New Perspectives , 2017, ArXiv.

[25]  Paul R. Prucnal,et al.  Microring Weight Banks , 2016, IEEE Journal of Selected Topics in Quantum Electronics.

[26]  Eugenio Culurciello,et al.  An Analysis of Deep Neural Network Models for Practical Applications , 2016, ArXiv.

[27]  L. Chrostowski,et al.  Wavelength tuning and stabilization of microring-based filters using silicon in-resonator photoconductive heaters. , 2015, Optics express.

[28]  W. Freude,et al.  Connecting Silicon Photonic Circuits to Multicore Fibers by Photonic Wire Bonding , 2015, Journal of Lightwave Technology.

[29]  Chetan Jhurani,et al.  A GEMM interface and implementation on NVIDIA GPUs for multiple small matrices , 2013, J. Parallel Distributed Comput..

[30]  Michal Lipson,et al.  Power insensitive silicon microring resonators. , 2012, Optics letters.

[31]  P. Dumon,et al.  Silicon microring resonators , 2012 .

[32]  John E. Heebner,et al.  Optical Microresonators: Theory, Fabrication, and Applications , 2007 .

[33]  Jack J. Dongarra,et al.  A set of level 3 basic linear algebra subprograms , 1990, TOMS.

[34]  H J Shaw,et al.  All-single-mode fiber resonator. , 1982, Optics letters.