Modulation format recognition in a UVLC system based on an ultra-lightweight model with communication-informed knowledge distillation

Abstract

Modulation format recognition (MFR) is a key technology for adaptive optical systems, but it faces significant challenges in underwater visible light communication (UVLC) due to the complex channel environment. Recent advances in deep learning have enabled remarkable achievements in image recognition, owing to the powerful feature extraction of neural networks (NN). However, the high computational complexity of NN limits their practicality in UVLC systems. This paper proposes a communication-informed knowledge distillation (CIKD) method that achieves high-precision and low-latency MFR with an ultra-lightweight student model. The student model consists of only one linear dense layer under a communication-informed auxiliary system and is trained under the guidance of a high-complexity and high-precision teacher model. The MFR task involves eight modulation formats: PAM4, QPSK, 8QAM-CIR, 8QAM-DIA, 16QAM, 16APSK, 32QAM, and 32APSK. Experimental results show that the student model based on CIKD can achieve comparable accuracy to the teacher model. After knowledge transfer, the prediction accuracy of the student model can be increased by up to 87%. Besides, it is worth noting that CIKD’s inference accuracy can reach up to 100%. Moreover, the parameters constituting the student model in CIKD correspond to merely 18% of the parameters found in the teacher model, which facilitates the hardware deployment and online data processing of MFR algorithms in UVLC systems.