SIGNALING FOR ADDITIONAL TRAINING OF NEURAL NETWORKS FOR MULTIPLE CHANNEL CONDITIONS

A method of wireless communication by a user equipment (UE) includes receiving, from a base station, a configuration to train a neural network for multiple different signal to noise ratios (SNRs) of a channel estimate for a wireless communication channel. The method also includes determining a current SNR of the channel estimate is above a first threshold value. The method further includes training the neural network based on the channel estimate, to obtain a first trained neural network. The method still further includes perturbing the channel estimate to obtain a perturbed channel estimate, and training the neural network based on the perturbed channel estimate, to obtain a second trained neural network. The method includes reporting, to the base station, parameters of the first trained neural network along with the channel estimate, and parameters of the second trained neural network.