Source-linked AI summary
Efficient Beam Alignment in Millimeter Wave Systems Using Contextual Bandits
Morteza Hashemi, Ashutosh Sabharwal, C. Emre Koksal, Ness B. Shroff
TL;DR
Dynamic mmWave links require repeated beam alignment, but searching the full angular space incurs overhead. The paper uses contextual structure in a bandit model to reduce that search, reporting accurate optimal-beam detection and lower delay overhead than exhaustive search.
Problem
In dynamic mmWave systems, transmitter and receiver beams must be repeatedly aligned, while searching the full angular domain incurs high delay overhead.
Source: §I · p. 1Method
The paper models beam pairs as arms in a structured bandit and uses contextual correlation and unimodality to guide search and propose an asymptotically optimal algorithm.
Source: §I · pp. 1–2Results
More than 85% of iterations correctly predict the optimal beam direction, and delay overhead is significantly reduced versus exhaustive search.
Source: D. Beam Alignment Accuracy and Delay… · p. 7Takeaways & Limitations
The regret bound does not depend on the number of beam matchings, a property relevant to MIMO systems with many beam combinations.
Source: §I · p. 2, §VII · p. 8
Abstract
from arXiv · showhide
In this paper, we investigate the problem of beam alignment in millimeter wave (mmWave) systems, and design an optimal algorithm to reduce the overhead. Specifically, due to directional communications, the transmitter and receiver beams need to be aligned, which incurs high delay overhead since without a priori knowledge of the transmitter/receiver location, the search space spans the entire angular domain. This is further exacerbated under dynamic conditions (e.g., moving vehicles) where the access to the base station (access point) is highly dynamic with intermittent on-off periods, requiring more frequent beam alignment and signal training. To mitigate this issue, we consider an online stochastic optimization formulation where the goal is to maximize the directivity gain (i.e., received energy) of the beam alignment policy within a time period. We exploit the inherent correlation and unimodality properties of the model, and demonstrate that contextual information improves the performance. To this end, we propose an equivalent structured Multi-Armed Bandit model to optimally exploit the exploration-exploitation tradeoff. In contrast to the classical MAB models, the contextual information makes the lower bound on regret (i.e., performance loss compared with an oracle policy) independent of the number of beams. This is a crucial property since the number of all combinations of beam patterns can be large in transceiver antenna arrays, especially in massive MIMO systems. We further provide an asymptotically optimal beam alignment algorithm, and investigate its performance via simulations.
I. INTRODUCTION
This section frames beam alignment as a costly, recurring search problem in dynamic mmWave systems. It proposes using contextual structure in a bandit model to reduce search overhead and derives a beam-count-independent regret bound with an asymptotically optimal algorithm.
I. INTRODUCTION
- Directional communication requires aligned transmitter and receiver beams; alignment overhead scales with device mobility and beam-resolution product, while exhaustive search over N possible directions has O(N^2) complexity.With a 7 degree beam width, an 18 degree misalignment reduces link budget by around 17 dB.Source: §I · p. 1
- Successive beam alignments are stochastically correlated, and received power is approximately unimodal, allowing search near the highest-index beam rather than uniformly across directions.In the 16-direction example, the proposed search is limited to the neighborhood of the beam with maximum index.Source: §I · pp. 1–2, Figure 1 · p. 2
- The optimization objective is to maximize expected received energy within a given time period.Source: §I · p. 2
- The structured Multi-Armed Bandit model uses contextual information to exploit correlations among beam matchings instead of exploring them heuristically.The paper represents each transmit–receive beam pair as an arm and seeks to maximize expected received energy over a time horizon.Source: §I · p. 2
- The regret lower bound does not scale with the number of beam matchings, and the proposed Unimodal Beam Alignment algorithm is asymptotically optimal.The bound exploits the problem's structure, while UBA is based on OSUB.Source: §I · p. 2
A. Beam Alignment
This section contrasts contextual beam alignment with prior search approaches, frames structured rewards through multi-armed bandits, and models analog beam selection using a matrix of candidate beam pairs.
A. Beam Alignment
- The paper focuses on standalone mmWave systems, using contextual information to reduce beam-search space and overhead compared with prior approaches.Prior approaches described include clustered localization, hierarchical search, channel-structure methods, and scanning schemes.Source: A. Beam Alignment · p. 2
B. Multi-Armed Bandit
- The paper casts mmWave beam alignment as a unimodal bandit problem and derives its regret bound, building on bandit models with correlated rewards.Unimodal bandits capture correlation among arms' average rewards.Source: B. Multi-Armed Bandit · p. 3
A. System Model
- With analog beamforming, one direction is selected at a time; beam alignment then seeks the largest entry of Y, while the proposed method uses correlations across Y to reduce the search space.Exhaustive search examines all Nr × Nt entries; analog beamforming uses a single RF chain at each end.Source: A. System Model · p. 3
B. Problem Statement
The problem is to select a transmitter–receiver beam pair that maximizes expected received energy during a finite alignment period. Beam pairs are characterized by angular misalignment and the probability of successful matching.
B. Problem Statement
- Misalignment δ_i,j is the angular mismatch between transmitter beam i and receiver beam j.The candidate set contains K = N_t × N_r possible beam pairs.Source: B. Problem Statement · pp. 3–4
- A matching succeeds when received energy exceeds threshold τ; its success probability is θ_k, assumed time-invariant in a fixed situation.For fixed transmit power, received energy p_k is nonincreasing with misalignment.Source: B. Problem Statement · p. 4
- The policy selects a beam pair to maximize expected beacon energy within the finite alignment horizon T.Each beam examination takes an average time of T̄, limiting how many pairs can be tested.Source: B. Problem Statement · p. 4
IV. EQUIVALENT MULTI-ARMED BANDIT MODEL
The paper models each transmitter–receiver beam pair as a bandit arm whose reward reflects received energy. It uses correlation and approximate unimodality across misalignments to guide search toward high-reward beam pairs.
IV. EQUIVALENT MULTI-ARMED BANDIT MODEL
- Each transmitter–receiver beam pair is an arm with Bernoulli reward p_k on success and zero on failure; its mean reward is μ_k = p_kθ_k.There are K = N_t × N_r arms.Source: §IV · p. 4
A. Contextual Information
- Success probabilities decrease with misalignment, while expected received energy is modeled as approximately unimodal with a unique maximum.The experimental observation was made in clear line of sight; the model assumes highly directional arrays with negligible sidelobes.Source: A. Contextual Information · pp. 4–5
- A graph orders beam-pair arms by expected reward and provides a monotonically increasing path to the unique maximum, avoiding uniform exploration of every node.The structure lets a policy avoid pulling some non-optimal arms.Source: A. Contextual Information · p. 4
- In a 40 GHz clear-line-of-sight experiment, received power approximately followed a unimodal pattern as the receiver horn rotated through misalignment angles.The measurement used a stationary transmitter and a receiver swept in 2-degree increments.Source: A. Contextual Information · p. 5, Figure 5 · p. 5
A. Regret Analysis
Regret measures expected reward lost relative to an oracle over time T. The analysis contrasts a structural bound whose factor is independent of beam-pair count with an unstructured bound that grows linearly with that count.
A. Regret Analysis
- Regret is expected reward loss over T compared with an oracle policy, and uncorrelated arms yield best-algorithm regret of O(K log(T)).Here K is the number of arms, which equals the number of possible transmitter–receiver beam pairs.Source: A. Regret Analysis · p. 5
- The structural regret-bound factor c(θ) sums a constant number of terms independent of K, unlike c′(θ), which grows linearly with the number of beam pairs.Without exploiting structural properties, the regret lower bound uses c′(θ).Source: A. Regret Analysis · p. 5
- The UBA procedure terminates beam search when ψ(t) ≥ Ψ and then proceeds to communication using beam index k.The passage specifies the stopping condition but not a quantitative performance result for the algorithm.Source: A. Regret Analysis · p. 6
B. Unimodal Beam Alignment (UBA) Algorithm
UBA uses a leader-based, KL-UCB-style search restricted to nearby arms, then adds a peak-to-average stopping rule to complete beam alignment. The paper states that its regret matches the derived lower bound and that the algorithm is asymptotically optimal.
B. Unimodal Beam Alignment (UBA) Algorithm
- UBA selects the highest-index arm near the arm with the highest empirical reward, using a KL-UCB-style index.The algorithm is based on OSUB and uses the leader’s neighborhood to restrict selection.Source: B. Unimodal Beam Alignment (UBA)… · p. 6
- Theorem 2 states a finite-time regret bound for UBA, and the preceding text says its regret matches the lower bound.The supplied theorem passage does not include the bound’s formula.Source: B. Unimodal Beam Alignment (UBA)… · p. 6
- UBA terminates when the current beam’s received energy divided by the average of previously received energies reaches threshold Ψ.The algorithm then declares the current beam direction optimal; Ψ = 4 is described as acceptable for LOS detection, though the threshold may vary by environment.Source: B. Unimodal Beam Alignment (UBA)… · p. 6
- The complete alignment procedure alternates transmitter and receiver steering, using UBA in place of exhaustive search at each scan stage.One side starts quasi-omnidirectionally while the other scans, then their roles reverse.Source: B. Unimodal Beam Alignment (UBA)… · p. 6
VI. NUMERICAL RESULTS
The evaluation compares UBA with exhaustive search under directional and quasi-directional conditions, measuring regret and, with a stopping rule, accuracy and delay. With eight beam directions, UBA substantially reduces regret in both conditions.
A. Setup
- The simulations compare UBA with exhaustive search, which scans all directions, under directional and quasi-directional success-probability conditions.Directional conditions have very high or very low success probabilities; quasi-directional conditions have smaller variance in success probabilities.Source: A. Setup · p. 6
- The evaluation measures regret against an oracle and also examines beam-alignment accuracy and delay when peak-to-average termination is used.The transmitter beam is fixed while the receiver scans for the optimal direction.Source: A. Setup · p. 7
B. Regret Performance
- UBA greatly reduces regret versus exhaustive beam sampling in both directional and quasi-directional scenarios with 8 beam directions.Regret rises over time but at a decreasing rate, and both methods have lower regret in the directional scenario.Source: B. Regret Performance · p. 7
C. Scaling with the Size of System
UBA’s regret performance does not degrade between 8 and 16 beam pairs, and its stopping rule can reduce alignment delay substantially in simulations. The conclusion reiterates that the regret bound is independent of the number of beam combinations.
C. Scaling with the Size of System
- UBA performs better than exhaustive beam sampling at both K = 8 and K = 16, and its performance does not degrade as beam-pair count increases.Figure 7 presents regret for the two beam-pair counts.Source: C. Scaling with the Size of System · p. 7, Figure 7 · p. 7
D. Beam Alignment Accuracy and Delay Overhead
- 128 time slots for exhaustive scanning fall to an average of 12.5 with UBA, at the cost of some error in detecting the optimal beam.For 8 beams, more than 85% of iterations correctly identify the optimal direction; most detections lead to beam 1 in fewer than 5 slots.Source: D. Beam Alignment Accuracy and Delay… · p. 7, Figure 9 · p. 8, Figure 8 · p. 8
VII. CONCLUSION
- The conclusion states that UBA’s regret bound does not depend on the number of transmit–receive beam combinations.The paper presents this as relevant to MIMO settings, where the number of combinations grows quickly.Source: §VII · p. 8
APPENDIX A PROOF OF THEOREM 2
The proof separates rounds according to whether the optimal arm is the leader, then bounds regret contributions using neighborhood-restricted selection and concentration inequalities. It concludes that the component terms are bounded and that the theorem follows.
APPENDIX A PROOF OF THEOREM 2
- The proof partitions rounds into those where the optimal arm is leader and those where another arm leads.It then analyzes regret contributions under these two cases.Source: Appendix A · p. 8
- When the optimal arm leads, the algorithm behaves like UCB restricted to that arm’s neighborhood, yielding the cited upper-bound approach.The proof also bounds how often a non-optimal arm is leader in the neighborhood-restricted problem.Source: Appendix A · pp. 8–9
- The proof uses concentration inequalities and bounded play counts and variance to show regret components are bounded by constants.It combines the partial bounds to conclude the theorem.Source: Appendix A · p. 9