Local learning for stable backpropagation-free neural network training towards physical learning
Organizations: Department of Materials Science and Engineering, Delft University of Technology, Mekelweg 2, Delft, 2628CD, The Netherlands · Department of Precision and Microsystems Engineering, Delft University of Technology, Mekelweg 2, Delft, 2628CD, The Netherlands · Department of Intelligent Systems, Delft University of Technology, Van Mourik Broekmanweg 6, Delft 2628XE, The Netherlands
Abstract
While backpropagation and automatic differentiation have driven deep learning's success, the physical limits of chip manufacturing and rising environmental costs of deep learning motivate alternative learning paradigms such as physical neural networks. However, most existing physical neural networks still rely on digital computing for training, largely because backpropagation and automatic differentiation are difficult to realize in physical systems. We introduce FFzero, a forward-only learning framework enabling stable neural network training without backpropagation or automatic differentiation. FFzero combines layer-wise local learning, prototype-based representations, and directional-derivative-based optimization through forward evaluations only. We show that local learning is effective under forward-only optimization, where backpropagation fails. FFzero generalizes to multilayer perceptron and convolutional neural networks across classification and regression. Using a simulated photonic neural network as an example, we demonstrate that FFzero provides a viable path toward backpropagation-free in-situ physical learning.
Figures & tables
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Notation | Value |
| MLP for MNIST classification, MNIST regression, | ||
| and FashionMNIST classification: | ||
| Size of training dataset | - | |
| Size of test dataset | - | |
| Batch size | - | |
| Number of training epochs | - |
| Parameter | Notation | Value |
| CNN for FashionMNIST classification: | ||
| Size of training dataset | - | |
| Size of test dataset | - | |
| Number of training epochs | - | |
| Learning rate | ||
| Step size in DD |