Method for AI-based channel estimation with varying PRB set size

A non-transitory storage medium having stored therein instructions that are executable by one or more hardware processors to perform operations comprising:

• receiving, by a pre-processing unit, a rough channel estimated array;
• splitting, by the pre-processing unit, the rough channel estimated array into a group of smaller arrays that each have a size that is smaller than a size of the rough channel estimated array;
• providing, by the pre-processing unit, the smaller arrays to a single neural network (NN);
• processing, by the NN, the smaller arrays to generate respective refined channel estimation outputs for each of the smaller arrays;
• and
• combining, by a post-processing unit, the smaller arrays having the respective refined channel estimation outputs to generate an output array with a size that is the same as the size of the rough channel estimated array.