Method for data augmentation for channel state information feedback

An 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 at least to perform:
• receiving, from a network node, a request for training data for data augmentation algorithm;
• transmitting, to the network node, a first part of a plurality of parts of the training data, wherein properties of data in the first part of the training data reflect distribution of the training data;
• wherein the apparatus is caused to perform either:
• i) computing a first set of precoding matrix indices based on the training data;
• receiving, from the network node, a second set of precoding matrix indices computed based on data samples generated by the network node based on the first part of the training data transmitted to the network node;
• comparing the first set of precoding matrix indices and the second set of precoding matrix to verify the data samples generated by the network node;
• and
• based on determining that a number of indices of the second set of precoding matrix indices verified as correct remains below a pre-set threshold, transmitting a second part of the plurality of parts of the training data to the network node;
• or
• ii) before transmitting the first part of the training data, the apparatus is caused to perform:
• generating data samples based on one or more parts of the plurality of parts of the training data, wherein properties of data in the one or more parts reflect distribution of the training data;
• verifying the generated data samples;
• iteratively repeating generating the data samples and verifying the generated data samples to determine a suitable number of the one or more parts of the training data to enable generation of a desired set of training data;
• and wherein the first part of the training data to be transmitted to the network node comprises the suitable number of the one or more parts of the training data.