Power Control and MSE Analysis

From 'Add Up the Signals' to a Power-Control Problem

Section §16.1 imposed a single design rule — bkhk=ηb_k h_k = \eta for all kk — and read off the MSE σ2/η2\sigma^2/|\eta|^2. Making the MSE small means making η|\eta| large. But η|\eta| is bounded by the weakest channel: to satisfy bk2Pk/E[sk2]|b_k|^2 \leq P_k / \mathbb{E}[|s_k|^2] for every user, one sets ηhkPk/E[sk2]|\eta| \leq |h_k| \sqrt{P_k / \mathbb{E}[|s_k|^2]} across all kk.

The point is that AirComp's MSE is a water-filling problem in reverse — the receive amplitude is limited by the worst channel, not the average. Strong channels sit below their power budgets. The section formalizes this, derives the zero-forcing and MMSE receivers, and characterizes when — if ever — dropping weak users helps.

,

Theorem: Zero-Forcing AirComp MSE

Fix a common receive amplitude target η>0\eta > 0. Users transmit with bk=η/hkb_k = \eta / h_k, subject to bk2σs2Pk|b_k|^2 \sigma_s^2 \leq P_k where σs2=E[sk2]\sigma_s^2 = \mathbb{E}[|s_k|^2]. Write γk=hk2Pk/σs2\gamma_k = |h_k|^2 P_k / \sigma_s^2 (per-user effective channel gain).

  1. Feasibility. The magnitude-alignment choice bk=η/hkb_k = \eta / h_k is feasible for every kk iff η2minkγk\eta^2 \leq \min_k \gamma_k.

  2. MSE. Under feasibility, the zero-forcing post-processor y^=r/η\hat{y} = r / \eta achieves MSE(η)  =  σ2η2.\mathsf{MSE}(\eta) \;=\; \frac{\sigma^2}{\eta^2}.

  3. Optimal receive amplitude. The MSE is minimized by η=minkγk\eta^{\star} = \sqrt{\min_k \gamma_k}, giving MSE  =  σ2minkγk.\mathsf{MSE}^{\star} \;=\; \frac{\sigma^2}{\min_k \gamma_k}.

,

Example: Computing the ZF MSE for Four Heterogeneous Users

Four users share an AirComp MAC. Channel gains hk2={1.0,0.6,0.3,0.1}|h_k|^2 = \{1.0, 0.6, 0.3, 0.1\}. Per-user budget Pk=1P_k = 1. Source variance σs2=1\sigma_s^2 = 1. Noise variance σ2=0.01\sigma^2 = 0.01. Compute η\eta^{\star} and MSE\mathsf{MSE}^{\star} under zero-forcing.

Definition:

MMSE AirComp Receiver

Fix transmit scalings bkb_k (not necessarily aligned). Let α=khkbk\alpha = \sum_k h_k b_k denote the sum of channel-scaled gains. The receiver's MMSE estimator for y=ksky = \sum_k s_k takes the form y^MMSE=cr\hat{y}_{\text{MMSE}} = c \cdot r, with cc chosen to minimize E[y^MMSEy2]\mathbb{E}[|\hat{y}_{\text{MMSE}} - y|^2].

Assuming i.i.d.\ sources (sks_k zero- mean with variance σs2\sigma_s^2), the optimal linear estimator is c  =  ασs2α2σs2+σ2.c^{\star} \;=\; \frac{\alpha^{*} \sigma_s^2}{|\alpha|^2 \sigma_s^2 + \sigma^2}. When the transmit scalings satisfy magnitude alignment (bkhk=ηb_k h_k = \eta, α=nη\alpha = n \eta) and σs2\sigma_s^2 is large, c1/αc^{\star} \to 1/\alpha — recovering zero-forcing. For small σs2\sigma_s^2 (low source energy), MMSE shrinks the estimate toward zero to trade bias for variance reduction.

When ZF Is Optimal and When It Is Not

Zero-forcing inverts the channel perfectly: y^=ksk+w/η\hat{y} = \sum_k s_k + \mathbf{w}/\eta — no bias, but the noise is amplified by 1/η1/\eta. MMSE accepts a small bias in exchange for lower noise, giving uniformly lower total MSE.

