Skip to main content
Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

Learned-SBL-GAMP based hybrid precoders/combiners in millimeter wave massive MIMO systems

  • Shoukath Ali K. ,

    Roles Conceptualization, Data curation, Investigation, Software, Supervision, Validation, Visualization, Writing – original draft, Writing – review & editing

    shoukathlk@gmail.com (SAK); arfatkhan@kku.ac.th (AAK)

    Affiliation Department of Electronics and Communication Engineering, Presidency University, Itgalpura, Rajanukunte, Yelahanka, Bengaluru, Karnataka, India

  • Arfat Ahmad Khan ,

    Roles Conceptualization, Formal analysis, Funding acquisition, Investigation, Methodology, Project administration, Resources, Software, Supervision, Validation, Visualization, Writing – original draft, Writing – review & editing

    shoukathlk@gmail.com (SAK); arfatkhan@kku.ac.th (AAK)

    Affiliation Department of Computer Science, College of Computing, Khon Kaen University, Khon Kaen, Thailand

  • Perarasi T,

    Roles Conceptualization, Data curation, Software, Validation, Visualization

    Affiliation Department of Electronics and Communication Engineering, Bannari Amman Institute of Technology, Anna University, Erode, Tamil Nadu, India

  • Ateeq Ur Rehman,

    Roles Conceptualization, Data curation, Formal analysis, Funding acquisition, Methodology, Project administration, Writing – review & editing

    Affiliation Department of Electrical Engineering, Government College University, Lahore, Pakistan

  • Khmaies Ouahada

    Roles Funding acquisition, Validation, Visualization

    Affiliation Department of Electrical and Electronic Engineering Science, University of Johannesburg, Johannesburg, South Africa

Abstract

In Millimeter-Wave (mm-Wave) massive Multiple-Input Multiple-Output (MIMO) systems, hybrid precoders/combiners must be designed to improve antenna gain and reduce hardware complexity. Sparse Bayesian learning via Expectation Maximization (SBL-EM) algorithm is not practically feasible for high signal dimensions because estimating sparse signals and designing optimal hybrid precoders/combiners using SBL-EM still provide high computational complexity for higher signal dimensions. To overcome the issues of high computational complexity along with making it suitable for larger data sets, in this paper, we propose Learned-Sparse Bayesian Learning with Generalized Approximate Message Passing algorithm (L-SBL-GAMP) algorithm for designing optimal hybrid precoders/combiners suitable for mmWave Massive MIMO systems. The L-SBL-GAMP algorithm is an extension of the SBL-GAMP algorithm that incorporates a Deep Neural Network (DNN) to improve the system performance. Based on the nature of the training data, the L-SBL-GAMP can design the optimal Hybrid precoders/combiners, which enhances the spectral efficiency of mmWave massive MIMO systems. The proposed L-SBL-GAMP algorithm reduces the iterations, training overhead, and computational complexity compared to the SBL-EM algorithm. The simulation results unveil that the proposed L-SBL-GAMP provides higher achievable rates, better accuracy, and low computational complexity compared to the existing algorithm, such as Orthogonal Matching Pursuit (OMP), Simultaneous Orthogonal Matching Pursuit (SOMP), SBL-EM and SBL-GAMP for mmWave massive MIMO architectures.

Introduction

During the last few decades, many end users are connected to the internet due to the technological enhancements in communication devices [1, 2]. As a result, lower frequency bands have been congested. To tackle this issue, scientists have explored new ways, such as equipping base stations with massive antennas [3, 4], spectrum coding, and network densification algorithms [5]. These ways have solved the issues; however, they are not enough to meet the ever-increasing demand of users. This situation has shifted the focus of research toward millimeter-wave frequencies (mmWave) [6].

The spectrum of mmWave ranges from 30GHz-300GHz, and it is deemed a perfect candidate to meet the requirements of 5G networks, as they require higher spectral and energy efficiencies [7]. However, the main issues associated with mmWave frequencies are related to high path losses and severe blockages [8]. It has been seen that the large antenna array at the base station helps to significantly reduce high path losses [9]. Meanwhile, a large antenna array can easily be installed at the base station owing to the small wavelength of mmWave frequencies. Therefore, the incorporation of mmWave with large antenna arrays at the base station is one of the most suitable options to fulfill the requirements of a current and future generation of communication networks. A large antenna array at the base station is another promising wireless technology to fulfill the requirements of spectral and energy efficiencies for the current and future wireless communication systems [10].

Contrary to the conventional MIMO systems [11] where each transceiver chain is equipped with the individual Digital to Analogue Convertor (DAC), Analogue to Digital Convertor (ADC), frequency matches, and amplifiers, researchers opt for hybrid architectures in mmWave communication systems to reduce the cost. In hybrid architecture, the precoders and combiners are designed by a combination of analog and digital precoders and combiners for processing the transmitting and receiving signals along with reducing the cost. We get high beamforming gain due to the analog domain and high multiplexing gain with reduced spectral efficiency due to the digital domain [12, 13]. Also, along with the cost reduction in mmWave systems, the complexity and the power efficiency of the system get improved because of using fewer RF chains compared to the antennas.

