Sum-Rate Maximization via WMMSE

Sum Rate: The Utilitarian Objective

Sum rate Rsum=βˆ‘klog⁑2(1+SINRk)R_{\text{sum}} = \sum_k \log_2(1 + \text{SINR}_k) is the natural metric when the goal is aggregate throughput β€” the operator's perspective on spectral efficiency. It rewards giving more resources to stronger users (water-filling), and the RIS can amplify this effect by coherently aligning the strongest user's channel. Sum-rate maximization is the right objective for hot-spot deployments where average throughput matters more than per-user fairness. We develop the WMMSE-AO algorithm in detail, since it is reused throughout this chapter and the advanced architectures (Ch. 11–13).

Definition:

The RIS-Aided Sum-Rate Problem

The RIS-aided sum-rate maximization problem is

β€…β€Šmax⁑W,Ξ¦βˆ‘k=1Klog⁑2 ⁣(1+∣hk,effHvk∣2βˆ‘jβ‰ k∣hk,effHvj∣2+Οƒ2)β€…β€Š\boxed{\;\max_{\mathbf{W}, \boldsymbol{\Phi}} \sum_{k=1}^{K} \log_2\!\left(1 + \frac{|\mathbf{h}_{k,\text{eff}}^H \mathbf{v}_{k}|^2}{\sum_{j \neq k} |\mathbf{h}_{k,\text{eff}}^H \mathbf{v}_{j}|^2 + \sigma^2}\right)\;}

subject to tr(WHW)≀Pt\text{tr}(\mathbf{W}^{H} \mathbf{W}) \leq P_t and βˆ£Ο•n∣=1|\phi_n| = 1 for all nn. The outer optimization is over the joint variable (W,Ξ¦)(\mathbf{W}, \boldsymbol{\Phi}); the problem is non-convex in both directions.

Theorem: WMMSE Equivalence for MU-RIS

Fix Ξ¦\boldsymbol{\Phi}. The RIS sum-rate problem is equivalent to the WMMSE problem

min⁑W,{uk},{wk}βˆ‘kwkekβˆ’log⁑2wks.t.Β tr(WHW)≀Pt,\min_{\mathbf{W}, \{u_k\}, \{w_k\}} \sum_k w_k e_k - \log_2 w_k \quad \text{s.t.}\ \text{tr}(\mathbf{W}^{H} \mathbf{W}) \leq P_t,

where ek=E[∣ukβˆ—ykβˆ’sk∣2]e_k = \mathbb{E}[|u_k^* y_k - s_k|^2] is user kk's MSE under scalar combiner uku_k, and wk>0w_k > 0 is the WMMSE weight. The optima satisfy

  • uk⋆=hk,effHvk/(βˆ‘j∣hk,effHvj∣2+Οƒ2)u_k^\star = \mathbf{h}_{k,\text{eff}}^H \mathbf{v}_{k} / (\sum_j |\mathbf{h}_{k,\text{eff}}^H \mathbf{v}_{j}|^2 + \sigma^2),
  • wk⋆=1/ek⋆w_k^\star = 1 / e_k^\star,
  • vk⋆=wkuk⋆(βˆ‘jwj∣uj∣2hj,effhj,effH+ΞΌI)βˆ’1hk,eff\mathbf{v}_{k}^\star = w_k u_k^\star (\sum_j w_j |u_j|^2 \mathbf{h}_{j,\text{eff}} \mathbf{h}_{j,\text{eff}}^H + \mu \mathbf{I})^{-1} \mathbf{h}_{k,\text{eff}},

where ΞΌβ‰₯0\mu \geq 0 is the Lagrange multiplier for the power constraint (found by bisection).

The WMMSE identity (Chapter 5) applies unchanged when the channels are any complex matrices, including those reshaped by the RIS. We reuse the WMMSE trick here: introduce auxiliary variables (uk,wk)(\mathbf{u}_k, w_k) and convert the hard log-sum problem into a block-convex problem with closed-form updates.

,

AO for MU-RIS Sum-Rate Maximization

