The present disclosure enables a base station to group UEs based on similar PMIs for a combined transmission. The apparatus may receive a plurality of PMIs from a plurality of UEs. In one aspect, each PMI in the plurality of PMIs may be received from a different UE. The apparatus may also assign each UE of the plurality of UEs to a UE group. In another aspect, each UE group may be associated with a different PMI set in a plurality of PMI sets. The apparatus may further assign a transmission scheme to each UE group. In one configuration, the apparatus may determine one or more preferred spatial layers for each UE (e.g., based on the PMIs) assigned to a UE group, and schedule a combined transmission for the UE group based on the determined one or more spatial layers.