Beam report enhancements for beam prediction

A method for wireless communication at a user equipment (UE), comprising:

• receiving control signaling identifying, for a plurality of beams, a set of quantized values from a plurality of sets of quantized values configured at the UE for beam management reporting for the plurality of beams, wherein the control signaling comprises an indication of a codebook that is one of a plurality of codebooks configured at the UE for beam management reporting for the plurality of beams, and wherein the set of quantized values is identified by the codebook and the codebook is based on one or more machine learning models used at the network node for beam predictions or other beam management procedures;
• receiving a set of reference signals that correspond to a set of beams of the plurality of beams, each reference signal of the set of reference signals associated with a beam of the plurality of beams;
• and
• transmitting a report indicating a set of power values that are based at least in part on the set of quantized values and the set of reference signals received by the UE, each power value of the set of power values associated with a respective beam of the set of beams.

A user equipment (UE) for wireless communication, comprising:

• means for receiving control signaling identifying, for a plurality of beams, a set of quantized values from a plurality of sets of quantized values configured at the UE for beam management reporting for the plurality of beams, wherein the control signaling comprises an indication of a codebook that is one of a plurality of codebooks configured at the UE for beam management reporting for the plurality of beams, and wherein the set of quantized values is identified by the codebook and the codebook is based on one or more machine learning models used at the network node for beam predictions or other beam management procedures;
• means for receiving a set of reference signals that correspond to a set of beams of the plurality of beams, each reference signal of the set of reference signals associated with a beam of the plurality of beams;
• and
• means for transmitting a report indicating a set of power values that are based at least in part on the set of quantized values and the set of reference signals received by the UE, each power value of the set of power values associated with a respective beam of the set of beams.

A non-transitory computer-readable medium storing code for wireless communication, the code comprising instructions executable by one or more processors to:

• receive control signaling identifying, for a plurality of beams, a set of quantized values from a plurality of sets of quantized values configured at the non-transitory computer-readable medium for beam management reporting for the plurality of beams, wherein the control signaling comprises an indication of a codebook that is one of a plurality of codebooks configured at the UE for beam management reporting for the plurality of beams, and wherein the set of quantized values is identified by the codebook and the codebook is based on one or more machine learning models used at the network node for beam predictions or other beam management procedures;
• receive a set of reference signals that correspond to a set of beams of the plurality of beams, each reference signal of the set of reference signals associated with a beam of the plurality of beams;
• and
• transmit a report indicating a set of power values that are based at least in part on the set of quantized values and the set of reference signals received by the non-transitory computer-readable medium, each power value of the set of power values associated with a respective beam of the set of beams.