RECURRENT EQUIVARIANT INFERENCE MACHINES FOR CHANNEL ESTIMATION

Methods, systems, and devices for wireless communications are described. A wireless device may receive an assignment of a set of resources associated with a channel where the set of resources includes a first subset of resources allocated for data transmission and a second subset of resources allocated for a reference signal. The wireless device may generate multiple channel estimations per layer of the channel and perform a refinement operation utilizing the estimations to generate a channel estimation associated with multiple layers. Each iteration of the refinement operation may include generating respective gradients associated with each per layer channel estimation; generating a current set of values of a latent variable; and modifying the channel estimations.