Furthermore, due to the above-mentioned propagation losses at the channel in mmWave systems, the response of the channel gets sparse because the propagation characteristics of mmWave are different compared to microwave systems [14, 15]. Also, due to the sparse channel in mmWave, the nature of the hybrid precoders gets sparse. Many attempts have been made in the literature to address the hybrid precoding problem using various algorithms [16]. In [17, 18], the authors propose a codebook-based hybrid precoding design by considering the sparse nature of the mmWave channel. Moreover, Orthogonal Matching Pursuit (OMP) is deemed a good candidate for designing hybrid precoders. However, with the increment in the number of transceivers, the computational complexity gets increased in OMP [19, 20]. As a result, it brings a poor estimation of the channel matrix due to the increment in the computational complexity, which reduces the spectral efficiency. Another approach, named as greedy Simultaneous Orthogonal Matching Pursuit (SOMP) technique, is used to reduce the system’s computational complexity. Although the computation complexity in terms of the number of iterations gets reduced in SOMP, the algorithm still leads to poor performance when we increase the dimension of transceiver chains [21] which reduces the spectral efficiency.

In [22], the authors propose a greedy frequency selective hybrid precoding algorithm based on Gram-Schmidt orthogonalization to reduce the complexity during the designing of precoders/combiners. The Gram-Schmidt orthogonalization algorithm is based on the greedy selection of columns of the transceiver matrix by considering minimal spatially active channel components to reduce the approximation error. In [23], the authors propose a frequency-selective hybrid precoding based on maximizing mutual information with the help of quantized codebooks. However, similar to the issues in OMP, the proposed schemes lead to poor performances when the number of transceiver elements gets increased.

In [24], the authors propose a Sparse Bayesian Learning (SBL) estimation scheme for the evaluation of posterior functions to the prior information, improving the channel’s estimation accuracy. Motivated by this research, the authors discuss the computation of sparse channels with the help of SBL estimation schemes [2527]. They developed a Single Measurement Vector (SMV) for SBL-based mmWave MIMO channel estimation and then moved to the development of Multiple Measurement Vectors (MMV) for multi-user communication.

In [28], the authors propose a novel SBL framework to estimate the spatial signatures and temporal varying characteristics of the sparse virtual channel model. To avoid the unacceptable complexity, the expectation maximization (EM) algorithm is applied to approximate the maximum likelihood (ML) solution iteratively. Furthermore, in [29], the authors estimate the channel of a massive MIMO system by using the EM-based Variational Bayesian (EM-VB). By exploiting the channel reciprocity between the uplink and downlink, the downlink channel is estimated. In terms of methods, the authors use the SBL via EM approach for the computation of a time-varying channel in both of these papers.

In terms of precoder designing, the SBL method is used to minimize the error between the desired and the estimated precoders. For the estimation of missing information, the Expectation Maximization (EM) based SBL method is used during the designing of hybrid precoders, as EM-based SBL assists to compute the posterior distribution of hidden variables [30, 31]. In addition, it has been seen that the SBL works well for signal recovery, and in fact, it shows better performance compared to greedy methods and other basis pursuit methods. Despite exhibiting good performance, the dimension of the signal gets increased because we have to compute inverse elements having NxN size.

In [32], the authors propose a Generalized Approximate Message Passing (GAMP) algorithm for the estimation of the mmWave channel in a computationally efficient way. The integration of approximate message passing with EM assists to compute the distribution of true posterior approximately. Moreover, it has been seen that the performance of SBL with EM and GAMP is the same in terms of channel estimation accuracy, but the complexity is less in the latter one [33]. In this paper, we work on the design of hybrid precoders/combiners for mmWave massive MIMO hybrid architecture. The main purpose of using the hybrid architecture is to reduce the cost along with the power consumption, as RF chains are less than the transmitting antennas in the hybrid architecture. The nature of the precoders gets sparse in mmWave systems, we have proposed a deep learning-based architecture for designing the optimal hybrid precoders/combiners by getting motivated by the success of SBL via the GAMP algorithm for the estimation of mmWave channels in [3436]. Firstly, the SBL via the GAMP algorithm is used for designing optimal hybrid precoders/combiners for mmWave massive MIMO systems in a computationally efficient way. To further improve the estimation accuracy and the channel overhead, we have proposed a deep learning-based approach, named Learned-SBL-GAMP (L-SBL-GAMP) algorithm, to design the hybrid precoders/combiners in mmWave massive MIMO systems. Actually, in the L-SBL-GAMP algorithm, all T layers are used to estimate the hybrid precoders/combiners and its matrices are sparse. The simulation results unveil the effectivities of the proposed deep learning-based method, where it can be seen that the proposed deep learning-based algorithm outperforms the existing methods (LS, OMP, SOMP, SBL-EM, and SBL-GAMP) by providing the reduced Normalized Mean Square Error (NMSE) value and the increased spectral efficiency.

In summary, Table 1 provides the pros/cons of the recent state-of-the-art work in the literature.

thumbnail
Table 1. A brief comparison of the pros/cons of the recent state-of-the-art work in the literature.

https://doi.org/10.1371/journal.pone.0289868.t001

