ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
731
Citations
21
Influential Citations
Nature
Venue
2022
Year
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 .
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.
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.
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.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba