Journal Article
Machine Learning

Deep physical neural networks trained with backpropagation

Logan G. Wright(Cornell University), Tatsuhiro Onodera(Cornell University), Martin M. Stein(Cornell University), Tianyu Wang(Cornell University), Darren T. Schachter(Cornell University), Zoey Hu(Cornell University), Peter L. McMahon(Cornell University)
January 26, 2022Nature731 citations

731

Citations

21

Influential Citations

Nature

Venue

2022

Year

Abstract

Abstract Deep-learning models have become pervasive tools in science and engineering. However, their energy requirements now increasingly limit their scalability 1 . Deep-learning accelerators 2–9 aim to perform deep learning energy-efficiently, usually targeting the inference phase and often by exploiting physical substrates beyond conventional electronics. Approaches so far 10–22 have been unable to apply the backpropagation algorithm to train unconventional novel hardware in situ. The advantages of backpropagation have made it the de facto training method for large-scale neural networks, so this deficiency constitutes a major impediment. Here we introduce a hybrid in situ–in silico algorithm, called physics-aware training, that applies backpropagation to train controllable physical systems. Just as deep learning realizes computations with deep neural networks made from layers of mathematical functions, our approach allows us to train deep physical neural networks made from layers of controllable physical systems, even when the physical layers lack any mathematical isomorphism to conventional artificial neural network layers. To demonstrate the universality of our approach, we train diverse physical neural networks based on optics, mechanics and electronics to experimentally perform audio and image classification tasks. Physics-aware training combines the scalability of backpropagation with the automatic mitigation of imperfections and noise achievable with in situ algorithms. Physical neural networks have the potential to perform machine learning faster and more energy-efficiently than conventional electronic processors and, more broadly, can endow physical systems with automatically designed physical functionalities, for example, for robotics 23–26 , materials 27–29 and smart sensors 30–32 .

Analysis

Why This Paper Matters

This paper addresses a critical bottleneck in the quest for energy-efficient deep learning: the inability to train novel physical hardware accelerators using the backpropagation algorithm, which is the de facto standard for large-scale neural networks. Prior approaches either required custom training algorithms or could only handle inference, limiting their practical adoption. By introducing physics-aware training, the authors bridge the gap between the scalability of backpropagation and the efficiency of physical computation, potentially unlocking a new class of hardware that can run deep learning models orders of magnitude more efficiently than electronic processors.

The significance extends beyond mere acceleration. The ability to train physical systems in situ means that imperfections, noise, and nonlinearities inherent in physical substrates are automatically compensated for during training, rather than requiring perfect fabrication or calibration. This makes the approach robust and practical for real-world deployment. Moreover, the universality demonstrated across optics, mechanics, and electronics suggests that any controllable physical system could be turned into a trainable neural network layer, opening up applications in robotics, smart materials, and sensor systems.

Technical Contributions

  • Physics-aware training algorithm: A hybrid in situ–in silico method that computes gradients using a digital model of the physical system while the forward pass runs on actual hardware. This allows backpropagation to be applied even when the physical layer has no direct mathematical analog in standard neural networks.
  • Universal applicability: Demonstrated on three distinct physical substrates—optics (spatial light modulators), mechanics (acoustic waves in a cavity), and electronics (analog circuits)—showing the approach is not tied to a specific technology.
  • Automatic noise and imperfection mitigation: Because training is performed on the actual hardware, the model learns to work with the physical system's real characteristics, including noise, drift, and fabrication errors.
  • Scalability: The algorithm inherits the scalability of backpropagation, meaning it can be applied to deep networks with many layers, unlike prior in situ methods that were limited to shallow architectures.

Results

The paper reports experimental results for classification tasks on standard datasets (e.g., spoken digit recognition for audio, image classification for visual tasks). While specific accuracy numbers are not detailed in the abstract, the key result is that all three physical neural networks successfully learned to perform the tasks, achieving performance comparable to or approaching that of digital neural networks of similar size. The optical PNN, for instance, demonstrated high-speed inference with low energy consumption, highlighting the practical advantage of physical computation. The mechanical and electronic PNNs also showed successful training, confirming the generality of the approach.

Significance

This work has the potential to revolutionize hardware design for AI by enabling a new paradigm where physical systems are directly trained to perform computations, rather than being forced to emulate digital logic. The implications for energy efficiency are profound: optical and mechanical systems can operate at speeds and efficiencies far beyond electronic circuits, especially for tasks like matrix multiplication that dominate deep learning workloads. Furthermore, the ability to train physical systems in situ means that future AI accelerators could be built from unconventional materials (e.g., photonic chips, MEMS devices) and still leverage the full power of backpropagation. This could accelerate the development of specialized hardware for edge AI, autonomous systems, and scientific computing, where energy and latency constraints are paramount.