Multi-Wavelength Weighting Strategy of Diffractive Deep Neural Network

Jianan Feng,Hang Chen,Dahai Yang,Junbo Hao,Jie Lin,Peng Jin
DOI: https://doi.org/10.2139/ssrn.4388387
2023-01-01
Abstract:The diffractive deep neural network (D2NN) exhibits potential advantages to perform large-scale computational task in terms of high speed, low power consumption, parallelism and scalability. Currently, D2NN with cascaded diffractive surfaces is usually designed for single illumination. In this article, we propose a scheme of D2NN for multiple wavelengths based on the wavelength weighting coefficient. With the wavelength weighting loss function, multi-wavelength D2NN could be optically classify the images of handwritten digits by detecting the intensity power. Relying on the above mechanism, the wavelength-weighting D2NN has numerically achieved a test accuracy of more than 83.3% over the MNIST dataset for three wavelengths illumination. Using spatial light modulation, we constructed a single-layer wavelength-weighting D2NN and experimentally verified handwritten digits classification under 3 wavelengths illumination. Furthermore, this scheme is potential to realize incoherent beam diffractive deep neural network. Our method can be also extended to intelligent machine vision system and hyperspectral processor.
What problem does this paper attempt to address?