The main contributions and novelties of the paper are summarized as follows:

  1. In this paper, we provide the mathematical modeling of optimal precoders/combiners for mmWave massive MIMO hybrid architecture.
  2. To design hybrid precoders/combiners, we utilize the L-SBL via the GAMP algorithm, which reduces the computational complexity of the system compared to the EM-based SBL approach.
  3. Due to the fixed nature of hyperparameters during the iterations in SBL via GAMP, the estimation accuracy of precoders gets reduced. Therefore, to tackle this issue, we propose a deep learning-based approach named learned SBL via GAMP.
  4. In the proposed methodology, the hyperparameters get effectively tuned during each iteration, which assists in achieving the minimum loss function. The proposed DNN architecture can design the optimal hybrid precoders/combiners.
  5. To validate the effectiveness of the proposed methodology, the experimental results are provided where it can be seen that the proposed methodology outperforms the existing approaches (LS, OMP, SOMP, SBL-EM, and SBL-GAMP) by providing less Normalized Mean Square Error (NMSE), losses, and the enhanced spectral efficiency.

Notations

we opt for the following notations in this paper: A(.j) and A(i,j) represent the ith row, jth column, and (i, j)th element of a matrix A, respectively. [⋅]* represents a conjugate operator, [⋅]H represents a conjugate transpose operator, and [⋅]T represents a transpose operator. a is a vector and a is a scalar. E(⋅) represents an expectation operator, ‖XF represents a Frobenius norm of X, and denotes the l2 norm of X.

System model

The architecture of hybrid precoders and combiners for the mmWave massive MIMO system is depicted in Fig 1. Let Nt be the transmitting elements at the transmission section, and several receiving antennas at the reception side. represents the RF chains at the transmission, and represents the RF chains at the receiver side. It also NS represents the number of data streams transmitted via RF chains at the transmitter and receiver sides. denotes the baseband precoder, and is the RF precoder at the transmitter side. denotes the baseband combiner, and represents the RF combiner. Furthermore, we impose total power constraints to normalize the baseband precoder FBB, so that [31].

The numerical model for the channel with L paths is represented in Eq (1), which shows the channel correlation between the transmitting and receiving antennas. The geometric channel model is expressed as HNr×Nt, and it is written as: (1)

Where represents the complex gain for the lth path. φr,l,θr,l denote the Angles of Arrivals (AoAs) received for the lth path. Denote the Angles of Departures (AoDs) at the transmitting antenna for the lth path. at(⋅) denotes the transmit array response and ar(⋅) denotes the receive array response. Due to mmWave frequencies, there are a small number of available channel paths L.

Problem definition

This section deals with the designing of hybrid precoders/combiners that optimally increase the system’s achievable rate while suppressing channel training overhead. For the given system and channel models, the achievable rate can be written as [37]: (2)

Where F = FRFFBB denotes the analog/digital precoder, and denotes the analog/digital combiners. The covariance matrix can be written as: (3)

The signal-to-noise ratio (SNR) of the system can be written as: (4)

Where represents the total transmit power, and denotes the noise power.

The designing of optimal hybrid precoders and combiners is required to improvise mutual information between the baseband precoder and RF precoder, which can be written as [31]: (5) (6)

The direct maximization process is difficult; therefore, an ideal digital precoder and combiner method is used. The ideal digital precoder and combiner can be written as: (7) (8)

Where denotes the right singular vector matrices, and denotes the left singular vector matrices. The values of matrices are estimated from the Singular Value Decomposition (SVD) H.

The problem of maximizing the mutual information concerning the precoders FBB and FRF can be well approximated by the sparse matrix reconstruction problem, and it is formulated as below:

The equivalent precoders and combiners design problem for the mmWave MIMO system to achieve the best approximation to the ideal precoder Fopt and ideal combiner Wopt is shown below in Eqs (9) and (11) [31]: (9) (10)

Where denotes the dictionary matrix of transmit array response, and denotes the dictionary matrix of receive array response.

The first constraint arises because can have only nonzero rows corresponding to the number of active RF chains, while the second constraint .

(11)(12)

Sparse Bayesian learning (SBL) algorithm

In this section, a two-layer hierarchical prior model is proposed to calculate the sparsity of a Bayesian learning model. The first layer represents the Gaussian prior distribution [32]: (13)

Where denotes the non-negative hyperparameter, which controls the sparsity coefficient . In the next layer, hyperpriors are computed with the help of Gamma distributions: (14) (15) (16)

The sparse signal and the hyperparameters are learned using the Expectation-Maximization (EM) algorithm. In the EM algorithm, FBB is a hidden variable. Also, using the iterative method, we strengthen a lower bound value on probability p(α,γ|Fopt). The EM algorithm works in two steps named E-step and M-Step, respectively. Firstly, the FBB is computed, and it is given as: (17)

From the posterior function , the Gaussian distribution of its mean and covariance matrix is written as: (18) (19)

The posterior obeys the Gaussian distribution, and its parameters are: (20) (21) (22)

The Q-function is maximized in M-steps by keeping {α,γ}, which leads to the following update rules: (23) (24)

Where is the expectation concerning? It is observed that the EM algorithm for every iteration has to update the posterior distribution with G×G computation inverse. Hence, the EM-based sparse Bayesian has complexity in the range of O(G3) flops, which increases further concerning larger dimensions [32].

