Streamlined optical training of large-scale modern deep learning architectures with direct feedback alignment
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2024
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866908294900088832 |
|---|---|
| author | Wang, Ziao Müller, Kilian Filipovich, Matthew Launay, Julien Ohana, Ruben Pariente, Gustave Mokaadi, Safa Brossollet, Charles Moreau, Fabien Cappelli, Alessandro Poli, Iacopo Carron, Igor Daudet, Laurent Krzakala, Florent Gigan, Sylvain |
| author_facet | Wang, Ziao Müller, Kilian Filipovich, Matthew Launay, Julien Ohana, Ruben Pariente, Gustave Mokaadi, Safa Brossollet, Charles Moreau, Fabien Cappelli, Alessandro Poli, Iacopo Carron, Igor Daudet, Laurent Krzakala, Florent Gigan, Sylvain |
| contents | Modern deep learning relies nearly exclusively on dedicated electronic hardware accelerators. Photonic approaches, with low consumption and high operation speed, are increasingly considered for inference but, to date, remain mostly limited to relatively basic tasks. Simultaneously, the problem of training deep and complex neural networks, overwhelmingly performed through backpropagation, remains a significant limitation to the size and, consequently, the performance of current architectures and a major compute and energy bottleneck. Here, we experimentally implement a versatile and scalable training algorithm, called direct feedback alignment, on a hybrid electronic-photonic platform. An optical processing unit performs large-scale random matrix multiplications, which is the central operation of this algorithm, at speeds up to 1500 TeraOPS under 30 Watts of power. We perform optical training of modern deep learning architectures, including Transformers, with more than 1B parameters, and obtain good performances on language, vision, and diffusion-based generative tasks. We study the scaling of the training time, and demonstrate a potential advantage of our hybrid opto-electronic approach for ultra-deep and wide neural networks, thus opening a promising route to sustain the exponential growth of modern artificial intelligence beyond traditional von Neumann approaches. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2409_12965 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Streamlined optical training of large-scale modern deep learning architectures with direct feedback alignment Wang, Ziao Müller, Kilian Filipovich, Matthew Launay, Julien Ohana, Ruben Pariente, Gustave Mokaadi, Safa Brossollet, Charles Moreau, Fabien Cappelli, Alessandro Poli, Iacopo Carron, Igor Daudet, Laurent Krzakala, Florent Gigan, Sylvain Emerging Technologies Disordered Systems and Neural Networks Machine Learning Applied Physics Optics Modern deep learning relies nearly exclusively on dedicated electronic hardware accelerators. Photonic approaches, with low consumption and high operation speed, are increasingly considered for inference but, to date, remain mostly limited to relatively basic tasks. Simultaneously, the problem of training deep and complex neural networks, overwhelmingly performed through backpropagation, remains a significant limitation to the size and, consequently, the performance of current architectures and a major compute and energy bottleneck. Here, we experimentally implement a versatile and scalable training algorithm, called direct feedback alignment, on a hybrid electronic-photonic platform. An optical processing unit performs large-scale random matrix multiplications, which is the central operation of this algorithm, at speeds up to 1500 TeraOPS under 30 Watts of power. We perform optical training of modern deep learning architectures, including Transformers, with more than 1B parameters, and obtain good performances on language, vision, and diffusion-based generative tasks. We study the scaling of the training time, and demonstrate a potential advantage of our hybrid opto-electronic approach for ultra-deep and wide neural networks, thus opening a promising route to sustain the exponential growth of modern artificial intelligence beyond traditional von Neumann approaches. |
| title | Streamlined optical training of large-scale modern deep learning architectures with direct feedback alignment |
| topic | Emerging Technologies Disordered Systems and Neural Networks Machine Learning Applied Physics Optics |
| url | https://arxiv.org/abs/2409.12965 |