Fronthaul Quantization-Aware MU-MIMO Precoding for Sum Rate Maximization ††thanks: This work was supported by the Knut and Alice Wallenberg Foundation.
Abstract
This paper considers a multi-user multiple-input multiple-output (MU-MIMO) system where the precoding matrix is selected in a baseband unit (BBU) and then sent over a digital fronthaul to the transmitting antenna array. The fronthaul has a limited bit resolution with a known quantization behavior. We formulate a new sum rate maximization problem where the precoding matrix elements must comply with the quantizer. We solve this non-convex mixed-integer problem to local optimality by a novel iterative algorithm inspired by the classical weighted minimum mean square error (WMMSE) approach. The precoding optimization subproblem becomes an integer least-squares problem, which we solve with a new algorithm using a sphere decoding (SD) approach. We show numerically that the proposed precoding technique vastly outperforms the baseline of optimizing an infinite-resolution precoder and then quantizing it. We also develop a heuristic quantization-aware precoding that outperforms the baseline while having comparable complexity.
Index Terms:
Sum rate maximization, weighted minimum mean square error, quantization-aware precoding.I Introduction
Multi-user multiple-input multiple-output (MU-MIMO) systems enable high data rates through spatial multiplexing of multiple user equipments (UEs) on the same time-frequency resource [1]. A base station (BS) equipped with multiple antennas and channel state information (CSI) can transmit simultaneously to several UEs using different beamforming directivity to increase the sum rate, which is controlled by the precoding method. In precoding matrix design, it is most common to maximize the sum rate, or even the weighted sum rate, under a constraint on the total transmit power [2]; however, sum rate maximization is known to be NP-hard [3]. One popular approach for sum-rate maximization is the iterative weighted minimum mean square error (WMMSE) algorithm, which finds locally optimal solutions with affordable computational complexity [4]. This paper utilizes a novel iterative algorithm inspired by WMMSE.
A 5G BS typically consists of two main components: an advanced antenna system (AAS) and a baseband unit (BBU). The AAS is a box containing the antenna elements and their respective radio units (RUs). The BBU performs the digital processing related to the received uplink data and transmitted downlink data. The AAS and BBU are connected through a digital fronthaul. The integration of antennas and radios into a single box has made massive MU-MIMO practically feasible [5] and enabled the BBU to be virtualized in an edge cloud through migration to the centralized radio access network architecture [6]. The new implementation bottleneck is the limited fronthaul capacity and the quantization errors it creates.
Both the uplink/downlink data and combining/precoding coefficients are sent over this digital fronthaul and must be quantized to a finite resolution. This paper proposes a novel linear block-level quantization-aware precoding technique that maximizes the sum rate.
I-A Prior Work
Downlink MU-MIMO systems have been widely studied in previous literature regarding impairments in analog hardware [7] and the effect of low-resolution digital-to-analog converters [8]. These prior works are characterized by distortion created either in the RU, analog domain, or converters. Therefore, the transmitted signal is distorted after the precoding. The effect of limited fronthaul capacity is studied in [9], but the precoding design was not quantized. In [10], the authors proposed a fronthaul quantization-aware precoding design that minimizes the sum MSE, which will generally not maximize the sum rate. Many previous studies suggest designing a precoding matrix by maximizing the sum rate, often using the WMMSE approach; see [11, 12] and references therein. Nevertheless, they consider ideal hardware or other types of distortion than precoding quantization.
I-B Contributions
This paper proposes a transmit precoding design that finds a local optimum to the sum-rate maximization problem subject to a transmit power constraint over a limited-capacity fronthaul connection that only accepts precoding matrix elements from a discrete quantization codebook. The main contributions are:
-
•
We propose maximizing the sum rate by solving a quantization-aware precoding problem. As this mixed-integer problem is non-convex, we rewrite it following the iterative WMMSE algorithm to find a local optimum. Each iteration of the proposed iterative algorithm contains a new integer least-square problem that minimizes the weighted MSE at each UE. The solution is obtained in a new way inspired by sphere decoding (SD). We consider a reduced-complexity variation on the proposed algorithm where only the last iteration is quantization-aware.
-
•
We define quantization-unaware precoding as a baseline and then recommend a low-complexity heuristic algorithm to sequentially refine the quantization-unaware precoding columns to improve the sum rate. The complexity is comparable to quantization-unaware precoding; thus, massive MIMO scenarios can be effectively handled.
-
•
We provide numerical results to compare different quantization-aware algorithms with quantization-unaware precoding baseline in terms of sum rate with the correlated Rician fading and a uniform planar array (UPA).
II System model
We consider a single-cell MU-MIMO downlink system, where the BS contains an AAS with antenna-integrated radios and serves single-antenna UEs. The AAS is connected to a BBU through a limited-capacity fronthaul link, which is modeled as a finite-resolution quantizer. The precoding matrix is computed, and the data symbol vector is encoded at the BBU and then sent over the fronthaul to the AAS. As data symbols are bit sequences from a channel code, we can transmit them over the fronthaul without quantization errors as they are already quantized. We can then map the data symbols obtained from the BBU to modulation symbols at the AAS. However, the precoding matrix computed at the BBU based on CSI is quantized due to the digital fronthaul. The quantized precoding matrix is then multiplied with the UEs’ data symbols at the AAS, and finally, the product is transmitted wirelessly.
Before analyzing the proposed quantization-aware precoding in Section III, we introduce the problem formulation and quantization scheme in the following subsections.
II-A Problem Formulation
The BBU uses its available CSI to select the downlink precoding matrix . As the main focus of this work is on managing quantization errors in a precoding matrix that maximizes the sum rate, we consider perfect CSI. However, the same algorithms can be used if the BBU has imperfect CSI but treats it as perfect. We postpone the channel estimation part for future work. The transmitted data symbol to the UE is denoted by , which has zero mean and normalized unit power, and the corresponding channel vector represents a narrowband channel and might be one subcarrier of a multi-carrier system. The algorithm developed in this paper can be applied individually to each subcarrier. The received signal at the UE is
(1) |
where is the quantized linear precoding vector for UE and represents the independent additive complex Gaussian receiver noise with power . For later use, we define the total received signal as , the data symbols vector as ( is the finite set of constellation points such as a QAM alphabet), the channel matrix as , and the precoding matrix as .
The fronthaul quantization alphabet set is defined as
(2) |
We assume the same quantization alphabet is used for the real and imaginary parts. Here contains the set of real-valued quantization labels, denotes the number of quantization levels, and is the number of quantization bits per real dimension. Note that becomes the complex-number set in the case of infinite resolution. The quantized precoding matrix and the data symbols vector are sent separately over the fronthaul. The precoded signal vector is calculated at the AAS as , where the scaling factor is computed at the AAS, denotes the maximum transmit power, and denotes the matrix trace. This scaling factor ensures that the , where is the Euclidean norm so that the maximum power is always utilized, despite the finite-resolution quantization.
The achievable rate is , where the signal-to-interference-plus-noise-ratio (SINR) depends on the precoding matrix as
(3) |
We want to maximize the sum rate of this downlink channel under the mentioned maximum transmit power constraint, and we define this problem as
[2] P∈P^M ×K ∑_k=1^K log_2 (1 + SINR_k (P))P_1: \addConstrainttr(PP^H) ≤q, where the optimization variable is with elements for and . Problem is not convex since the utility is non-concave and the search space is discrete, so it is hard to find the optimal solution [11].
II-B Quantization Scheme
We are modeling the digital fronthaul as a quantizer. In practice, uniform quantization is often used, so we assume our quantizer function is a symmetric uniform quantization with step size . Each entry of the quantization labels is defined as
(4) |
Furthermore, we let , where , specify the set of the quantization thresholds. For uniform quantizers, the thresholds are
(5) |
The quantizer function can be uniquely described by the set of quantization labels and the set of quantization thresholds . The quantizer maps an input to the quantized output , where the set is defined in (2), if and . The step size of the quantizer should be chosen to minimize the distortion between the quantized output and the unquantized input. The optimal step size depends on the dynamic range of the input, which in our case depends on the precoding scheme and channel model. We select the step size to minimize the distortion under the maximum-entropy assumption that each input element to the quantizer is distributed , where the variance is selected so that the sum power of the elements matches with the power constraint (II-A). The corresponding optimal step size for the normal distribution was numerically found in [13].
III Proposed WMMSE Algorithm
As the optimization problem is non-convex, the global optimal solution is challenging to find. We instead target finding a local optimum. Inspired by [4], we will rewrite as an equivalent iterative WMMSE problem for which a local optimum can be found through alternating optimization. In the following, we decompose this equivalent optimization problem into a sequence of convex subproblems.
Let denote the estimate at UE of the transmitted data symbol . It is obtained from the received signal using the receiver gain (also known as the precoding factor [8]). For a given receiver gain, the MSE in the data detection as UE becomes
(6) |
The MSE in (6) is a convex function of . We can select the value of that minimizes the MSE for given as
(7) |
By plugging optimal into (6), we can see that is equal to .
Now by defining the auxiliary weight , we formulate a weighted sum MMSE problem subject to the same total transmit power constraint as in (II-A):
[2] P∈P^M ×K , β,d ∑_k=1^K ( d_k e_k(P, β_k) - log_2 (d_k)) P_2: \addConstrainttr(PP^H) ≤q, where is a vector containing all receiver gains and is a vector containing all the UE weights in the weighted MSE. The problem is equivalent to in the sense that the optimal is the same for both problems. This equivalence comes from the fact that the optimal weight for the UE is
(8) |
so (III) then becomes .
The cost function in (III) is convex in each individual optimization variable, which is the key reason for considering this equivalent problem formulation.
For fixed and (e.g., calculated as in (7) and (8)), we have the WMMSE problem {mini!}[2] P∈P^M ×K∑_k=1^K d_k e_k(P, β_k) P_3: \addConstrainttr(PP^H) ≤q, which is a mixed-integer convex problem. It can be solved using general-purpose methods, such as CVX [14]. By iterating between updating using (7), using (8), and by solving , we obtain a block coordinate descent algorithm that will converge to a stationary point (for the same reasons as in [4]). This algorithm is summarized in Figure 1.
We can initialize the algorithm using any precoding matrix, including those obtained using classical infinite-resolution precoding schemes. The Wiener filtering (WF) precoding scheme (also known as regularized zero-forcing) is the most desirable one in this context since it can be derived by minimizing the sum MSE [15]. Hence, we suggest setting the initial precoding matrix as .
III-A Efficient Implementation of Quantization-Aware Precoding
The main complexity in the proposed algorithm originates from solving . Instead of using a general-purpose solver, we will propose an efficient dedicated algorithm. We can rewrite the objective function (III) as
(9) |
where is a diagonal matrix with elements from the vector with UE weights on the main diagonal. The expression in (9) can be expanded as
(10) |
where we introduce the notation .
We first notice that is a so-called integer least-squares problem due to uniform quantizer that we chose. The search space is a scaled finite subset of the infinite integer lattice. A technique that has previously been proposed as an efficient algorithm to solve closest lattice point problems in the Euclidean sense is called sphere decoding (SD). SD has significantly lower average computational complexity than a naive exhaustive search [16]. The basic principle of SD is to reduce the number of search points in a skewed lattice that lies within a hypersphere of radius , which can speed up the process of finding the solution without loss of optimality. In this paper, we consider the Schnorr-Euchner SD (SESD) algorithm [17], where the enumeration sorts candidate symbols in a zig-zag manner. SESD improves the basic SD algorithm by first checking the smallest child node of the parent node in each layer because the first found the feasible solution is often quite suitable and quickly enables a reduction of the search radius. Thus, many branches can be pruned, and the calculation complexity can be further lowered.
We propose using the SESD algorithm to approximately solve for fixed values of and . However, the classical SD framework does not contain any power constraint. To adapt our problem to match with the form required by SD algorithms, we need to proceed as follows. First, we rewrite the objective function (10) using the Lagrange multiplier as
(11) |
When minimizing (11) with respect to and , we can drop the constant term and have problem , given in (III-A) at the top of the next page. Although strong duality does not hold for the integer least-squares problem , provides a good approximation. For solving , we can utilize the SD algorithm for a fixed value of . We then make a bisection search over to find the value that gives a solution that satisfies the power constraint in (III) near equality.
For a fixed value of , and by vectorizing and using , we obtain ,
—l— P∈P^M ×K, λ≥0 tr(P^H(H^HD^H DH + λI_M )P- diag(d) DHP - (diag(d) DHP)^H ) - λq .P_4:
—l— p_i∈P^M , i=1,…,K ∑_i=1^K ( p_i^H ( H^HD^H DH+λI_M )p_i -f_i^Tp_i - ( f_i^Tp_i)^H), P_5:
which finds a suboptimal precoding matrix and has separable objective functions that each only depends on one of the optimization variables. This feature enables parallel optimization of for . Thus, in addition to the more efficient search strategy, the reformulation of problem also significantly reduces the dimension of each subproblem [10]. By defining , we can obtain the equivalent formulation of each term of the objective function in (III-A) as
(12) |
where is obtained from the Cholesky decomposition and . We can now minimize (12) by the classical SESD algorithm. We call this approach the Proposed SD-based WMMSE algorithm.
III-B Quantization-Unaware Precoding
The conventional WMMSE-based algorithms in [11, 12] (among others) compute a precoding matrix from that maximizes the sum rate with infinite resolution and the available CSI. Hence, the naive baseline approach would be to compute such a precoding matrix and then quantize each entry using so that the result can be sent over the fronthaul. In this case, the BBU follows Fig. 1 but solves in the domain instead of , and becomes a continuous convex optimization problem that is solvable using any general-purpose convex solver. We will refer to this quantization-unaware precoding approach as the Unaware WMMSE. After the WMMSE algorithm has converged, the final precoding matrix is obtained by quantizing each entry as .
III-C Combined Quantization-Aware and -Unaware Precoding
Although the proposed SD approach is tailored for the problem, there is a limit to how large setups ( and ) it can handle before the run time of the Proposed SD-based WMMSE approach becomes an issue. As we run the SD algorithm in each iteration, the algorithm’s average complexity over iterations becomes for some [16].
An alternative would be to search for a precoding matrix in for iterations and then use the SD-based algorithm only for the final iteration. In this case, we will first identify UE gains and weights that are suitable for sum-rate maximization with infinite-resolution precoding and then compute the corresponding optimized quantization-aware precoding. The order of complexity would be for some . We call this method Half-aware WMMSE.
III-D Heuristic Quantization-Aware Precoding
Although the Half-aware WMMSE scheme has lower complexity than the Proposed SD-based WMMSE, the complexity grows exponentially with the number of antennas , which makes scenarios with many antennas intractable. Hence, we believe it should primarily be seen as a benchmark for designing quantization-aware precoding schemes with low complexity. In this subsection, we propose such a heuristic precoding scheme. The proposed scheme is an add-on to the Unaware WMMSE precoding. After we quantize the precoding matrix to obtain , we refine the elements sequentially. We consider the second closest quantization levels in both the real and imaginary dimensions according to Euclidean distance [10]. This search gives us three alternative ways of quantizing each element in the precoding matrix . We call this method the Heuristic quantization-aware.
As the matrix elements are refined sequentially, we must order the elements properly. We propose to start by updating the column of the quantized precoding matrix corresponding to the UE with the highest generated interference , where since this might improve the performance the most.111We have noticed experimentally that this leads to the largest improvement in sum rate at high SNR. Then for that specific UE , for each transmit antenna , we identify the four nearest points in to the element from the original unquantized precoding matrix . We evaluate the sum rate
(13) |
for the four different options obtained with while all other elements are fixed. We then replace the corresponding element in with the option that achieves the largest sum rate. The rest of the UEs are ordered based on decreasing generated interference, and the precoding elements are updated similarly.
IV Numerical Results
This section compares the sum rates achieved by the aforementioned precoding approaches as a function of the SNR. The sum rate is calculated using Monte Carlo simulations for the case of Gaussian signaling.
IV-A Channel Model
We consider spatially correlated Rician fading channels composed of a line-of-sight (LoS) path component and a non-line-of-sight (NLoS) path component as
(14) |
where is the Rician factor, while and are the LoS and NLoS components, respectively. We assume that the AAS is a UPA; thus, the LoS channel matrix can be modeled as in [18, Ch. 7].
We model the NLoS channel as , which is spatially correlated Rayleigh fading. The spatial correlation matrix is generated following the local scattering model from [18, Ch. 2]. The locations of the UEs are randomly generated with the same elevation angle but uniformly distributed azimuth angles , seen from the UPA.
IV-B Results and Discussion
We assume the UPA consists of entries in the form of a array. The number of quantization levels is , the number of UEs is , and the Rician factor is . The UEs have the same SNR, which is defined as , where is the common channel variance.
![Refer to caption](x1.png)
Fig. 2 presents the convergence behavior of the proposed WMMSE algorithm in Fig. 1 for the cases when is solved exactly using CVX (denoted by CVX-based WMMSE) and approximately using the proposed SD-based method. We consider dB and generate one user drop. The proposed algorithm reaches a stationary point after iterations. We notice that, despite the approximations made to lower the complexity in the SD-based method, the difference in the sum rate is negligible after convergence. The difference between the sum rate at the starting point and the stationary point approximately shows the improvement of the proposed sum-rate maximization algorithm compared to our previous results in [10], which considered sum MSE minimization.
![Refer to caption](x2.png)
Fig. 3 depicts the average sum rate as a function of the SNR for the different precoding schemes. The top curve, WMMSE (infinite res), considers the ideal case without quantization and outperforms all the quantized precoding schemes since the rate increases linearly (in dB scale) at high SNR. In all quantized precoding schemes, the sum rate converges to specific limits at high SNR since the interference cannot be canceled entirely due to the limited precoding resolution; i.e., the system is interference-limited at high SNR. The gap between WMMSE (infinite res) and our novel Proposed SD-based WMMSE precoding is remarkably smaller than the gap between Unaware WMMSE and WMMSE (infinite res), where we quantized the precoding matrix used for WMMSE (infinite res). Our algorithm provides twice the rate at high SNR. The lower-complexity Half-aware WMMSE approach also outperforms the Unaware WMMSE algorithm, but there is a substantial gap to Proposed SD-based WMMSE. This gap demonstrates the importance of computing quantization-aware weights in the WMMSE algorithm instead of relying on those obtained with infinite resolution. The proposed Heuristic quantization-aware precoding reaches nearly the same performance as Half-aware WMMSE and, thus, performs vastly better than Unaware WMMSE precoding. Note that the complexity of Heuristic quantization-aware is polynomial with and , just as Unaware WMMSE, and therefore is implementable in the same scenarios (e.g., massive MIMO).
V Conclusions
A 5G site often consists of an AAS connected to a BBU via a digital fronthaul with limited capacity. Hence, the precoding matrix that is computed at the BBU must be quantized to finite precision. We have proposed a novel WMMSE-based framework for quantization-aware precoding that finds a local optimum to the sum rate maximization problem. Moreover, a reduced-complexity SD algorithm was proposed to find an approximate but tight solution. We demonstrated that the sum rate can be doubled at high SNR by being quantization-aware both when selecting the weights in the WMMSE formulation and when selecting the precoding matrix. Besides, a heuristic quantization-aware precoding was proposed to outperform the quantization-unaware baseline while having comparable complexity. Finally, we note that the framework can be easily modified also to solve weighted sum rate problems.
References
- [1] D. Gesbert, M. Kountouris, R. W. Heath, C.-B. Chae, and T. Sälzer, “Shifting the MIMO paradigm,” IEEE Signal Process. Mag., vol. 24, no. 5, pp. 36–46, Sep. 2007.
- [2] E. Björnson, G. Zheng, M. Bengtsson, and B. Ottersten, “Robust monotonic optimization framework for multicell MISO systems,” IEEE Trans. Signal Proc., vol. 60, no. 5, pp. 2508–2523, 2012.
- [3] Y.-F. Liu, Y.-H. Dai, and Z.-Q. Luo, “Coordinated beamforming for MISO interference channel: Complexity analysis and efficient algorithms,” IEEE Trans. Signal Proc., vol. 59, no. 3, pp. 1142–1157, 2010.
- [4] Q. Shi, M. Razaviyayn, Z.-Q. Luo, and C. He, “An iteratively weighted MMSE approach to distributed sum-utility maximization for a MIMO interfering broadcast channel,” IEEE Trans. Signal Process., vol. 59, no. 9, pp. 4331–4340, 2011.
- [5] E. Björnson, L. Sanguinetti, H. Wymeersch, J. Hoydis, and T. L. Marzetta, “Massive MIMO is a reality—What is next? Five promising research directions for antenna arrays,” Digit. Signal Process., vol. 94, pp. 3–20, 2019.
- [6] M. Peng, C. Wang, V. Lau, and H. V. Poor, “Fronthaul-constrained cloud radio access networks: Insights and challenges,” IEEE Wirel. Commun., vol. 22, no. 2, pp. 152–160, 2015.
- [7] E. Björnson, J. Hoydis, M. Kountouris, and M. Debbah, “Massive MIMO systems with non-ideal hardware: Energy efficiency, estimation, and capacity limits,” IEEE Trans. Inf. Theory, vol. 60, no. 11, pp. 7112–7139, 2014.
- [8] S. Jacobsson, G. Durisi, M. Coldrey, T. Goldstein, and C. Studer, “Quantized precoding for massive MU-MIMO,” IEEE Trans. Commun., vol. 65, no. 11, pp. 4670–4684, 2017.
- [9] P. Parida, H. S. Dhillon, and A. F. Molisch, “Downlink performance analysis of cell-free massive MIMO with finite fronthaul capacity,” in VTC-Fall, Chicago, IL, USA, 2018, pp. 1–6.
- [10] Y. Khorsandmanesh, E. Björnson, and J. Jaldén, “Optimized precoding for MU-MIMO with fronthaul quantization,” arXiv preprint arXiv:2209.01868, 2022.
- [11] S. S. Christensen, R. Agarwal, E. De Carvalho, and J. M. Cioffi, “Weighted sum-rate maximization using weighted MMSE for MIMO-BC beamforming design,” IEEE Trans. Wirel. Commun., vol. 7, no. 12, pp. 4792–4799, 2008.
- [12] X. Zhao, S. Lu, Q. Shi, and Z.-Q. Luo, “Rethinking WMMSE: Can its complexity scale linearly with the number of BS antennas?” arXiv preprint arXiv:2205.06225, 2022.
- [13] D. Hui and D. L. Neuhoff, “Asymptotic analysis of optimal fixed-rate uniform scalar quantization,” IEEE Trans. Inf. Theory, vol. 47, no. 3, pp. 957–977, 2001.
- [14] M. Grant and S. Boyd, “CVX: Matlab software for disciplined convex programming, version 2.1,” 2014.
- [15] M. Joham, W. Utschick, and J. A. Nossek, “Linear transmit processing in MIMO communications systems,” IEEE Trans. Signal Process., vol. 53, no. 8, pp. 2700–2712, 2005.
- [16] J. Jaldén and B. Ottersten, “On the complexity of sphere decoding in digital communications,” IEEE Trans. Signal Process., vol. 53, no. 4, pp. 1474–1484, 2005.
- [17] E. Agrell, T. Eriksson, A. Vardy, and K. Zeger, “Closest point search in lattices,” IEEE Trans. Inf. Theory, vol. 48, no. 8, pp. 2201–2214, 2002.
- [18] E. Björnson, J. Hoydis, and L. Sanguinetti, “Massive MIMO networks: Spectral, energy, and hardware efficiency,” Foundations and Trends in Signal Processing, vol. 11, no. 3-4, pp. 154–655, 2017.