The optimal hybrid precoder and combiner are designed with the help of SBL based EM method. The SBL-EM method achieves high spectral efficiency and it increases the training overhead with computational complexity. To minimize the number of iterations and complexity, we use the SBL via the GAMP algorithm. The SBL via the GAMP algorithm does not require an inverse operation. It only needs a simple multiplication operation.

Proposed learned SBL-GAMP architecture

In this section, we first discuss the SBL via the GAMP algorithm and then move to the proposed L-SBL-GAMP algorithm.

SBL-GAMP algorithm

The GAMP algorithm attains the target with a smaller number of iterations and training overhead [32]. The Bayesian iterative technique, named GAMP, is adapted to estimate the optimal hybrid precoder of mmWave systems. The proposed algorithm provides low computational complexity and finds the estimation of a posterior distribution conditioned on the prior distribution FBB. In the EM algorithm, the posterior estimate distribution FBB is used to swap the true posterior distribution. Moreover, the hyperparameters are considered as known and fixed in the GAMP algorithm. Depending on the estimated values FBB, its parametric values are updated in the M-step.

In the GAMP algorithm, we use the central-limit-theorem to approximate variable and factor nodes, which enhances the performance of the system. The SBL-GAMP framework adopts two important approximation methods. In the first approximation method, the GAMP assumes posterior independence among hidden variables {FBBn}, which θ = {α,γ} denotes the hyperparameters.

The true posterior distribution approximates as p(FBBn|Fopt,θ), and it is computed as: (25)

Let and are different variables that are updated for every process value of the GAMP algorithm. With the help of Gaussian Prior Distribution, the approximate posterior of mean and variance is given in Eqs (26) and (27).

(26)(27)

denotes the noiseless output of the other approximation method, where specifies the mth row in AT. The GAMP equates to the exact marginal posterior p(Fm|Foptm,θ): (28)

Where and are the parameters that are iteratively updated during the iterative process of GAMP.

Using the Gaussian prior distribution defines the mean and variance, and it is given in Eqs (29) and (30).

(29)(30)

With the above two approximation methods, the two following important scalar functions are defined as gin(⋅) and gout(⋅), which are utilized for the proposed approach. gin(⋅) denotes the scalar function of input, which defines the posterior mean , and it is expressed by using Minimum Mean Squared Error (MMSE) mode [34]: (31)

To obtain the estimated variance , we take the partial derivative with respect to : (32)

Moreover, gout(⋅) denotes the output scalar function, and its posterior mean is defined as: (33)

Taking the partial derivative of to obtain the posterior variance : (34)

The GAMP algorithm finds the posterior mean and variance FBBn at the k iteration for the given input gin(⋅) and output gout(⋅) scalar functions. Finally, the approximate posterior distributions for the variables FBB Fm = ATFBB are derived by using the proposed GAMP algorithm. The GAMP algorithm does not require an inverse operation. It only needs a simple multiplication operation, which can be scaled as O(NtG). Thus, its computational complexity gets considerably reduced.

As far as learning hyperparameters is concerned, the EM method is used to estimate hyperparameters by considering hidden variables FBB and the iteratively improving Q-function: (35)

To find the number of M-steps for the analysis of hyperparameters {αn}, we take the first-order derivative of the Q-function for αn: (36) (37) (38)

Where represents the expectation operator regarding p(FBB|Fopt,θ(t)). The approximate posterior distribution FBBn is used due to the unavailability of a true posterior. The GAMP algorithm is used to approximate the posterior distribution by replacing the true posterior distribution. From the Gaussian distribution process, the approximate posterior distribution defines the mean and variance, which are given in Eqs (26) and (27) (39) (40)

Here, hyperparameters are required to be updated. The function of a hyperparameter is to inverse noise variance. The approximate posterior distribution of output without any disturbance is considered in the GAMP approach. The F is then assigned to the hidden variables for understanding noise variance, and it is given as: (41)

From the partial derivative of Q-function regarding γ projects: (42) (43)

〈⋅〉 specifies the expectation for , and the approximate posterior distribution Fm. We know that Fm tracks the Gaussian distribution in terms of mean and variance, and it is expressed as: (44) where and from (29)–(30) with relocated with and γ is changed with γ(t). By equating the derivative to zero, the updated rule for: (45)

As of now, the GAMP-based sparse Bayesian learning algorithm is derived. For a better understanding, we have summarized the SBL-GAMP approach in algorithm 1 [32].

Algorithm 1: SBL-GAMP

Initialization: For known θ(t); assign and are initialized for prior distribution with mean and variance. Continue the process until Let ε be pre-specified error tolerance.

Initialization: given and γ(0)

Based on derivates of the GAMP algorithm, hyperparameters are updated according to (40) and (45) using the EM method.

However, in SBL via GAMP, the hyperparameter is considered fixed for all the iterations, which reduces the estimation accuracy of precoders. To overcome this issue, the learned SBL via GAMP is proposed. In the learned SBL via GAMP, for each layer, the hyperparameters get changed and tuned to obtain the minimum loss function.

Learned SBL-GAMP architecture

