120 GOPS Photonic Tensor Core in Thin-film Lithium Niobate for Inference and in-situ Training
120 GOPS Photonic Tensor Core in Thin-film Lithium Niobate for Inference and in-situ Training
Photonics offers a transformative approach to artificial intelligence (AI) and neuromorphic computing by enabling low-latency, high-speed, and energy-efficient computations. However, conventional photonic tensor cores face significant challenges in constructing large-scale photonic neuromorphic networks. Here, we propose a fully integrated photonic tensor core, consisting of only two thin-film lithium niobate (TFLN) modulators, a III-V laser, and a charge-integration photoreceiver. Despite its simple architecture, it is capable of implementing an entire layer of a neural network with a computational speed of 120 GOPS, while also allowing flexible adjustment of the number of inputs (fan-in) and outputs (fan-out). Our tensor core supports rapid in-situ training with a weight update speed of 60 GHz. Furthermore, it successfully classifies (supervised learning) and clusters (unsupervised learning) 112 * 112-pixel images through in-situ training. To enable in-situ training for clustering AI tasks, we offer a solution for performing multiplications between two negative numbers.
Wangning Cai、Tianye Wang、Xun Guan、Jingxiang Song、Wei Ke、Wei Shi、Shangxuan Yu、Mohamed Rahim、Yuntao Zhu、Nicolas A. F. Jaeger、Mengyue Xu、Omid Esmaeeli、Sudip Shekhar、Grzegorz Pakulski、Siyuan Yu、Mustafa Hammood、Mohammed Al-Qadasi、Jens Schmid、Yanmei Lin、Lukas Chrostowski、Zhongjin Lin、Xinlun Cai、Pedro Barrios、Arman Safarnejadian、Bhavin J. Shastri、Weihong Jiang、Zibo Zheng、Leslie A. n Rusch、Hugh Morison、Matthew Mitchell
光电子技术计算技术、计算机技术
Wangning Cai,Tianye Wang,Xun Guan,Jingxiang Song,Wei Ke,Wei Shi,Shangxuan Yu,Mohamed Rahim,Yuntao Zhu,Nicolas A. F. Jaeger,Mengyue Xu,Omid Esmaeeli,Sudip Shekhar,Grzegorz Pakulski,Siyuan Yu,Mustafa Hammood,Mohammed Al-Qadasi,Jens Schmid,Yanmei Lin,Lukas Chrostowski,Zhongjin Lin,Xinlun Cai,Pedro Barrios,Arman Safarnejadian,Bhavin J. Shastri,Weihong Jiang,Zibo Zheng,Leslie A. n Rusch,Hugh Morison,Matthew Mitchell.120 GOPS Photonic Tensor Core in Thin-film Lithium Niobate for Inference and in-situ Training[EB/OL].(2023-11-28)[2025-05-31].https://arxiv.org/abs/2311.16896.点此复制
评论