When beamforming (e.g., via a millimeter wave system (mmW)) is used for wireless communication, a base station may transmit beams that are directed to certain directions. Due to the directional nature of the beams in the mmW system, an approach to determine a beam that provides a desirable gain is studied. The apparatus may be a user equipment (UE). The apparatus receives, from a base station, a plurality of signals through a plurality of beams of the base station, each of the plurality of beams corresponding to a respective antenna port of a plurality of antenna ports of the base station. The apparatus receives from the base station a number of beams whose information should be fed back to the base station. The apparatus performs channel estimation for each beam of the plurality of beams from the plurality of antenna ports based on the plurality of signals.