A method by a user equipment, UE (250, 260, 320, 420, 820), with one or more receive antenna ports, each of the receive antenna ports associated with a plurality of receive beam ports, the method comprising: receiving, by the UE, information from a network (100, 130, 200) indicating configurations of one or more transmit antenna ports, each of the transmit antenna ports associated with a plurality of transmit beam ports, the information further indicating a plurality of reference signals, each of the reference signals associated with one of the transmit beam ports for the transmit antenna ports; receiving, by the UE, a subset of the reference signals; measuring, by the UE (250, 260, 320, 420, 820), receive signal quality for each of the subset of the reference signals; determining, by the UE, a selected receive beam port for each of the receive antenna ports according to the receive signal quality measured; deriving, by the UE, one or more reporting sets (510, 520, 832, 834), the one or more reporting sets corresponding to one of the receive antenna ports, each of the reporting sets indicating one or more selected transmit beam ports for the receive antenna port, the one or more reporting sets derived according to the receive signal quality measured; transmitting, by the UE, the one or more reporting sets (510, 520, 832, 834) to the network (100, 130, 200); and receiving, by the UE, a data transmission from the selected transmit beam ports using the selected receive beam port for each of the receive antenna ports, wherein the information indicating the plurality of reference signals is a channel state information, CSI, reporting configuration, and wherein transmitting the one or more reporting sets (510, 520, 832, 834) comprises transmitting a CSI report including the one or more reporting sets. receiving, by the UE, information from a network (100, 130, 200) indicating configurations of one or more transmit antenna ports, each of the transmit antenna ports associated with a plurality of transmit beam ports, the information further indicating a plurality of reference signals, each of the reference signals associated with one of the transmit beam ports for the transmit antenna ports; receiving, by the UE, a subset of the reference signals; measuring, by the UE (250, 260, 320, 420, 820), receive signal quality for each of the subset of the reference signals; determining, by the UE, a selected receive beam port for each of the receive antenna ports according to the receive signal quality measured; deriving, by the UE, one or more reporting sets (510, 520, 832, 834), the one or more reporting sets corresponding to one of the receive antenna ports, each of the reporting sets indicating one or more selected transmit beam ports for the receive antenna port, the one or more reporting sets derived according to the receive signal quality measured; transmitting, by the UE, the one or more reporting sets (510, 520, 832, 834) to the network (100, 130, 200); and receiving, by the UE, a data transmission from the selected transmit beam ports using the selected receive beam port for each of the receive antenna ports, wherein the information indicating the plurality of reference signals is a channel state information, CSI, reporting configuration, and wherein transmitting the one or more reporting sets (510, 520, 832, 834) comprises transmitting a CSI report including the one or more reporting sets.