arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.02429cs.ETcs.LGphysics.optics
收藏

光子卷积神经网络与预训练的在 situ 训练

Photonic convolutional neural network with pre-trained in situ training

Saurabh Ranjan, Sonika Thakral, Amit Sehgal

AI总结:

本文提出并验证了完全基于光子的卷积神经网络(PCNN),在光学领域实现MNIST图像分类,达到94%的测试准确率,通过全光处理和微环谐振非线性性提升能效。

摘要:

光子计算是一种具有潜力克服电子冯·诺依曼架构能耗瓶颈的计算范式。通过率和能耗是互补金属氧化物半导体(CMOS)芯片的基本限制,因此卷积神经网络(CNN)正在革新机器学习、计算机视觉和其他基于图像的应用。在本工作中,我们提出并验证了完全基于光子的卷积神经网络(PCNN),在光学领域实现MNIST图像分类,达到94%的测试准确率。与现有依赖频繁光-电-光转换(O/E/O)的架构不同,我们的系统利用Mach-Zehnder干涉仪(MZI)网格、波分复用(WDM)池化和基于微环谐振的非线性性进行相干处理。最大池化单元完全在硅光子学上实现,不需要光-电或电转换。为克服训练物理相移器参数的挑战,我们引入了一种混合训练方法,部署了一个数学精确的可微数字双胞胎用于离线反向传播,随后通过同时扰动随机近似(SPSA)算法进行在 situ 微调。我们的评估显示对热交叉talk具有显著的鲁棒性(在严重耦合情况下仅降级0.43%的准确率),并且在单图像推断中比最先进的电子GPU的能效提高了100到242倍。

英文摘要:

Convolutional neural networks (CNNs) have transformed image processing, but the energy consumption and inference latency of electronic based implementations remain fundamental bottlenecks. These limitations have motivated the search for alternative hardware architectures beyond Complementary metal-oxide-semiconductor (CMOS) chips. Optical systems can perform linear matrix operations at the speed of light with extremely low energy dissipation, making them attractive for CNN acceleration. However, building a fully coherent photonic CNN that performs both linear and nonlinear operations and training it efficiently remains an open challenge. Here we present a fully photonic convolutional neural network (PCNN) that executes image classification in the optical domain, including convolution, max-pooling, nonlinear activation, and fully connected layers. The network achieves 94.49 percent accuracy on the MNIST dataset distributed across Mach Zehnder Interferometer (MZI) meshes, weighted Multimode Interferometer (MMI) trees, and a microring resonator based nonlinearity. A mathematically exact differentiable digital twin, enables backpropagation for ex situ pre training, reaches 97.45 percent digital accuracy. Trained phases are transferred one-to-one to the photonic hardware and refined via a gradient free algorithm that estimates the full gradient with only two forward passes. The architecture exhibits inherent robustness to non idealities, under the compound effect of propagation loss, MZI insertion loss, fabrication disorder, and thermal crosstalk. A bottom-up power analysis yields 10.83 W static chip consumption and 843 ns inference latency, translating to 220 to 330 times greater energy efficiency than state of the art electronic GPUs for single-image inference.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