An apparatus is disclosed, which may include: at least one processor; and at least one memory storing instructions that, when executed by the at least one processor, cause the apparatus at least to perform: receiving downlink channel estimates between a plurality of access point (AP) antennas of one or more APs and a plurality of user equipment (UE) antennas of one or more UEs; iteratively updating a set of power allocation weights for the AP antennas using weights extracted from a precoder; computing the precoder based on the received channel estimates and a previous set of power allocation weights; performing the iterative updates until a stopping criterion is reached; and outputting data representative of a precoder computed from the received channel estimates and the updated set of power allocation weights.