Joint Design of Hybrid Beamforming and Phase Shifts for IRS-Assisted Multi-User mmWave Systems

Abstract

This paper presents a joint design approach for intelligent reflecting surface (IRS)-assisted multi-user millimeter-wave (mmWave) systems. Our goal is to maximize the sum-rate of all users by optimizing the hybrid beamforming at the base station and the low-resolution phase shifters (e.g., 1 bit) at the IRS. To address this, we first adopt a zero-force (ZF) technique to design fully-digital (FD) beamforming and develop a cross-entropy optimization (CEO) framework-based iterative algorithm to calculate IRS phase shifts. Specifically, in this framework, the probability distributions of IRS elements are updated by minimizing the CE, which can generate a solution close to the optimal one with a sufficiently high probability. Then, based on the obtained FD beamforming, an alternating minimization method is applied to acquire hybrid beamforming. Simulation results show that our proposed joint design scheme can achieve enhanced performance compared to the existing schemes while maintaining a lower computational complexity.

1. Introduction

Future communication systems are expected to support various emerging applications and services, such as human-computer interaction, autonomous driving, the internet of everything, artificial intelligence, and virtual reality [1]. These advanced services and applications necessitate high-rate data transmission. As a promising technology, millimeter wave (mmWave) communication with wider available bandwidth has the potential to offer gigabits-per-second communication data rates [2,3,4]. However, owing to the high carrier frequency, the mmWave signal experiences severe path loss as compared to the signal with lower frequency bands. On the other hand, the short wavelength and high directivity make mmWave signals vulnerable to blockage events, resulting in link interruption.

To overcome this challenge, intelligent reflecting surface (IRS), which is a reconfigurable planar array composed of massive passive reflecting elements, has been introduced into mmWave communication systems [5,6,7]. More precisely, by independently controlling the coefficient of each IRS element, the incident signal from the base station (BS) is reflected to the desired direction, thereby improving the quality of the received signals and coverage of mmWave communication. In addition, the IRS is also low-cost and energy-efficient without requiring any extra radio frequency (RF) chain.

Recently, IRS-assisted mmWave communications have drawn significant research attention. Specifically, the IRS-assisted single-user mmWave system was investigated in [8,9,10,11,12]. The authors of [8] adopted the semidefinite relaxation (SDR) technique to optimize the problem of reflecting beamforming design. Then, ref. [9] introduced a branch-and-bound (BnB) algorithm to obtain globally optimal solutions for both active and passive beamforming at the access point (AP) and the IRS. In [10], a geometric mean decomposition-based beamforming algorithm was presented to enhance bit error rate (BER) performance in IRS-aided mmWave hybrid MIMO systems. Leveraging the asymptotic orthogonality of array response vectors, the authors in [11] proposed a joint design of the IRS reflection matrix and hybrid beamforming. Ref. [12] reframed the IRS design as a more tractable optimization problem and developed a Riemannian manifold optimization (RMO)-based scheme. Further extending to multi-user scenarios, works such as refs. [13,14,15,16,17,18,19] explored the general IRS-aided multi-user communication systems. Ref. [13] introduced a gradient-projection method aimed at minimizing the mean square error (MSE) between the received symbols and the transmitted symbols. The weighted sum rate maximization problem was studied in [14] via a block coordinate descent algorithm to obtain high-quality suboptimal solution. With a focus on double IRS configurations, ref. [15] proposed an alternating optimization strategy for designing transmit beamforming at the BS and phase shifts at the IRSs. Notably, the aforementioned studies primarily assume continuous phase shifters (PSs), which are often impractical due to hardware cost and power constraints. In response, ref. [16] proposed a successive refinement (SR) algorithm that iteratively optimizes discrete phase shifts element by element. Then, ref. [17] developed a low-convexity machine learning-based beamforming design, and ref. [18] proposed a two-layer penalty-based algorithm to decouple optimization variables in SINR constraints, incorporating manifold optimization for discrete phase shift design. To further enhance system performance under the SR algorithm, ref. [19] presented a coordinate descent method (CDM)-based approach for reflecting beamforming design. Nevertheless, this performance improvement comes at the cost of sacrificing the computational complexity.

In this paper, we focus on an IRS-assisted multi-user mmWave communication system. Our goal is to maximize the overall sum-rate for all users through the joint optimization of hybrid beamforming at the BS and low-resolution phase shifters (e.g., 1 bit) at the IRS. To tackle this challenging non-convex optimization problem, we first decouple the design of the IRS from the hybrid beamforming using the zero-force (ZF) technique. Then, we adopt the idea of the cross-entropy optimization (CEO) framework, which was developed originally for the machine learning domain, to address the IRS design subproblem. Specifically, this framework starts with random sampling based on the probability distributions of IRS elements. After that, it successively refines these probability distributions by minimizing the cross-entropy, progressively yielding a solution that converges toward the optimum with a sufficiently high probability. In addition, according to the obtained ZF beamforming, an alternating minimization method is applied to acquire hybrid beamforming. Simulation results demonstrate that our proposed scheme can achieve better performance than the existing schemes, while at a much lower computational complexity.

