Backpropagation-free Training of Analog AI Accelerators
Deep learning has achieved remarkable success in diverse fields in recent years. However, this growth presents significant challenges, particularly in terms of energy consumption during both training and inference phases. While there have been efforts to improve energy efficiency during the inference phase, efficient training of deep learning models remains a largely unaddressed challenge. The training method for digital deep learning models typically relies on backpropagation, a process that is difficult to implement physically due to its reliance on precise knowledge of forward-pass computations in neural networks. To overcome this issue, We present a physics-compatible deep neural network architecture, augmented by a biologically-inspired learning algorithm referred to as physical local learning (PhyLL). This framework allows for the direct training of deep physical neural networks, comprising layers of physical nonlinear systems. Notably, our approach dispenses with the need for detailed knowledge of the specific properties of these nonlinear physical layers. Our approach outperforms state-of-the-art hardware-aware training methods by enhancing training speed, reducing digital computations and power consumption in physical systems, particularly in optics.
Authors
- Romain Fleury (ORCID: https://orcid.org/0000-0002-9486-6854)
- Babak Rahmani (ORCID: https://orcid.org/0000-0002-0856-9635)
- Matthieu Malléjac (ORCID: https://orcid.org/0000-0003-0896-638X)
- Philipp del Hougne (ORCID: https://orcid.org/0000-0002-4821-3924)
- Ali Momeni (ORCID: https://orcid.org/0000-0001-9707-8290)
Institutions
- Centre National de la Recherche Scientifique (FR)
- Institut d'Électronique et des Technologies du numéRique (FR)
- Microsoft Research (United Kingdom) (GB)
- École Polytechnique Fédérale de Lausanne (CH)
- Université de Rennes (FR)
Publication Details
- Published
- 2024-09-09
- DOI
- https://doi.org/10.1109/metamaterials62190.2024.10703263
- Primary Topic
- Neural Networks and Reservoir Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.00