The bias-free property of ZF matters when the aggregated value feeds into a downstream estimator that assumes unbiased inputs (e.g., an FL gradient update). For applications that only care about total squared error, MMSE dominates. §16.4 returns to this choice through the FL lens of Chapter 17.

AirComp MSE vs. Transmit Power and User Count

Explore how the zero-forcing AirComp MSE scales with the common per-user transmit budget PP and the number of users nn. Channels are drawn from a Rayleigh fading distribution; the minimum channel gain governs the MSE (Theorem 16.2.1). As nn grows, the minimum fades deteriorate in distribution — so MSE increases with nn, despite more signals superimposing. This exposes the worst-user bottleneck.

Parameters
30
10
-20

Theorem: Threshold User Selection

Let A[n]\mathcal{A} \subseteq [n] be any subset of users. Restricting AirComp to A\mathcal{A} yields aggregate kAsk\sum_{k \in \mathcal{A}} s_k with MSE(A)  =  σ2minkAγk.\mathsf{MSE}(\mathcal{A}) \;=\; \frac{\sigma^2}{\min_{k \in \mathcal{A}} \gamma_k}. The threshold scheduling rule A(τ)={k:γkτ}\mathcal{A}(\tau) = \{k : \gamma_k \geq \tau\} is Pareto-optimal: no other subset achieves both a smaller MSE and a larger A|\mathcal{A}|.

⚠️Engineering Note

Deploying AirComp Scheduling

Practical guidance for AirComp user selection:

  • Choose τ\tau from the target MSE. Given a tolerance MSEtol\mathsf{MSE}^{\text{tol}}, set τ=σ2/MSEtol\tau = \sigma^2/\mathsf{MSE}^{\text{tol}}. Users with γk<τ\gamma_k < \tau are excluded.

  • Fairness concerns. Threshold scheduling systematically excludes weak-channel users. In FL, this can introduce statistical bias (weak users often correlate with specific data distributions — rural clients, edge devices). Consider rotating the excluded set across rounds.

  • CSIT quality. Channel estimation error δk=h^khk\delta_k = \hat{h}_k - h_k translates into alignment error δkbk\delta_k \cdot b_k. A quick rule of thumb: keep δk2hk2/10|\delta_k|^2 \leq |h_k|^2 / 10 to stay within 0.50.5 dB of the ideal-CSIT MSE.

  • Truncate aggressively or run more rounds? An aggressive threshold yields lower MSE per round but excludes users; running more rounds with a lax threshold includes everyone but costs channel uses. The optimal trade-off is FL-application- specific (§17.3).

Practical Constraints
  • CSIT error must be h/10\leq |h|/\sqrt{10} for 0.50.5 dB MSE loss

  • Threshold τ\tau determines included users

  • Rotation across rounds mitigates bias

  • Trade round count against per-round threshold

📋 Ref: Yang-Jiang-Shi 2020; Cao-Yang-Jiang 2020

Key Takeaway

The AirComp MSE is dominated by the weakest user's effective channel gain γk=hk2Pk/σs2\gamma_k = |h_k|^2 P_k / \sigma_s^2. The zero-forcing receiver — the workhorse of AirComp in practice — achieves MSE=σ2/minkγk\mathsf{MSE}^{\star} = \sigma^2/\min_k \gamma_k. Threshold user selection traces a Pareto-optimal MSE-vs.-sample-size curve. In FL contexts (Chapter 17), this trade-off is a core design knob: decreasing MSE may cost statistical representativeness.

Quick Check

In zero-forcing AirComp with n=4n = 4 users, the effective channel gains are γk={0.8,0.5,0.2,0.4}\gamma_k = \{0.8, 0.5, 0.2, 0.4\} and σ2=0.02\sigma^2 = 0.02. What is MSE\mathsf{MSE}^{\star}?

0.02/0.8=0.0250.02 / 0.8 = 0.025

0.02/0.2=0.10.02 / 0.2 = 0.1

0.02/γˉ=0.02/0.4750.0420.02 / \bar{\gamma} = 0.02 / 0.475 \approx 0.042

0.02n/kγk0.02 \cdot n / \sum_k \gamma_k