Transfer Learning Strategy in Neural Network Application for Underwater Visible Light Communication System

Abstract

Post-equalization using neural network (NN) is a promising technique that models and offsets the nonlinear distortion in visible light communication (VLC) channels, which is recognized as an essential component in the incoming 6G era. NN post-equalizer is good at modeling complex channel effects without previously knowing the law of physics during the transmission. However, the trained NN might be weak in generalization, and thus consumes considerable computation in retraining new models for different channel conditions. In this paper, we studied transfer learning strategy, growing DNN models from a well-trained ‘stem model’ instead of exhaustively training multiple models from randomly initialized states. It extracts the main feature of the channel first whose signal power balances the signal-to-noise ratio and the nonlinearity, and later focuses on the detailed difference in other channel conditions. Compared with the exhaustive training strategy, stem-originated DNN models achieve 64% of the working range with five times the training efficiency at most or more than 95% of the working range with 150% higher efficiency. This finding is beneficial to improving the feasibility of DNN application in real-world UVLC systems.