In an aspect of the disclosure, a method, a computer-readable medium, and an apparatus are provided. The apparatus may be wireless equipment. The wireless equipment selects a first subset of beams to be utilized for beam management. The beams are from a set of first type of beams used for communication with a base station or a UE. The wireless equipment measures signals transmitted on a second subset of beams. The beams are from the set of first type of beams or from a set of second type of beams. The wireless equipment measures the signals over a time window. The wireless equipment inputs the measurements to a computational model. The wireless equipment receives predictions of channel measurements on the first subset of beams from the computational model.