This disclosure relates to techniques for providing an artificial intelligence based framework for performing channel state information reporting in a wireless communication system. A cellular base station may provide system information for a cell to a wireless device. The system information may indicate that the cell supports artificial intelligence based channel state information reporting. The wireless device may provide wireless device capability information to the cellular base station. The capability information may indicate that the wireless device supports artificial intelligence based channel state information reporting. The wireless device may determine an artificial intelligence model to use to perform channel state information reporting with the cell, and may perform channel state information reporting using the selected artificial intelligence model.