Complexity: O(Tβ‹…(TWMMSEKNt3+Cpassive))O(T \cdot (T_{\text{WMMSE}} K N_t^{3} + C_{\text{passive}})), T∼20T \sim 20
Input: channels {hk,d,hk,2,H1}\{\mathbf{h}_{k,d}, \mathbf{h}_{k,2}, \mathbf{H}_1\}, power PtP_t, tolerance Ο΅\epsilon.
Output: (W⋆,Φ⋆)(\mathbf{W}^\star, \boldsymbol{\Phi}^\star) achieving a local optimum.
1. Initialize Ξ¦(0)\boldsymbol{\Phi}^{(0)} (all ones or random unit-modulus).
2. Repeat t=0,1,2,…t = 0, 1, 2, \ldots:
3. \quad Compute effective channels hk,eff(t)\mathbf{h}_{k,\text{eff}}^{(t)}.
4. \quad Active (WMMSE) inner loop:
5. \quad\quad Update {uk}\{u_k\} (MMSE), {wk}\{w_k\} (from MSE), {vk}\{\mathbf{v}_{k}\} (regularized least-squares + power bisection).
6. \quad\quad Iterate until sum-rate plateau (∼5\sim 5-1010 inner iterations).
7. \quad Passive (RIS) update: use Ch. 6 algorithm on the multi-user QCQP
maxβ‘βˆ£Ο•n∣=1βˆ‘kwk{2β„œ(...)βˆ’...}\max_{|\phi_n|=1} \sum_k w_k \{2\Re(...) - ...\}, a sum of weighted quadratics.
8. \quad Check convergence: ∣Rsum(t+1)βˆ’Rsum(t)∣<Ο΅|R_{\text{sum}}^{(t+1)} - R_{\text{sum}}^{(t)}| < \epsilon.
9. return (W,Ξ¦)(\mathbf{W}, \boldsymbol{\Phi}).

The passive subproblem in multi-user setting has rank-KK quadratic form (one term per user in the objective) β€” higher rank than single-user, so element-wise BCD has larger gaps. Manifold or SDR preferred for Kβ‰₯4K \geq 4.

Sum Rate vs. Number of Users

Sweep the number of users KK from 1 to 16 and plot the RIS-aided sum rate alongside the no-RIS baseline. The RIS gain grows with KK up to a saturation point (where the DoF floor min⁑(Nt,K)\min(N_t, K) caps the multiplexing gain). Increase NN to see the saturation shift upward.

Parameters
64
8
10
15

Example: WMMSE Iteration for a 2-User System

A 2-user RIS-aided MISO system has Nt=4N_t = 4, N=16N = 16, Pt/Οƒ2=10Β dBP_t / \sigma^2 = 10\text{ dB}. Given initial Ξ¦(0)=I\boldsymbol{\Phi}^{(0)} = \mathbf{I}, execute one WMMSE inner iteration and one RIS element-wise sweep; check that the sum rate increased.

Sum Rate Favors the Strong

At sum-rate optimum, stronger users receive more power and their channels are further amplified by the RIS. The optimization is Matthew-effect: the rich get richer. Users with poor direct channels get virtually none of the power budget unless the RIS can make them competitive. This is not a flaw of the algorithm β€” it is the efficient-market outcome of sum-rate maximization. If you dislike the outcome, switch objectives (max-min fairness, Section 7.3, or proportional-fair).

πŸ”§Engineering Note

WMMSE Runtime in MU-RIS Deployment

Each WMMSE inner iteration is O(Nt3+KNt2)O(N_t^{3} + K N_t^{2}). For Nt=16,K=8N_t = 16, K = 8: ∼104\sim 10^4 flops, under 10 μs10\,\mu\text{s} on modern hardware. The RIS element-wise sweep cost dominates: O(NK)O(N K) for a sum-of-quadratics multi-user objective.

Typical total optimization time per coherence block for N=256,K=4,Nt=16N = 256, K = 4, N_t = 16: ∼2\sim 2 ms (element-wise), ∼50\sim 50 ms (manifold), ∼5\sim 5 s (SDR). Only the first two are real-time-feasible.

Practical Constraints
  • β€’

    WMMSE inner iterations: 5-10 for Ο΅=10βˆ’3\epsilon = 10^{-3} convergence.

  • β€’

    Outer AO iterations: 10-20; warm-starting reduces to 3-5.

  • β€’

    Memory: O(NNtK)O(N N_t K) for effective-channel storage.

Common Mistake: Don't Drop Users at Sum-Rate Optimum

Mistake:

"At low SNR, sum rate drops weak users entirely (zero power). So I just remove them from the active list."

Correction:

The weak users are part of the problem: serving them eventually demands fairness, and their CSI is needed for active-inactive user tracking. More pragmatically: at even slightly higher SNR, dropped users might become feasible β€” and re-entry costs time. Modern RAN schedulers allocate a minimum per-user rate; WMMSE with a per-user rate floor enforces this without explicit user dropping. See Ch. 17 for deployment considerations.