The architecture of L-SBL-GAMP with multiple layers is shown in Fig 2. Each layer of the L-SBL-GAMP is similar to an iteration of the SBL-GAMP algorithm. The inputs of the L-SBL-GAMP network are the Fopt (Ideal digital Precoder) and AT (transmit array response). The outputs of the L-SBL-GAMP network are baseband precoder and trainable parameters θ = {W} = {α,γ}. In algorithm 2, for the first layer, fix them to learn W1 = {α1,γ1} and re-learn W1. In layer 2, fix the to learn them W2 = {α2,γ2} and re-learn W0, W1 and W2. Similarly, the process will continue till the T layer, and it is shown in algorithm 2.

Algorithm 2: Learned SBL-GAMP

Initialization: Input: Training set For t = 1,…,T do

Initialize: Wt = Wt−1

for R steps do

Learn Wt = {αt,γt} with fixed to minimize the loss function

First estimate optimal precoder

and then find the loss function

End for

for R steps do

Re-Learn to minimize the loss function

Estimate optimal precoder

and then calculate the loss function

End for

End for

Output: θT−1

Algorithm 2, θ = {W} = {α,γ} denotes the set of learnable parameters in the t-th layer. The total number of L-SBL-GAMP layers is denoted as T, and R denotes the total number of mini-batches used in the training of the L-SBL-GAMP layer. The output vector is related to the sparse baseband precoder vector : (46)

The loss function used in the training of the L-SBL-GAMP layer is the mean square error between the true sparse vector and the current training layer output. The expression for the mean square error loss function is given by: (47) where m denotes the number of training samples in a mini-batch and G represents the function learned by L-SBL-GAMP. A layer-by-layer method is used, and the Adam optimizer is used in the back-propagation process to optimize the final value. The L-SBL-GAMP network is implemented in Python using the neural network libraries Keras and TensorFlow.

After estimating the sparse precoder , we are following a two-step procedure to extract the baseband precoder . We calculate the transmit baseband precoder matrix from . Firstly, we separate the G − K rows to get . The rows of the matrix are then removed to finally obtain the matrix. The transmit RF precoder FRF can be extracted by keeping the columns corresponding to the rows extracted from .

The above same procedure is utilized to calculate the baseband combiner and WRF RF combiner. They can be extracted from by retaining the columns corresponding to the rows extracted from .

Moreover, and is the optimum MMSE combiner.

The estimated baseband precoder, RF precoder, baseband combiner, and RF combiner are substituted in Eq (2) to obtain the achievable rate or Spectral Efficiency [37].

Finally, the Normalized Mean Square Error (NMSE) is used to evaluate the performance of the L-SBL-GAMP network: (48)

Computational complexity and performance analysis

Sparse Bayesian Learning (SBL) is a popular method for sparse signal recovery or optimal hybrid precoder design. SBL-EM (Expectation-Maximization) and SBL-GAMP (Gaussian Approximation Message Passing) are two variants of SBL that differ in their algorithms for estimating the sparse signal or designing the hybrid precoder.

In terms of performance, the accuracy of the optimal hybrid precoder design using SBL-EM is almost the same compared to SBL-GAMP, but at the cost of higher computational complexity.

The L-SBL-GAMP algorithm is an extension of the SBL- GAMP algorithm that incorporates a deep neural network to improve performance. The L-SBL-GAMP approach can achieve high accuracy while reducing the computational complexity compared to SBL-EM. However, the performance of the L-SBL-GAMP approach may depend on the quality and quantity of training data used to train the neural network.

Overall, while the L-GAMP-SBL algorithm has a higher computational complexity than GAMP-SBL and reduces computational complexity than SBL-EM, the improved performance it provides may make it the best choice in many practical applications where accuracy is critical. The choice of algorithm ultimately depends on the specific requirements of the application and the trade-off between computational complexity and performance.

SBL-EM has a computational complexity of O(G3), where G is the number of grid matrices. The SBL-GAMP algorithm does not require an inverse operation. It only needs a simple multiplication operation, which can be scaled as O(NtG). Thus, its computational complexity gets considerably reduced.

The computational complexity of L-SBL- GAMP is dominated by the training process, which typically has a complexity of O(ETNtG), where E is the number of epochs, T is the number of training samples or layers, Nt is the number of transmit antenna and G is the number of grid matrix or dictionary matrix.

In general, the computational complexity of L-SBL- GAMP is higher than that of SBL- GAMP, due to the added complexity of the training process. However, this added complexity may be justified if the improved accuracy of L-SBL-GAMP leads to better performance in practice. It is also worth noting that there are techniques for reducing the computational complexity of the training process, such as using mini-batch training or pre-training the neural network on a smaller dataset before fine-tuning it on the full dataset.

The L-SBL- GAMP algorithm has a higher computational complexity than SBL- GAMP, as it involves additional processing steps required to train and use the deep neural network. However, the computational complexity of L-SBL-GAMP is still significantly lower than that of SBL-EM.

Overall, while the L-SBL- GAMP algorithm has a higher computational complexity than SBL-GAMP, it is still significantly more efficient than SBL-EM, and the improved performance it provides may make it the best choice in many practical applications.

Results

In this section, the proposed L-SBL-GAMP algorithm is used to estimate hybrid precoders/combiners in the hybrid architecture. We assume 64 antennas are employed both at the transmitting and receiving ends, and 4 RF chains are considered. The channel matrix dataset is generated. A random noise is added to the communication link, as it happens in the conventional communication system. A data point in the dataset is nothing, but a pair of noisy channel and codebook indices. This created dataset is used to train the proposed deep learning neural network, which is utilized to find the loss function. Our model is implemented using the Keras libraries [36] with a Theano backend. The Adam optimizer is used with a momentum of 0.5, a batch size of 512, and a 0.005 learning rate is used. The performance of the proposed L-SBL-GAMP algorithm is compared with the existing algorithms, such as OMP, SOMP, and SBL-EM. The following parameters are used to simulate the proposed and the existing algorithms.

