Efficient Training of Large-Scale Optical Neural Network Based on Evolutionary Algorithm

Zhiwei Yang,Yihang Lai,Tian Zhang,Jian Dai,Kun Xu
DOI: https://doi.org/10.1109/icocn63276.2024.10648457
2024-01-01
Abstract:In this article, we introduce a novel learning mechanism based on evolutionary algorithm to train subspace optical neural networks (SONNs). To demonstrate the effectiveness of our proposed covariance matrix adaptation evolution strategy (CMA-ES), the trained SONNs are applied in a typical MNIST dataset classification task. The calculated results demonstrate that the accuracy and stability of the training method are competitive with the traditional gradient algorithm, but the SONNs trained by the CMA-ES can achieve 1.5 times acceleration.
What problem does this paper attempt to address?