Notation: Throughout this paper, matrices are denoted by bold uppercase letters, vectors by bold lowercase letters, and scalars by regular italic letters. The symbols ${(·)}^T$, ${(·)}^H$, and ${(·)}^{-1}$ represent the transpose, Hermitian transpose, and matrix inverse, respectively. The Frobenius norm is expressed as ${\parallel ·\parallel }_F$, and ${\mathbf{I}}_n$ denotes the $n\times n$ identity matrix. The entry in the i-th row and j-th column of matrix $\mathbf{A}$ is written as ${\left[\mathbf{A}\right]}_{i,j}$. Statistical expectation is denoted by $\mathbb{E}[·]$, and ${\mathbb{C}}^{x\times y}$ represents the space of $x\times y$ complex-valued matrices.

2. System Model and Problem Formulation

2.1. System Model

We focus on an IRS-assisted multi-user mmWave system, as illustrated in Figure 1, where a BS equipped with N antennas and $N_{\mathrm{RF}}$ RF chains simultaneously serves with K single-antenna users through an IRS with M reflecting elements. In contrast to the fully-digital (FD) beamforming architecture, the BS employs a hybrid beamforming structure to lower hardware cost and power consumption. Assuming that the direct path between the BS and the users is blocked, the received signal at all K users, after the reflection from IRS, is given by the following:

$$\begin{array}{c}\hfill \mathbf{y}=\mathbf{R}\mathbf{\Phi }\mathbf{G}{\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}\mathbf{s}+\mathbf{n},\end{array}$$

(1)

where $\mathbf{s}\in {\mathbb{C}}^{K\times 1}$ denotes the transmitted symbol vector satisfying $\mathbb{E}\left[\mathbf{s}{\mathbf{s}}^H\right]=\mathbf{I}$. The matrices ${\mathbf{F}}_{\mathrm{D}}=[{\mathbf{d}}_1,\dots ,{\mathbf{d}}_K]\in {\mathbb{C}}^{N_{\mathrm{RF}}\times K}$ and ${\mathbf{F}}_{\mathrm{RF}}\in {\mathbb{C}}^{N\times N_{\mathrm{RF}}}$ are the digital baseband precoder and analog RF precoder, respectively. The vector $\mathbf{n}\in {\mathbb{C}}^{K\times 1}\sim \mathcal{CN}(0,{\sigma }_k^2\mathbf{I})$ represents the additive Gaussian noise. The channel matrix from the BS to the IRS is denoted by $\mathbf{G}\in {\mathbb{C}}^{M\times N}$, and $\mathbf{R}={[{\mathbf{r}}_1,\dots ,{\mathbf{r}}_K]}^H\in {\mathbb{C}}^{K\times M}$ is the channel matrix from the IRS to the users, with ${\mathbf{r}}_k\in {\mathbb{C}}^{M\times 1}$ being the channel vector to the k-th user. The phase shift matrix of the IRS is defined as $\mathbf{\Phi }=\mathrm{diag}\left(\mathbf{v}\right)\in {\mathbb{C}}^{M\times M}$, where $\mathbf{v}={\left\{v_m\right\}}_{m=1}^M={[e^{j{\theta }_1},e^{j{\theta }_2},\dots ,e^{j{\theta }_M}]}^T$ and ${\theta }_m\in (0,2\pi )$ is the phase shift introduced by the m-th IRS reflecting element.

The signal-to-interference-plus-noise ratio (SINR) at the k-th user can be expressed as follows:

$$\begin{array}{c}\hfill {\gamma }_k={\displaystyle \frac{|{\mathbf{r}}_k^H\mathbf{\Phi }\mathbf{G}{\mathbf{F}}_{\mathrm{RF}}{\mathbf{d}}_k{|}^2}{{\sum }_{i=1,i\ne k}^K{|{\mathbf{r}}_k^H\mathbf{\Phi }\mathbf{G}{\mathbf{F}}_{\mathrm{RF}}{\mathbf{d}}_i|}^2+{\sigma }_k^2}}.\end{array}$$

(2)

