Neural network based channel state information feedback

Various aspects of the present disclosure generally relate to neural network based channel state information (CSI) feedback. In some aspects, a device may obtain a CSI instance for a channel, determine a neural network model including a CSI encoder and a CSI decoder, and train the neural network model based at least in part on encoding the CSI instance into encoded CSI, decoding the encoded CSI into decoded CSI, and computing and minimizing a loss function by comparing the CSI instance and the decoded CSI. The device may obtain one or more encoder weights and one or more decoder weights based at least in part on training the neural network model. Numerous other aspects are provided.