Noise-resilient photonic neural networks through adaptive quantization

Abstract Photonic neural networks have emerged as a promising solution to overcome limitations of traditional hardware for neuromorphic computations, offering advantages in bandwidth, latency, and power efficiency. However, their performance is constrained by the limited precision of analog photonic computing, which is affected by inherent noise sources such as thermal and shot noise, and distortions. These effects degrade the photonic neural network performance, reducing the bit resolution achievable in photonic hardware typically to 2–4 bits. Traditional quantization strategies fail to account for these noise contributions, resulting in a substantial accuracy loss during inference. This paper introduces an adaptive quantization method called Adaptive-Quantization Photonic-Aware Neural Network (AQ-PANN) to address the challenges posed by different noise sources in analog photonic hardware. The proposed method uses a learnable step size quantization scheme to achieve high accuracy and stability under varying noise levels, introducing a scheme that unifies quantization step adaptation with noise injection exactly where photonic distortions arise. This design incurs only a minor training-time overhead, as it involves learning a small number of per-layer quantization step sizes and does not affect inference. Experimental evaluations on three commonly used test datasets (MNIST, SVHN, and CIFAR-10) with different bit resolutions show the robustness of AQ-PANN. On MNIST, an accuracy drop of only 2% was observed from low to high noise levels in a 4-bit configuration, while traditional DoReFa quantization suffered a 29% drop. For the SVHN dataset, AQ-PANN obtained a mean accuracy of 92% under high noise with 4-bit quantization, outperforming DoReFa by over 45%. On CIFAR-10, AQ-PANN maintained close to 60% accuracy under high noise in the 4-bit configuration, whereas DoReFa and PACT both collapsed below 40%. These results highlight the effectiveness of AQ-PANN in sustaining model performance across different noise intensities, enabling practical photonic neural network deployment.