A UE transmits to a BS an indication of a number of PTRS ports. The number of PTRS ports is a suggestion to the BS for allocating the indicated number of PTRS ports to the UE for transmission of PTRS from the BS to the UE to enable the UE to perform phase tracking. The method also includes allocating, by the BS, PTRS ports to the UE based on the indication of the number of PTRS ports. The indication may be included in a UCI message, MAC CE, or RRC message transmitted by the UE to the BS. The BS may map the allocated PTRS ports to DMRS ports corresponding to spatial streams transmitted by the BS. The UE may estimate CPE of each spatial stream, measure correlations of the estimated CPE among the spatial streams, and use the correlations to determine the suggested number of PTRS.