A user equipment (UE) in a wireless communication system includes a transceiver and a processor. The transceiver receives information about an antenna system of a base station. The information includes a number of collocated antenna groups, and a number of antenna modules for each type of antenna modules in each of the collocated antenna groups. A collocated antenna group has at least two types of antenna modules: a first module with first antenna type and a second module with second antenna type. The transceiver also receives configuration information for a channel state information-reference signal (CSI-RS) resource. The transceiver also receives a CSI-RS based on the configuration information and acquires measurements. The processor determines a subset of antenna modules based on a comparison between the measurements and a criterion. The processor generates CSI for the subset of antenna modules. The transceiver transmits a CSI report comprising the CSI.