Ml-based reception of data in a wireless communication system

An apparatus for reception of data in a wireless communication system, the apparatus comprising at least one processor;

• and at least one memory storing instructions that, when executed by the at least one processor, cause the apparatus to:
• receive a set of soft bits, referred to as set of raw soft bits, representing at least part of the data;
• input the set of raw soft bits into a machine learning model, to obtain an output comprising a set of symbols, referred to as set of refined soft symbols, and a set of one or more values indicating a signal disturbance level for the set of refined soft symbols;
• provide the set of refined soft symbols and the set of one or more values for enabling a decoding of the at least part of the data.