Efficient user clustering using a low-complexity Artificial Neural Network (ANN) for 5G NOMA systems

Non-orthogonal multiple access (NOMA) has gained considerable interest from the 3GPP community as a potential radio access strategy for the future fifth-generation (5G) wireless networks. Compared to orthogonal multiple access (OMA), NOMA is more efficient from the perspective of throughput performance making it more favorable for 5G systems. Existing NOMA techniques merely offer a rigid user grouping without exploring channel heterogeneity and diversity to cluster users, resulting in a poor throughput performance. An adaptive user clustering (AUC) approach has been proposed to search through all possible combinations to obtain the best clusters with the highest throughput. This scheme exploits the channel diversity of users to maximize throughput, however, the brute-force search (B-FS) method to find the optimal combinations results in a prohibitive complexity. In this paper, a novel artificial neural network (ANN) approach is proposed for user clustering in the downlink of the 5G NOMA system in order to maximize throughput performance at an acceptable complexity. In the proposed strategy, ANN model is first trained with the historical dataset, which contains the transmitting powers and channel gains of the downlink NOMA users, along with the information of the corresponding clusters which maximize the throughput performance of the system. Next, validation is performed to tune the values of hyper-parameters such as learning rate, length of training data, and epoch learned during training to validate cluster formation and to avoid over-fitting of the model. Finally, the ANN model is tested with the learned parameters and tuned hyper-parameters, to predict the formation of clusters and to evaluate the accuracy of the model. Simulation results demonstrate that the proposed scheme is able to obtain a significant reduction in terms of complexity with a performance of 98% for throughput (near-optimal throughput performance) when compared with the optimal approaches.