GPU-Based Implementation of Pruned Artificial Neural Networks for Digital Predistortion Linearization of Wideband Power Amplifiers

Abstract

This paper presents a feature selection technique based on $\ell _{1}$ regularization to select the most relevant weights of artificial neural networks (ANNs) for digital predistortion (DPD) linearization of wideband radio-frequency (RF) power amplifiers (PAs). The proposed pruning method is applied to the first hidden layer of a feed-forward real-valued time-delay neural network, commonly used for DPD purposes. In addition, this paper presents the ANN-based DPD implementation using a graphic processing unit (GPU) with compute unified device architecture (CUDA) units. Thanks to the proposed pruning strategy, it is possible to reduce the ANN complexity significantly, thereby achieving a higher data throughput with the GPU-based implementation. The trade-off among RF performance metrics, number of model parameters and throughput of the GPU implementation is evaluated considering the linearization of a high-efficiency pseudo-Doherty load modulated balanced amplifier (LMBA). The linearized PA operating at an RF frequency of 2 GHz delivers a mean output power of 40 dBm with approximately 50% power efficiency when excited with 5G new radio (NR) signals with up to 200 MHz bandwidth and an 8 dB peak-to-average power ratio (PAPR). The real-time GPU implementation of the ANN-based DPD can meet the linearity specifications with a throughput circa 1 GSa/s.