Department of Electronic Engineering, Tsinghua University, Beijing, China.
Beijing National Research Center for Information Science and Technology, Tsinghua University, Beijing, China.
Nature. 2024 Aug;632(8024):280-286. doi: 10.1038/s41586-024-07687-4. Epub 2024 Aug 7.
Optical computing promises to improve the speed and energy efficiency of machine learning applications. However, current approaches to efficiently train these models are limited by in silico emulation on digital computers. Here we develop a method called fully forward mode (FFM) learning, which implements the compute-intensive training process on the physical system. The majority of the machine learning operations are thus efficiently conducted in parallel on site, alleviating numerical modelling constraints. In free-space and integrated photonics, we experimentally demonstrate optical systems with state-of-the-art performances for a given network size. FFM learning shows training the deepest optical neural networks with millions of parameters achieves accuracy equivalent to the ideal model. It supports all-optical focusing through scattering media with a resolution of the diffraction limit; it can also image in parallel the objects hidden outside the direct line of sight at over a kilohertz frame rate and can conduct all-optical processing with light intensity as weak as subphoton per pixel (5.40 × 10- operations-per-second-per-watt energy efficiency) at room temperature. Furthermore, we prove that FFM learning can automatically search non-Hermitian exceptional points without an analytical model. FFM learning not only facilitates orders-of-magnitude-faster learning processes, but can also advance applied and theoretical fields such as deep neural networks, ultrasensitive perception and topological photonics.
光计算有望提高机器学习应用的速度和能源效率。然而,当前高效训练这些模型的方法受到数字计算机上的仿真限制。在这里,我们开发了一种称为完全前向模式(FFM)学习的方法,它在物理系统上实现了计算密集型的训练过程。因此,大多数机器学习操作都可以在现场高效地并行进行,减轻了数值建模的限制。在自由空间和集成光子学中,我们通过实验演示了具有给定网络规模的最先进性能的光学系统。FFM 学习表明,训练具有数百万个参数的最深层光学神经网络可以达到与理想模型相当的精度。它支持通过散射介质进行全光聚焦,分辨率达到衍射极限;它还可以以超过 1 kHz 的帧率并行地对隐藏在视线之外的物体进行成像,并且可以以低于每个像素 0.001 个光子的光强(室温下每瓦 5.40×10- 次操作/秒的能量效率)进行全光处理。此外,我们证明 FFM 学习可以在没有解析模型的情况下自动搜索非厄米特异常点。FFM 学习不仅可以加速学习过程,而且还可以推进深度学习神经网络、超灵敏感知和拓扑光子学等应用和理论领域的发展。