Disclosed is a method for wireless communication. The method comprises receiving, from a network node, a reference signal (RS) of a first type. An input signal of a second type is determined based on the reference signal. Signals of the second type are associated with training a first machine learning (ML) model applied by a UE to estimate a parameter used in wireless communications by the UE. The input signal of the second type is applied to the first ML model to estimate the parameter used in wireless communications by the UE.