Note that the hybrid beamforming requires satisfying the normalized power constraint $\left|\right|{\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}{\left|\right|}_F^2\le \rho$, where $\rho$ is the transmit power. Considering the practical hardware implementation limit, analog precoding and IRS reflecting elements should be realized by using the discrete PSs. Each element of the Q-bit resolution discrete PSs is restricted to the values of a set ${\mathcal{F}}_Q=\{e^{j0},e^{j△\theta },\dots ,e^{j(2^Q-1)△\theta }\}$, where $△\theta ={\displaystyle \frac{2\pi }{2^Q}}$. Thus, we can obtain the constraints $v_m\in {\mathcal{F}}_{Q_1}$ and ${\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{i,j}\in {\mathcal{F}}_{Q_2}$, $\forall i,j$, where $Q_1$ and $Q_2$ denote the resolution of IRS and analog precoding, respectively.

2.2. Channel Model

To capture the limited scattering features of mmWave propagation, we adopt the Saleh–Valenzuela channel model. Assuming the channel state information (CSI) of all links can be obtained by existing channel estimation schemes [20], the channel model with $L_1$ propagation paths for the BS-IRS links can be written as follows:

$$\begin{array}{c}\hfill \mathbf{G}=\sqrt{{\displaystyle \frac{NM}{L_1}}}\sum _{l=1}^{L_1}{\alpha }_l{\mathbf{a}}_{\mathrm{IRS}}({\omega }_l^r,{\varphi }_l^r){\mathbf{a}}_{\mathrm{BS}}^H\left({\eta }_l^t\right),\end{array}$$

(3)

where ${\alpha }_l$ denotes the complex, ${\omega }_l^r,{\varphi }_l^r$ and ${\eta }_l^t$ are azimuthal angle of arrival (AoA), elevational AoA, and angle of departure (AoD) of the l-th path between the BS and the IRS, respectively. The functions ${\mathbf{a}}_{\mathrm{BS}}\left(\eta \right)$ and ${\mathbf{a}}_{\mathrm{IRS}}(\omega ,\varphi )$ are respectively the array steering vectors at the BS and the IRS, which can be given by the following:

$$\begin{array}{c}\hfill {\mathbf{a}}_{\mathrm{BS}}\left(\eta \right)={\displaystyle \frac{1}{\sqrt{N}}}{[1,\dots ,e^{j{\displaystyle \frac{2\pi d}{\lambda }}n\sin \left(\eta \right)},\dots ,]}^T,\end{array}$$

(4)

$$\begin{array}{c}\hfill {\mathbf{a}}_{\mathrm{IRS}}(\omega ,\varphi )={\displaystyle \frac{1}{\sqrt{M}}}{[1,\dots ,e^{j{\displaystyle \frac{2\pi d}{\lambda }}(x\sin \left(\omega \right)\sin \left(\varphi \right)+y\cos \left(\varphi \right))},\dots ,]}^T,\end{array}$$

(5)

where $0\le n\le (N-1)$, $0\le \{x,y\}\le (\sqrt{M}-1)$. $\lambda$ and d are the signal wavelength and the antenna spacing. The channel vector ${\mathbf{r}}_k\in {\mathbb{C}}^{M\times 1}$ between the IRS and the k-th user, can be written as follows

$$\begin{array}{c}\hfill {\mathbf{r}}_k=\sqrt{{\displaystyle \frac{M}{L_2}}}\sum _{l=1}^{L_2}{\beta }_{k,l}{\mathbf{a}}_{\mathrm{IRS}}({\omega }_l^t,{\varphi }_l^t),\end{array}$$

(6)

where $L_2$ is the propagation paths for the IRS-users links. ${\beta }_{k,l}$, ${\omega }_l^t$ and ${\varphi }_l^t$ are the complex, azimuthal AoD, and elevational AoD of the l-th path for the k-th user, respectively.

2.3. Problem Formulation

In this paper, our objective is to maximize the sum-rate of all the K users by jointly designing the digital baseband precoding matrix ${\mathbf{F}}_{\mathrm{D}}$, the analog precoding matrix ${\mathbf{F}}_{\mathrm{RF}}$, and the IRS phase shift vector $\mathbf{v}$. The optimization problem can be formulated as follows:

$$\begin{array}{c}{\mathcal{P}}_1:\underset{\{{\mathbf{F}}_{\mathrm{D}},{\mathbf{F}}_{\mathrm{RF}},\mathbf{v}\}}{\mathrm{maximize}}f({\mathbf{F}}_{\mathrm{D}},{\mathbf{F}}_{\mathrm{RF}},\mathbf{v})={\displaystyle \sum _{k=1}^K}{\log }_2(1+{\gamma }_k)\\ \mathrm{s}.\mathrm{t}.\left\{\begin{array}{c}\parallel {\mathbf{F}}_{\mathrm{RF}}{{\mathbf{F}}_{\mathrm{D}}\parallel }_F^2\le \rho ,\hfill \\ {\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{i,j}\in {\mathcal{F}}_{Q_2},\forall i,j,\hfill \\ v_m\in {\mathcal{F}}_{Q_1},\forall m.\hfill \end{array}\right.\end{array}$$

(7)

Such an optimization problem is intractable, owing to the correlation among ${\mathbf{F}}_{\mathrm{D}}$, ${\mathbf{F}}_{\mathrm{RF}}$, and $\mathbf{v}$. To simplify this problem, we first consider an IRS-assisted multi-user mmWave system with an FD beamforming architecture, namely, $\mathbf{F}={\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}=[{\mathbf{f}}_1,\dots ,{\mathbf{f}}_K]$. When the optimal $\mathbf{F}$ is obtained, we can adopt an alternating minimization method in [21] to search for a hybrid beamforming matrix (${\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}$) to approach the optimal $\mathbf{F}$. Thus, the corresponding problem can be rewritten as follows:

$$\begin{array}{c}{\mathcal{P}}_2:\underset{\{\mathbf{F},\mathbf{v}\}}{\mathrm{maximize}}f(\mathbf{F},\mathbf{v})={\displaystyle \sum _{k=1}^K}{\log }_2(1+{\gamma }_k)\\ \mathrm{s}.\mathrm{t}.\left\{\begin{array}{c}{\parallel \mathbf{F}\parallel }_F^2\le \rho ,\hfill \\ v_m\in {\mathcal{F}}_{Q_1},\forall m.\hfill \end{array}\right.\end{array}$$

(8)

However, problem ${\mathcal{P}}_2$ is still challenging due to the coupled $\mathbf{F}$ and $\mathbf{v}$. Therefore, we will develop a ZF-based CEO design algorithm to solve problem ${\mathcal{P}}_2$. The details are presented in the next section.

3. Algorithm Design

In this section, we propose a ZF-based CEO algorithm to design IRS phase shift and FD beamforming by an alternating iterative procedure. After that, we utilize an alternating minimization method to optimize the hybrid beamforming matrix so that it could approach the performance of the FD beamforming matrix.

3.1. ZF-Based CEO Algorithm Design

The CEO framework is a probabilistic model-based iterative optimization algorithm. A key advantage of this framework is to optimize large-scale discrete variable spaces, which are inherent to IRS design problems. In addition, it can achieve substantially lower computational complexity relative to conventional exhaustive search and branch-and-bound techniques. This framework consists of the following key steps: (i) Sampling: Generating several random samples of candidate solutions based on a specified probability distribution. (ii) Selection: By sorting the objective function values in descending order, some top-performing candidate solutions are selected as the elite samples. (iii) Updating: Based on the selected elites, the probability distribution parameters are updated by minimizing the CE. (iv) Iteration: Repeat sampling and updating until convergence.

To characterize the sampling behavior in the CEO framework, we first define a probability parameter set ${\left\{p_{m,q}\right\}}_{q=1}^{2^{Q_1}}$, where $p_{m,q}=\mathrm{Pr}(v_m=e^{jq△\theta })$ denotes the probability that $v_m$ selects the q-th element from set ${\mathcal{F}}_{Q_1}$, and ${\sum }_{q=1}^{2^{Q_1}}{p}_{m,q}=1$. In the initialization phase, we assume that all the IRS phase shifts belong to ${\mathcal{F}}_{Q_1}$ with equal probability. At the i-th iteration, we generate A candidate IRS phase shifts ${\left\{{\mathbf{v}}^{\left[a\right]}\right\}}_{a=1}^A$ according to the probability distribution $\mathcal{H}({\mathbf{v}}^{\left[a\right]};{\mathbf{P}}^{\left(i\right)})$, written as follows:

$$\begin{array}{c}\hfill \mathcal{H}({\mathbf{v}}^{\left[a\right]};{\mathbf{P}}^{\left(i\right)})=\prod _{m=1}^M\sum _{q=1}^{2^{Q_1}}{p}_{m,q}^{\left(i\right)}{\mathbb{1}}_{\{v_m^{\left[a\right]}={\left[{\mathcal{F}}_{Q_1}\right]}_q\}},\end{array}$$

(9)

where $\mathbf{P}$ is a matrix composed of $p_{m,q}$, $m\in \{1,\dots ,M\}$, $q\in \{1,\dots 2^{Q_1}\}$. ${\mathbb{1}}_{\{·\}}$ denotes the indicator function for an event, and ${\left[{\mathcal{F}}_{Q_1}\right]}_q$ is the q-th element of ${\mathcal{F}}_{Q_1}$. When any IRS phase shift ${\mathbf{v}}^{\left[a\right]}$ is given, the sum-rate maximization problem ${\mathcal{P}}_2$ can be written as follows:

$$\begin{array}{c}{\mathcal{P}}_3:\underset{\mathbf{F}}{\mathrm{maximize}}f\left(\mathbf{F}\right)={\displaystyle \sum _{k=1}^K}{\log }_2(1+{\gamma }_k)\\ {\mathrm{s}.\mathrm{t}.\parallel \mathbf{F}\parallel }_F^2\le \rho ,\end{array}$$

(10)

By using a ZF technique, the FD beamforming matrix can be computed by the following:

$$\begin{array}{c}\hfill {\mathbf{F}}_{\mathrm{ZF}}={\mathbf{H}}_e^H{\left({\mathbf{H}}_e{\mathbf{H}}_e^H\right)}^{-1},\end{array}$$

(11)

where ${\mathbf{H}}_e\triangleq \mathbf{R}\mathbf{\Phi }\mathbf{G}$. To meet the transmit power constraint, the optimal FD beamforming matrix is given as follows:

$$\begin{array}{c}\hfill {\mathbf{F}}^{★}={\displaystyle \frac{\sqrt{\rho }}{\parallel {\mathbf{F}}_{\mathrm{ZF}}{\parallel }_F}}{\mathbf{F}}_{\mathrm{ZF}}\end{array}$$

(12)

According to the obtained FD beamforming matrix ${\mathbf{F}}^{★}$, problem ${\mathcal{P}}_2$ can be reformulated as follows:

$$\begin{array}{c}{\mathcal{P}}_4:\underset{\mathbf{v}}{\mathrm{maximize}}f\left(\mathbf{v}\right)={\displaystyle \sum _{k=1}^K}{\log }_2(1+{\gamma }_k)\\ \mathrm{s}.\mathrm{t}.\mathbf{v}\in \{{\mathbf{v}}^{\left[1\right]},\dots ,{\mathbf{v}}^{\left[A\right]}\},\end{array}$$

(13)

Then, we calculate A sum-rate ${\left\{f\left({\mathbf{v}}^{\left[a\right]}\right)\right\}}_{a=1}^A$ corresponding to the candidate IRS phase shifts, and sort these values ${\left\{f\left({\mathbf{v}}^{\left[a\right]}\right)\right\}}_{a=1}^A$ in descending order, which is written as follows:

$$\begin{array}{c}\hfill f\left({\mathbf{v}}^{\left[1\right]}\right)\ge f\left({\mathbf{v}}^{\left[2\right]}\right)\ge \dots \ge f\left({\mathbf{v}}^{\left[A_{elite}\right]}\right)\ge \dots \ge f\left({\mathbf{v}}^{\left[A\right]}\right)\end{array}$$

(14)

Note that only the top-performing candidate solutions are considered in the CEO framework. Therefore, we choose $A_{elite}$ IRS phase shifts with the largest sum-rate as elites to update $p_{m,q}^{(i+1)}$ by minimizing CE, which is equivalent to the following [22]:

$$\begin{array}{c}{\mathcal{P}}_5:{\mathbf{p}}_{m,q}^{(i+1)}=\underset{{\mathbf{p}}_{m,q}^{\left(i\right)}}{\mathrm{argmax}}{\displaystyle \frac{1}{A}}{\displaystyle \sum _{a=1}^{A_{\mathrm{elite}}}}\ln \mathcal{H}({\mathbf{v}}^{\left[a\right]};{\mathbf{P}}^{\left(i\right)})\\ \mathrm{s}.\mathrm{t}.{\displaystyle \sum _{q=1}^{2^{Q_1}}}{p}_{m,q}^{\left(i\right)}=1,\forall m.\end{array}$$

(15)

After that, we adopt the Lagrange multiplier method to solve problem ${\mathcal{P}}_5$. Specifically, the Lagrange function associated with (15) can be expressed as follows:

$$\begin{array}{c}\hfill \mathcal{L}={\displaystyle \frac{1}{A}}\sum _{a=1}^{A_{\mathrm{elite}}}\ln \mathcal{H}({\mathbf{v}}^{\left[a\right]};{\mathbf{P}}^{\left(i\right)})+\sum _{m=1}^M{\mu }_m\left(\sum _{q=1}^{2^{Q_1}}{p}_{m,q}^{\left(i\right)}-1\right),\end{array}$$

(16)

where ${\mu }_m$ is the Lagrange multiplier. The partial derivatives of the Lagrange function on $p_{m,q}^{\left(i\right)}$ is derived as follows:

$$\begin{array}{c}\hfill {\displaystyle \frac{\partial \mathcal{L}}{\partial p_{m,q}^{\left(i\right)}}}={\displaystyle \frac{1}{A}}\sum _{a=1}^{A_{\mathrm{elite}}}{\mathbb{1}}_{\{v_m^{\left[a\right]}={\left[{\mathcal{F}}_{Q_1}\right]}_q\}}+{\mu }_m{p}_{m,q}^{\left(i\right)},\end{array}$$

(17)

Setting ${\displaystyle \frac{\partial \mathcal{L}}{\partial p_{m,q}^{\left(i\right)}}}=0$, we can obtain the Lagrange multiplier as follows:

$$\begin{array}{c}\hfill {\mu }_m=-{\displaystyle \frac{A_{elite}}{A}},\end{array}$$

(18)

The probability $p_{m,q}^{(i+1)}$ can be updated by the following:

$$\begin{array}{c}\hfill p_{m,q}^{(i+1)}={\displaystyle \frac{{\sum }_{a=1}^{A_{\mathrm{elite}}}{\mathbb{1}}_{\{v_m^{\left[a\right]}={\left[{\mathcal{F}}_{Q_1}\right]}_q\}}}{A_{elite}}},\end{array}$$

(19)

Finally, a smoothing process is adopted to avoid the local optimum, which is given by the following:

$$\begin{array}{c}\hfill {\mathbf{P}}^{(i+1)}={\varpi }^{(i+1)}\times {\mathbf{P}}^{(i+1)}+(1-{\varpi }^{(i+1)})\times {\mathbf{P}}^{\left(i\right)}.\end{array}$$

(20)

where $0<\varpi <1$ is the smoothing step size. Repeating the above procedure, the probability distribution $\mathcal{H}(\mathbf{v};\mathbf{P})$ will be refined to generate a solution close to the optimal one with a sufficiently high probability. The proposed ZF-based CEO algorithm design is summarized in Algorithm 1, where I is the maximum number of iterations.

Algorithm 1 Proposed ZF-based CEO algorithm for solving (8).

1: Input: $\mathbf{R},\mathbf{G},A,A_{\mathrm{elite}}$, and I. 2: Initialization: $i=0, p_{m,q}^{\left(0\right)}={\displaystyle \frac{1}{2^{Q_1}}}$. 3: for  $i = 0$  to  I  do 4:     Generate A candidate $\mathbf{v}$ randomly as ${\left\{{\mathbf{v}}^{\left[a\right]}\right\}}_{a=1}^A$ based on the $\mathcal{H}({\mathbf{v}}^{\left[a\right]};{\mathbf{P}}^{\left(i\right)})$; 5:     Compute A corresponding FD beamforming matrices ${\left\{{\mathbf{F}}^{\left[a\right]}\right\}}_{a=1}^A$ based on the effective channel ${\mathbf{H}}_e^{\left[a\right]}\triangleq \mathbf{R}\mathrm{diag}\left({\mathbf{v}}^{\left[a\right]}\right)\mathbf{G}$ as in (12); 6:     Calculate the sum-rate ${\left\{f\left({\mathbf{v}}^{\left[a\right]}\right)\right\}}_{a=1}^A$ using (8) and sort them in descending as  $$f\left({\mathbf{v}}^{\left[1\right]}\right)\ge f\left({\mathbf{v}}^{\left[2\right]}\right)\ge \dots \ge f\left({\mathbf{v}}^{\left[A_{\mathrm{elite}}\right]}\right)\ge \dots \ge f\left({\mathbf{v}}^{\left[A\right]}\right);$$ 7:    Select elites as ${\mathbf{v}}^{\left[1\right]},{\mathbf{v}}^{\left[2\right]},\dots ,{\mathbf{v}}^{\left[A_{\mathrm{elite}}\right]}$; 8:    Update ${\mathbf{P}}^{(i+1)}$ according to (20); 9:     $i=i+1$; 10: end 11: Output: IRS phase shift vector ${\mathbf{v}}^{\left[1\right]}$, FD beamforming matrix ${\mathbf{F}}^{★}$.

Remark 1.

It has been verified in [23,24,25,26] that through appropriate selection of the smoothing parameter in (20), the CEO-based framework, utilizing discrete sampling distributions, can achieve probabilistic convergence to the global optimum, guaranteeing a probability of 1 for obtaining the optimal IRS phase shift. Ref. [26] shows more details of the theoretical convergence analysis of the CEO framework.

3.2. Hybrid Beamforming Design

Based on the obtained IRS phase shift vector and the optimal FD beamforming matrix, the hybrid beamforming design problem can be formulated as follows:

$$\begin{array}{c}{\mathcal{P}}_6:\underset{\{{\mathbf{F}}_{\mathrm{D}},{\mathbf{F}}_{\mathrm{RF}}\}}{\mathrm{minimize}}\left|\right|{\mathbf{F}}^{★}-{\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}{\left|\right|}_F^2\\ \mathrm{s}.\mathrm{t}.\left\{\begin{array}{c}\parallel {\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}{\parallel }_F^2\le \rho ,\hfill \\ {\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{i,j}\in {\mathcal{F}}_{Q_2},\forall i,j,\hfill \end{array}\right.\end{array}$$

(21)

To simplify this problem, the hybrid beamforming matrix is rewritten as follows:

$$\begin{array}{c}\hfill {\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}={\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,1}{\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{1,:}+\dots +{\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,n}{\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{n,:}+\dots +{\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,N_{\mathrm{RF}}}{\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{N_{\mathrm{RF}},:}\end{array}$$

(22)

where $n\in \{1,2,\dots ,N_{\mathrm{RF}}\}$. Assuming that both ${\mathbf{F}}_{\mathrm{RF}}$ and ${\mathbf{F}}_{\mathrm{D}}$ are given except the column vector ${\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,n}$ and the row vector ${\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{n,:}$, the hybrid beamforming problem can be decomposed into $N_{\mathrm{RF}}$ sub-problems. The n-th subproblem is expressed as follows:

$$\begin{array}{c}{\mathcal{P}}_7:\underset{\{{\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{n,:},{\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,n}\}}{\mathrm{minimize}}\left|\right|{\mathbf{F}}_n-{\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}{\left|\right|}_F^2\\ \mathrm{s}.\mathrm{t}.\left\{\begin{array}{c}\parallel {\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}{\parallel }_F^2\le \rho ,\hfill \\ {\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{i,j}\in {\mathcal{F}}_{Q_2},\forall i,j,\hfill \end{array}\right.\end{array}$$

(23)

where ${\mathbf{F}}_n={\mathbf{F}}^{★}-{\mathbf{F}}_{\mathrm{RF}}{\mathbf{F}}_{\mathrm{D}}+{\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,n}{\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{n,:}$ is defined as an auxiliary matrix. Specifically, we adopt an alternating minimization method of [21] to optimize ${\left[{\mathbf{F}}_{\mathrm{RF}}\right]}_{:,n}$ and ${\left[{\mathbf{F}}_{\mathrm{D}}\right]}_{n,:}$, and then update ${\mathbf{F}}_n$. This process continues until all column vectors of ${\mathbf{F}}_{\mathrm{RF}}$ and all row vectors of ${\mathbf{F}}_{\mathrm{D}}$ are obtained. Notably, the alternating minimization procedure is well-behaved and robust, with its convergence to a critical point having been proven in prior work [21].

3.3. Comparison of Computational Complexity

In this subsection, we analyze the computational complexity of the proposed ZF-CEO algorithm and choose the CDM [19] and SR [16] algorithms for comparison. For the first stage of the proposed ZF-CEO algorithm, the dominant cost lies in computing the effective channel matrix (Step 5), which requires $\mathcal{O}\left(KA\right(MN+M\left)\right)$ operations. In the second stage, applying the zero-forcing technique to obtain the optimal fully-digital beamforming matrix dominates the complexity, with a cost of $\mathcal{O}\left(A(K^{2}N+K^3)\right)$. The overall complexity of the ZF-CEO algorithm is $\mathcal{O}\left(A(KMN+KM+K^{2}N+K^3)\right)$. For the SR algorithm, the total complexity is $\mathcal{O}\left(2^{Q_1}(KMN+K^{2}M+K^3)\right)$. The total complexity of the CDM algorithm is $\mathcal{O}\left(2^{Q_1}(M{N}^3+2KM{N}^2+MN+KMN)\right)$. Assuming $\{M,N\}\gg K$, the dominant computational complexity of each algorithm is summarized in

Table 1. Comparison of computational complexity.

Algorithm	Dominant Complexity (per Iteration)	Runtime (s)
ZF-CEO	$\mathcal{O}\left(A(KMN+KM+K^{2}N+K^3)\right)$	1.26
SR	$\mathcal{O}\left(2^{Q_1}(KMN+K^{2}M+K^3)\right)$	1.32
CDM	$\mathcal{O}\left(2^{Q_1}(M{N}^3+2KM{N}^2+MN+KMN)\right)$	1.38

. It can be observed that the complexity of the proposed ZF-CEO algorithm is of the same order as that of the SR algorithm, while being slightly lower than that of the CDM algorithm. This indicates that the computational complexity of our proposed algorithm is practically an acceptable cost for IRS-assisted multi-user mmWave communication systems. In addition, we present the runtime of the proposed ZF-CEO, SR, and CDM algorithms in Table 1. It can be seen that the runtime of our algorithm is comparable to that of the other two algorithms. This is because the iterative sampling and probability updates involved in the CEO framework introduce non-negligible constant overhead, despite the relatively lower theoretical complexity of our proposed algorithm.

4. Simulation Results

We conduct the simulation results to evaluate the performance of the proposed algorithm in this section. For simulation, a three-dimensional geometry is adopted: a uniform linear array (ULA) at the BS and a uniform planer array (UPA) at the IRS are located in the $x-z$ plane and the $y-z$ plane, respectively. K single-antenna users are randomly distributed within a circular area centered at (0, 148 m). Considering the channel modeling realism and the imperfect CSI, the random angular offset parameters ${ϵ}_1\in [-3^{\circ },3^{\circ }]$ and ${ϵ}_2\in [-5^{\circ },5^{\circ }]$ are, respectively, introduced to the AoA and AoD. Moreover, consistent with the parameter configuration established in [19], the other simulation parameter values are summarized in

Table 2. Simulation parameters.

Parameters	Assumption
BS location ($x_{\mathrm{BS}}$, $y_{\mathrm{BS}}$, $z_{\mathrm{BS}}$)	(2 m, 0, 10 m)
IRS location ($x_{\mathrm{IRS}}$, $y_{\mathrm{IRS}}$, $z_{\mathrm{IRS}}$)	(0, 148 m, 10 m)
Number of users	$K=4$
Number of antennas	$N=64$
Number of reflecting elements	$M=100$
Number of propagation paths	$L_1=L_2=7$
Azimuth AoA/AoD	$\omega \in (-{\displaystyle \frac{\pi }{2}},{\displaystyle \frac{\pi }{2}}),\eta \in (-{\displaystyle \frac{\pi }{2}},{\displaystyle \frac{\pi }{2}})$
Elevation AoA	$\varphi \in (-{\displaystyle \frac{\pi }{2}},{\displaystyle \frac{\pi }{2}})$
Transmit power	$\rho =30$ dBm
Noise power	${\sigma }^2=-90$ dBm

.

Figure 2 shows the sum-rate versus the transmit power for $Q_1=1$ bit, $Q_1=2$ bit, and $Q_1=5$ bit phase shift resolutions. The proposed ZF-CEO algorithm is evaluated against two benchmark schemes: (1) the CDM algorithm [19]; (2) the SR algorithm [16], which has been widely used as a performance baseline for IRS-assisted multi-user systems. As observed in Figure 2, the sum-rate rises with increasing transmit power across all algorithms. Furthermore, when the resolution of IRS $Q_1$ increases, the sum-rate performance will increase over the entire considered $\rho$ range. Most importantly, the proposed ZF-CEO algorithm outperforms the CDM and the SR algorithms when $Q_1=5$ bits, and the performance gap becomes more pronounced when $Q_1=1$ bit. For instance, at a transmit power of 20 dBm and $Q_1=1$ bit, the gain of ZF-CEO exceeds 30% and 10% compared to SR and CDM algorithms. In conclusion, the proposed ZF-CEO algorithm can achieve a reasonable trade-off. It offers a significant performance advantage in low-resolution scenarios (e.g., 1 bit) with only a marginal increase in practical runtime compared to the existing algorithms.

Next, we examine the convergence behavior of our proposed ZF-CEO algorithm, as illustrated in Figure 3. Figure 3 plots the sum-rate against the number of iterations I and the number of candidates A, with parameters set as $A_{elite}/A=0.2$, $Q_1=1$ bit, and $\rho =30$ dBm. It can be seen that the sum-rate first rapidly increases and then stabilizes after a modest number of iterations (e.g., $I=40$), confirming the efficiency of the proposed ZF-CEO algorithm. Moreover, while increasing A yields noticeable sum-rate gains when A is relatively small, this improvement diminishes once A exceeds a certain threshold. This suggests that a moderately large candidate size (e.g., $A=150$) is sufficient to achieve near-optimal performance.

Figure 4 illustrates the sum-rate of different algorithms against the number of IRS elements $M=\{16,36,64,100,144,196,256\}$, under $Q_1=1$ bit and $\rho =30$ dBm. As expected, the sum-rate performance of all algorithms grows as the number of IRS elements M. However, the performance gain diminishes as the number of IRS elements increases. In Figure 4, we can see that the proposed ZF-CEO algorithm achieves enhanced performance compared to the CDM and SR algorithms across the considered range of M, with the performance gap widening slightly as M grows larger. This trend confirms that the proposed ZF-CEO method offers greater benefits in systems equipped with a larger IRS.

In Figure 5, we plot the achievable sum-rate for $N=\{64,100,144,196\}$ when $Q_1=1$ bit and $\rho =30$ dBm. It is observed that the sum-rate of all algorithms improves with increasing N. This indicates that the sum-rate performance of the proposed ZF-CEO algorithm can be improved by deploying more antennas at the BS. As shown in Figure 5, the proposed ZF-CEO algorithm has better performance than the CDM and SR algorithms over the whole N range in consideration. For example, at $N=100$, the performance gains of the proposed ZF-CEO algorithm over the CDM and SR algorithms are approximately $4.47\%$ and $17.6\%$, respectively.

5. Conclusions

In this paper, we studied an IRS-assisted multi-user mmWave system, aiming to maximize the overall sum-rate through joint optimization of the BS hybrid beamforming and the low-resolution phase shifters (e.g., 1 bit) at the IRS. To tackle this non-convex problem, the IRS design was first decoupled from the hybrid beamforming process using a ZF technique. Then, we developed a CEO framework-based iterative algorithm to calculate the discrete phase shifts of IRS. Simulation results showed that our proposed joint design scheme can improve the sum-rate performance, particularly under low-resolution phase shifter constraints, while at a much lower computational complexity.