Demodulation of modulation constellations with probabilistic amplitude shaping

A method for wireless communication at a network entity, comprising:

• determining a probabilistic amplitude shaping for a modulation constellation of a signal to be transmitted to a user equipment (UE);
• transmitting, to the UE, a probability distribution indicator that is associated with the probabilistic amplitude shaping;
• modulating the signal to be transmitted to the UE using the probabilistic amplitude shaping to generate a shaped modulation constellation;
• and
• transmitting the shaped modulation constellation to the UE.

An apparatus for wireless communication at a user equipment (UE), comprising:

• one or more processors;
• memory coupled with the one or more processors;
• and
• instructions stored in the memory and executable by the one or more processors to cause the apparatus to:
• receive an input signal from a transmitter on a wireless resource;
• estimate a channel noise between the UE and the transmitter associated with the wireless resource to determine a channel noise estimate;
• scale the input signal and the channel noise estimate to generate a scaled input signal and a scaled channel noise estimate, wherein the scaling is based at least in part on a probability distribution parameter that is associated with a probabilistic amplitude shaping that is applied at the transmitter to a modulation constellation of the input signal;
• and
• demap the modulation constellation of the input signal based at least in part on the scaled input signal and the scaled channel noise estimate.

An apparatus for wireless communication at a network entity, comprising:

• one or more processors;
• memory coupled with the one or more processors;
• and
• instructions stored in the memory and executable by the one or more processors to cause the apparatus to:
• determine a probabilistic amplitude shaping for a modulation constellation of a signal to be transmitted to a user equipment (UE);
• transmit, to the UE, a probability distribution indicator that is associated with the probabilistic amplitude shaping;
• modulate the signal to be transmitted to the UE using the probabilistic amplitude shaping to generate a shaped modulation constellation;
• and
• transmit the shaped modulation constellation to the UE.