Permutation-equivariant neural channel coding construction

An apparatus for wireless communication at a wireless communication device, comprising:

• one or more memories;
• and
• one or more processors, the one or more processors, individually or collectively and based at least in part on information stored in the one or more memories, being configured to:
• obtain a message vector;
• generate a set of message sub-vectors using the message vector;
• identify a set of codeword sub-vectors using the set of message sub-vectors, wherein a particular codeword sub-vector of the set of codeword sub-vectors corresponds to a particular message sub-vector of the set of message sub-vectors, and wherein identifying the particular codeword sub-vector is associated with:
• the particular message sub-vector, and
• a set of remaining message sub-vectors of the set of message sub-vectors;
• and
• transmit a communication carrying a codeword vector associated with the set of codeword sub-vectors.

A method of wireless communication performed by a wireless communication device, comprising:

• obtaining, by the wireless communication device, a message vector;
• generating, by the wireless communication device, a set of message sub-vectors using the message vector;
• generating, by the wireless communication device and using a common input linear layer, a set of message embedding vectors corresponding to the set of message sub-vectors;
• generating, by the wireless communication device and using an artificial neural network, a set of codeword embedding vectors corresponding to the set of message embedding vectors, wherein the artificial neural network comprises a transformer with a plurality of transformer encoder layers;
• generating, by the wireless communication device and using a common output linear layer and the set of codeword embedding vectors, a set of codeword sub-vectors, wherein the set of codeword sub-vectors is permutation-equivariant relative to the set of message sub-vectors;
• combining, by the wireless communication device, the set of codeword sub-vectors to generate a codeword vector;
• and
• transmitting, by the wireless communication device, a communication carrying the codeword vector.