Let the transmitting and receiving elements are considered as N = Nr = Nt = 64, where the count of RF chains is considered as , and several data streams are assumed for the simulation.

Fig 3 shows NMSE vs. SNR for the proposed and existing algorithms. The proposed L-SBL-GAMP algorithm provides better results compared to the OMP, SOMP, SBL-EM, and SBL-GAMP. In Fig 3, the accuracy of the optimal hybrid precoder design using SBL-EM is almost the same compared to the SBL-GAMP method, but computational complexity is high in SBL-EM compared to the SBL-GAMP. The proposed L-SBL-GAMP algorithm is an extension of the SBL- GAMP algorithm that incorporates a deep neural network to improve performance. The L-SBL-GAMP approach can achieve high accuracy while reducing the computational complexity compared to SBL-EM. Overall, the L-GAMP-SBL algorithm has a higher computational complexity than GAMP-SBL and lower computational complexity than the SBL-EM method. The choice of algorithm ultimately depends on the specific requirements of the application and the trade-off between computational complexity and performance.

The same parameters are used for both Figs 3 and 4. The spectral efficiency against SNR for the proposed L-SBL-GAMP and existing algorithms is shown in Fig 4. The proposed L-SBL-GAMP algorithm achieves better spectral efficiency with a small number of iterations compared to the other existing algorithms. The SBL-GAMP algorithm and SBL-EM method achieve approximately the same achievable rates, but the computational complexity is higher in the SBL-EM method compared to the SBL-GAMP and proposed L-SBL-GAMP algorithm.

Fig 5 shows the NMSE against the number of paths for the proposed algorithm. The path element L varies from 2 to 10. The large number of paths gives a high computational cost with high estimation error. This is due to the small neighboring angular distance. This leads to the uncertainty of adjacent points, which affects the estimation of hybrid precoder performance in mmWave systems. However, the proposed L-SBL-GAMP algorithm shows better performance compared to other existing algorithms.

The performance of the number of paths against the spectral efficiency with the proposed method is unveiled in Fig 6. The increased channel path decreases the spectral efficiency of the proposed algorithm. Moreover, increasing sparsity level leads to high errors in the precoder along with the high computational complexity in the mmWave channel.

thumbnail
Fig 6. Number of path Vs spectral efficiency (bits/s/Hz).

https://doi.org/10.1371/journal.pone.0289868.g006

In Fig 7, the Layer vs loss function is discussed for L-SBL-EM and the proposed L-SBL-GAMP algorithm. The proposed L-SBL-GAMP gives a reduction in the loss function compared to the L-SBL-EM network.

Figs 8 and 9 display the performance of the L-SBL-GAMP network concerning signal-to-noise ratio (SNR) versus spectral efficiency (SE) and SNR versus normalized mean squared error (NMSE) for varying numbers of RF chains. The results indicate that the proposed L-SBL-GAMP network achieves high spectral efficiency and superior accuracy as the number of RF chains increases. This improvement in spectral efficiency is due to the use of a larger set of random precoders, resulting in a decrease in estimation error through compressed sensing and an increase in the number of effective combining and precoding beam patterns. Consequently, it is preferable to use a larger number of RF chains to attain high spectral efficiency and reduce estimation error in mmWave systems. However, the use of a large number of RF chains increases the complexity of the mmWave architecture. Therefore, a trade-off between complexity and performance must be considered for real-time implementations.

thumbnail
Fig 8. SNR Vs spectral efficiency (bits/s/Hz) for various RF chain.

https://doi.org/10.1371/journal.pone.0289868.g008

The curves depicting the normalized mean squared error (NMSE) versus the number of antennas (N) for the proposed and existing approaches are presented in Fig 10. As N increases, the NMSE value decreases for both the proposed and existing algorithms. However, the least squares (LS) algorithm shows almost constant performance for different N values. The proposed L-SBL-GAMP network demonstrates superior performance compared to other approaches. Although the proposed method achieves better results with an increased number of antennas, there is a corresponding increase in computational complexity.

Fig 11 displays the number of antennas (N) versus the spectral efficiency for the proposed L-SBL-GAMP network and the existing method. the performance of the proposed L-SBL-GAMP network provides better spectral efficiency with an increase in the number of antennas for all hybrid systems.

Conclusion

In this paper, a deep learning architecture named L-SBL-GAMP is proposed to design optimal hybrid precoders/combiner for the mmWave massive MIMO hybrid architecture. The proposed L-SBL-GAMP algorithm provides better spectral efficiency, accuracy, and low computational complexity compared to the SBL-EM method. In the proposed L-SBL-GAMP algorithm, we utilized all the T layers for the computation of hybrid precoders/combiners under the sparse nature of the mmWave channel. The experimental and the numerical results showed the effectivities of the proposed deep learning-based method, where it could be seen that the proposed deep learning-based algorithm worked better than the existing state-of-the-art methods, such as LS, OMP, SOMP, SBL-EM, and SBL-GAMP, and provided increased spectral efficiency and less Normalized Mean Square Errors (NMSEs).

References

  1. 1. Kim K.S.; Kim D.K.; Chae C.B.; Choi S.; Ko Y.C.; Kim J.; et al. Ultrareliable and low-latency communication techniques for tactile Internet services. Proc. IEEE 2019, 107, 376–393.
  2. 2. Yan B., Wang G., Yu J., Jin X. and Zhang H., "Spatial-Temporal Chebyshev Graph Neural Network for Traffic Flow Prediction in IoT-Based ITS," in IEEE Internet of Things Journal, vol. 9, no. 12, pp. 9266–9279, 15 June15, 2022,
  3. 3. Khan A.A.; Uthansakul P.; Uthansakul M. Energy efficient design of massive MIMO by incorporating with mutual coupling. Int. J. Comm. Antenna Prop. 2017, 7, 198–207.
  4. 4. Ahmed A., Ahmed Q. Z., Almogren A., Haider S. K. and Rehman A. U., "Hybrid Precoding Aided Fast Frequency-Hopping for Millimeter-Wave Communication," in IEEE Access, vol. 9, pp. 149596–149608, 2021,
  5. 5. Dai L. and Zhang H., "Propagation-Model-Free Base Station Deployment for Mobile Networks: Integrating Machine Learning and Heuristic Methods," in IEEE Access, vol. 8, pp. 83375–83386, 2020,
  6. 6. Naqvi S. H. R., Ho P. H. and Peng L., "5G NR mmWave indoor coverage with massive antenna system," in Journal of Communications and Networks, vol. 23, no. 1, pp. 1–11, Feb. 2021,
  7. 7. Nusenu S. Y., Huaizong S. and Ye P., "Power Allocation and Equivalent Transmit FDA Beamspace for 5 G Mmwave NOMA Networks: Meta-Heuristic Optimization Approach," in IEEE Transactions on Vehicular Technology, 2022,
  8. 8. Arshad Jehangir, Rehman Abdul, Ateeq Ur Rehman Rehmat Ullah, and Seong Oun Hwang. 2020. "Spectral Efficiency Augmentation in Uplink Massive MIMO Systems by Increasing Transmit Power and Uniform Linear Array Gain" Sensors 20, no. 17: 4982. pmid:32887453
  9. 9. Rusek F., et al., “Scaling up MIMO: Opportunities and challenges with very large arrays,” IEEE Sig. Process. Mag., vol. 30, no. 1, pp. 40–60, Jan. 2013.
  10. 10. Pereira de Figueiredo F. A., "An Overview of Massive MIMO for 5G and 6G," in IEEE Latin America Transactions, vol. 20, no. 6, pp. 931–940, June 2022,
  11. 11. Rao Muhammad Asif Mustafa Shakir, Ateeq Ur Rehman Muhammad Shafiq, Rehan Ali Khan Wali Ullah Khan. “Performance evaluation of Spectral Efficiency for Uplink and Downlink Multi-Cell Massive MIMO Systems” in Journal of Sensors (JoS), vol. 2022, pp. 1–12.
  12. 12. Alkhateeb A., Leus G., and Heath R. Jr., “Limited Feedback Hybrid Precoding for Multi-User Millimeter Wave Systems,” IEEE Trans. Wireless Commun., vol. 14, no. 11, pp. 6481–6494, July 2015.
  13. 13. Rao Muhammad Asif Mustafa Shakir, Nebhen Jamel, Ateeq Ur Rehman Muhammad Shafiq and Choi Jin-Ghoo. “Energy Efficiency Trade-off with Spectral Efficiency in MIMO Systems” in Computers, Materials & Continua, vol. 70, No. 3, pp. 5889–5905, 2022.
  14. 14. Bogale T. E., and Le L. B., “Massive MIMO and mmWave for 5G Wireless HetNet,” IEEE Veh. Tech. Mag., vol. 11, no. 1, pp. 64–75, Mar. 2016.
  15. 15. Heath R. W Jr., Gonzalez-Prelcic N, Rangan S, Roh W, and Sayeed A, “An Overview of Signal Processing Techniques for Millimeter Wave MIMO Systems,” IEEE J. Sel. Topics in Sig. Process., vol. 10, no. 3, pp. 1–18, 2016.
  16. 16. Rao Muhammad Asif Jehangir Arshad, Shakir Mustafa, Noman Sohail M., Ateeq Ur Rehman. “Energy Efficiency Augmentation in Massive MIMO Systems Through Linear Precoding Schemes and Power Consumption Modeling” Wireless Communications and Mobile Computing., Volume 2020, pp: 1–13.
  17. 17. Ayach O. E., Rajagopal S., Abu-Surra S., Pi Z. and Heath R. W., "Spatially Sparse Precoding in Millimeter Wave MIMO Systems," in IEEE Transactions on Wireless Communications, vol. 13, no. 3, pp. 1499–1513, March 2014,
  18. 18. Méndez-Rial R., Rusu C., González-Prelcic N., and Heath R. W., “Dictionary-free hybrid precoders and combiners for mmWave MIMO systems,” in 2015 IEEE 16th International Workshop on Signal Processing Advances in Wireless Communications (SPAWC), pp. 151–155, Stockholm, Sweden, 2015.
  19. 19. Shoukath Ali K. and Sampath P., “Time Domain Channel Estimation for Time and Frequency Selective Millimeter Wave MIMO Hybrid Architectures: Sparse Bayesian Learning-Based Kalman Filter,” Wireless Personal Communications, vol. 117, no. 3, pp. 2453–2473, 2021.
  20. 20. Shoukath Ali K. and Sampath P., “Sparse Bayesian Learning Kalman Filter-based Channel Estimation for Hybrid Millimeter Wave MIMO Systems: A Frequency Domain Approach,” IETE Journal of Research, , 2021.
  21. 21. Rusu C., Méndez-Rial R., González-Prelcicy N., and Heath R. W., “Low complexity hybrid sparse precoding and combining in millimeter wave MIMO systems,” in IEEE International Conference on Communications (ICC), 2015. IEEE, 2015, pp. 1340–1345.
  22. 22. Alkhateeb A. and Heath R. W. Jr., “Frequency selective hybrid precoding for limited feedback millimeter wave systems,” IEEE Trans. Commun., vol. 64, no. 5, pp. 1801–1818, May 2016.
  23. 23. Alkhateeb A. and Heath R. W. Jr., “Gram Schmidt based greedy hybrid precoding for frequency selective millimeter wave MIMO systems,” in in Proc. of the IEEE International Conf. on Acoustics, Speech and Signal Processing (ICASSP), Shanghai, China, Mar. 2016.
  24. 24. Suryaprakash R. T., Pajovic M., Kim K. J., and Orlik P., “Millimeter wave communications channel estimation via Bayesian group sparse recovery,” in IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2016. IEEE, 2016, pp. 3406–3410.
  25. 25. Wipf D. P. and Rao B. D., “Sparse Bayesian learning for basis selection,” IEEE Transactions on Signal processing, vol. 52, no. 8, pp. 2153–2164, 2004.
  26. 26. Gorodnitsky I. F. and Rao B. D., “Sparse signal reconstruction from limited data using FOCUSS: A re-weighted minimum norm algorithm,” IEEE Transactions on signal processing, vol. 45, no. 3, pp. 600–616, 1997.
  27. 27. Chen S. S., Donoho D. L., and Saunders M. A., “Atomic decomposition by basis pursuit,” SIAM review, vol. 43, no. 1, pp. 129–159, 2001.
  28. 28. Ma J., Zhang S., Li H., Gao F. and Jin S., "Sparse Bayesian Learning for the Time-Varying Massive MIMO Channels: Acquisition and Tracking," in IEEE Transactions on Communications, vol. 67, no. 3, pp. 1925–1938, March 2019,
  29. 29. Liu Y., Zhang S., Gao F., Ma J. and Wang X., "Uplink-Aided High Mobility Downlink Channel Estimation Over Massive MIMO-OTFS System," in IEEE Journal on Selected Areas in Communications, vol. 38, no. 9, pp. 1994–2009, Sept. 2020,
  30. 30. Srivastava Suraj, Mishra Amrita, AnupamaRajoriya Aditya K. Jagannatham and GerdAscheid , “Quasi-Static and Time-Selective Channel Estimation for Block-Sparse Millimeter Wave Hybrid MIMO Systems: Sparse Bayesian Learning (SBL)-Based Approaches,” IEEE Transactions on Signal Processing, vol. 67, no. 5, pp. 1251–1266, 2018.
  31. 31. Srivastava Suraj; Mishra Amrita; Jagannatham Aditya K; GerdAscheid, SBL-Based Hybrid Precoder/ Combiner Design for Power and Spectrally Efficient Millimeter Wave MIMO Systems, in 2020 International Conference on Signal Processing and Communications (SPCOM), 28 August 2020.
  32. 32. Zou X., Li F., Fang J. and Li H., "Computationally efficient sparse Bayesian learning via generalized approximate message passing," 2016 IEEE International Conference on Ubiquitous Wireless Broadband (ICUWB), 2016, pp. 1–4,
  33. 33. Rangan S., “Generalized approximate message passing for estimation with random linear mixing,” in Proc. IEEE Int. Symp. Inf. Theory (ISIT), Saint Petersburg, Russia, Aug. 2011.
  34. 34. Al-Shoukairi M., Schniter P. and Rao B. D., "A GAMP-Based Low Complexity Sparse Bayesian Learning Algorithm," in IEEE Transactions on Signal Processing, vol. 66, no. 2, pp. 294–308, 15 Jan.15, 2018,
  35. 35. Vila J. P. and Schniter P., “Expectation-maximization Gaussian-mixture approximate message passing,” IEEE Trans. Signal Processing, no. 19, pp. 4658–4672, Oct. 2013.
  36. 36. Peter Rubin Jose, and Murthy Chandra R. "Learned-SBL: A deep learning architecture for sparse signal recovery." arXiv preprint arXiv:1909.08185 (2019).
  37. 37. Li X. and Alkhateeb A., "Deep Learning for Direct Hybrid Precoding in Millimeter Wave Massive MIMO Systems," 2019 53rd Asilomar Conference on Signals, Systems, and Computers, 2019, pp. 800–805,