A Geometry-aided Message Passing Method for AoA-Based Short Range MIMO Channel Estimation
Abstract
Short range channels commonly arise in millimeter wave (mmWave) wearable settings, where the length of the antenna arrays can be comparable to the distance between the radios. Conventional mmWave MIMO channel estimation techniques based on the far field assumption may perform poorly in short range settings due to the large angular spread and, hence, high available rank. We propose a geometry-aided message passing algorithm that exploits structure in short range line-of-sight (LoS) channels for spatial sub-Nyquist channel estimation. Our approach parametrizes the channel using angle-of-arrivals (AoAs) that are locally defined for subarrays of an antenna array. Furthermore, it leverages the dependencies between the local AoAs using factors based on the array geometry. We show that the LoS MIMO channel can be reconstructed using the derived local AoA estimates and the known transceiver geometry. The proposed approach achieves a reasonable rate with greatly reduced pilot transmissions when compared to exhaustive beam search-based local AoA estimation.
Index Terms:
AoA estimation, Mm-wave, message passingI Introduction
Millimeter wave radio architectures can be different from the architectures that are commonly used at lower carrier frequencies [1]. The smaller wavelengths at millimeter wave (mmWave) allow the use of a large number of antennas. The number of available radio frequency (RF) chains, however, can be far less than the number of antenna elements, to minimize the power consumption and the cost of mmWave systems [1]. The hybrid beamforming architecture, used in the IEEE 802.11ay standard, is one such example. High spectral efficiency can be achieved in a beamforming system if its antenna arrays are configured properly. Such configuration can be achieved if the channel between the transmitter (TX) and the receiver (RX) is known.
Channel estimation can be challenging in hybrid beamforming systems due to a limited number of RF chains [1]. To reduce the overhead in learning the channel, prior work has exploited low rank and sparse nature of mmWave channels [2]. Most of these algorithms, however, make a far field assumption, i.e., the distance between the TX and the RX is larger than the length of the antenna arrays at the TX and the RX [3]. In typical mmWave wearable settings, the far field assumption may not be applicable. For instance, short range line-of-sight (LoS) channels can have rank that is larger than one; a study on the rank of LoS channels with the transceiver distance can be found in [4]. In contrast, far field LoS channels have a rank one structure. Similarly, dictionaries used for compressive channel estimation in which far field channels have a sparse representation may not be appropriate in short range settings. Therefore, there is a need to develop algorithms that exploit the short range channel structure to minimize the training overhead for channel estimation.
In this paper, we propose a geometry-aided message passing algorithm for short range LoS channel estimation. We consider a point-to-point multiple-input multiple-output (MIMO) communication scenario, where the TX node is equipped with a fully digital architecture and a relatively small number of antennas, while the RX node comprises a large antenna array and a subarray-based hybrid beamforming architecture. Such a setup may be practical in an on-body mmWave sensor network. Assuming channel reciprocity, the estimated channel at the RX node can be used to design the corresponding hybrid beamforming structures for both data transmission and reception. We assume that the far field assumption holds for the subarray specific channels but not for the full channel. Under such an assumption, we define local angle-of-arrivals (AoAs) for each RX subarray. Our algorithm estimates the local AoAs using the pilots sent from the TX antennas, while exploiting the dependencies among the local AoAs that arise from the geometry of the RX antenna array. The full MIMO channel is then estimated from local AoA information corresponding to two outermost TX antennas of the TX antenna array. The proposed technique has low complexity and performs significantly better than the maximum likelihood approach that recovers local AoAs independently. Simulation results indicate that our algorithm can be used to greatly reduce the training required for LoS channel estimation when compared to conventional techniques.
II System and channel model
We consider a point-to-point111Extension to multiuser scenario is straightforward if each TX is assigned with orthogonal pilots. MIMO system with a subarray-based hybrid beamforming architecture at the RX illustrated in Fig 1. Let be the number of RF chains and corresponding analog subarrays at the RX. We use to denote the number of antennas at the RX. Each of the subarrays at the RX is considered to be spaced uniform linear array with antennas, where denotes the wavelength corresponding to the mmWave carrier frequency. Furthermore, the antennas at the RX are assumed to be collinear. Each subarray is equipped with an analog beamforming architecture with phase shifters. The TX is equipped with a fully digital architecture with RF chains and antennas.
We consider a narrowband setting and define as the channel matrix. The length of the antenna arrays at the TX and the RX are and , and the distance between the midpoint of the arrays at the TX and the RX is denoted by .

While the spacing between successive elements in each RX subarray is assumed to be , the spacing between the successive subarrays can be arbitrarily larger than . We assume that inter-subarray spacing is uniform across successive RX subarrays, and is determined by and . Similarly, the spacing between the TX antennas can be larger than . The assumptions like the use of uniform subarray spacing, and narrowband setting are made for simplicity of exposition. The ideas underlying our approach, however, can be extended to any TX/RX array geometry.
Now, we describe the channel and the system model in the hybrid beamforming setup. We use to denote the distance between the antenna at the RX and the antenna at the TX. The entry of the channel is then [4]
(1) |
We define , a vector in , as the channel between the subarray at the RX and the RF chain at the TX. Let be the pilot transmitted by the TX in the training slot. In the same slot, the RX applies conjugate transpose of to its subarray to acquire channel measurement . Under perfect synchronization, the channel measurement is
(2) |
where is circularly symmetric Gaussian noise with zero mean and variance . As has entries and the RX can acquire channel measurements in parallel, standard channel estimation based on exhaustive search requires a training overhead of . In this paper, we show that, by utilizing the known geometry of the RX antenna array, a reasonable approximation of can be estimated with a small fraction of pilot transmissions.
III Geometry-aided channel reconstruction
The idea underlying the proposed short range LoS channel estimation approach is best understood using local AoA estimation. For ease of exposition, consider a scenario in which the first antenna at the TX transmits known pilots and the rest of the antennas are inactive. For an indicator function , setting in (2) results in the desired condition. Ignoring the TX antenna index , we define as the local AoA made by the ray between the midpoints of the RX subarray and the first TX antenna222In general, denotes the local AoA between TX antenna and RX subarray . An identical AoA estimation process is carried out for each ., with the normal to the RX array. We make an assumption that the AoAs seen by the individual antennas within subarray are invariant, and can be approximated by the local AoA . The angles , however, can vary with the subarray index , as seen in Fig. 1. In mmWave wearable settings, the far field approximation may be valid for each subarray, while it may not hold true for the full array. For example, the length of a virtual reality headset can be comparable to the distance with a wearable. In this paper, we model as a realization of a random variable , and propose an algorithm to estimate the local AoAs from the channel measurements while exploiting the dependencies among .
III-A Construction of likelihood functions for local AoAs
As each subchannel is assumed to satisfy the far field approximation, the local AoAs can be estimated using standard compressed sensing (CS) [5]. We use shifted Zadoff-Chu (ZC) sequences as analog RX beamformer weights for channel acquisition, as ZC sequences can be realized in beamforming systems with unimodulus constraints, and have good properties for AoA estimation [6]. For instance, CS-based AoA estimation with ZC sequences is efficient, if the AoAs come from a set of angles that are defined by the discrete Fourier transform. We define as the ZC sequence used at the subarray of the RX. Spatially diverse channel measurements are acquired at the RX by applying different random circulant shifts of to its subarray.
For pilot transmission from the first antenna at the TX, the channel measurement at the RX subarray is
(3) |
We define a Vandermonde vector of length as . Under the far field assumption for subchannels, can be approximated as , where is an unknown complex gain. The channel measurement is then
(4) |
A collection of projections of , obtained using RX beam training vectors , is defined as . In this paper, the first beam training vectors, i.e., , are chosen as distinct random circulant shifts of . The vector is defined as , where denotes the element-wise product. The measurement is defined differently so that the unknown gain can be estimated. With defined as the first entry of , it can be observed from (4) that is a noisy version of . An estimate of is then .
The gain compensated channel measurements are defined as . The compression matrix associated with the subarray is defined as , where the row of is . From (4), it can be observed that . We ignore the errors in estimating to conclude that is a realization of . Thus, the scaled likelihood function is defined as
(5) |
A possible way to estimate is by maximizing in (5).
III-B Statistical dependency among local AoAs
In this section, we design geometry factors to model the strong dependencies among the local AoAs . For a specific , it can be observed from Fig. 1 that is a function of the distance between the first antenna at the TX and the first subarray of the RX. As this distance is typically bounded, takes a range of values that depend on . The maximum likelihood (ML) estimator that independently maximizes (5) over does not exploit such dependency, and may result in poor local AoA estimates.
We derive the geometry factor that represents the distribution of conditioned on . We define as the distance between the first TX antenna and the midpoint of the RX array, and as the distance of the RX subarray from the midpoint of the RX. For a given , the distance in Fig. 1 can be solved from
(6) |
and, subsequently, the local AoA can be expressed as
(7) |
From (6) and (7), it can be observed that there is a unique for a given and . Let be the mapping from and to , i.e., . We assume that . For example, it is reasonable to assume and in on-body communication setups. The distance is known to lie in . We assume that is uniformly distributed within this interval to get
(8) |
where and denotes the Dirac-delta function indicating . Similarly, other conditional distributions, i.e., , can be estimated with the arguments used to compute . The geometry factors do not depend on the channel measurements, and can be computed offline based on the array geometry at the RX.
III-C Geometry-aided message passing

Now, we explain our algorithm that combines information about from the likelihood functions (5) and the geometry factors (8). The factor graph [7] corresponding to our algorithm is shown in Fig. 2. The circular nodes in Fig. 2 denote random variables, and the rectangular nodes contain factors or likelihoods. A message flowing out of a node represents a distribution of a random variable, from the node’s perspective. Messages are sent between the circular nodes and the rectangular nodes to obtain better estimates of when compared to the ML estimator. Our algorithm includes a forward pass, i.e., sequential flow of messages among , and a backward pass in which message flows occur in the opposite direction. The messages in both directions are computed using the sum-product algorithm [7].
We explain how messages are constructed in the forward pass. In the first iteration of the forward pass, the likelihood function is sent to node , which forwards to the geometry factor that contains . Using the two functions, the geometry factor forwards message to node , defined by
(9) |
The distribution is essentially marginalized over . In other words, represents the belief about , i.e., a scaled probability distribution of believed by the geometry factor, using information about . Note that provides side information about that is independent of the measurements acquired by the second subarray. This side information comes from the observations in the first subarray, and the statistical dependency between and . The node combines information from the channel measurements, i.e., , with the one from , using
(10) |
The message in (10) is sent to the factor containing , which computes the belief about with and . The process of message flows continues until the last node with is reached. The messages in (9) and (10) can be generalized, by setting to and to , to obtain recursive equations. The forward pass computes message inflows using the recursive equations.
The forward pass does not exploit information about to generate side information about . The backward pass overcomes this issue by performing message flows in the opposite direction of the forward pass. In the first iteration, message from the node, i.e., , flows into the geometry factor containing . The geometry factor then sends a belief about , defined as , based on . Similarly, the backward pass is computed using
(11) | ||||
(12) |
It can be observed that nodes and do not receive side information in the forward and backward passes. Therefore, we set and , where denotes a uniform distribution over .
At the end of forward and backward passes, each node obtains side information in the form of and , and also has access to the likelihood . The three sources of information about can be combined by defining a new distribution , i.e.,
(13) |
Finally, the local AoA estimate with our geometry-aided message passing algorithm is defined as . For computational tractability, the integrals in (8), (9), and (11) are computed using a discrete sum, assuming angular resolution of . Thus, the complexity of the algorithm is . It can be noticed that the factor graph in our algorithm models dependencies among adjacent local AoAs, and does not include factors like for . Incorporating such factors can result in short cycles, which may not be desirable from a message passing perspective [8].
III-D MIMO channel reconstruction from local AoA estimates
A reasonable approximation of the LoS MIMO channel matrix can be estimated at the RX from the local AoA estimates. First, we note that the RX knows the geometry of its antenna array, and can compute the position vectors of all the RX antenna elements relative to a common reference. Let and denote the position vectors of the first and RX subarrays. We define as a unit vector along the antenna array at the RX. It can be observed from Fig. 1 that the first TX antenna is at a distance of from the first RX subarray. Furthermore, the direction of this TX antenna, relative to the RX array, can be found by rotating in the anti-clockwise direction by . For a rotation matrix given by
(14) |
the position vector of the first TX antenna is defined as where can be computed using triangulation from multiple local AoA estimates. The position vector of the TX antennas can be derived by the algorithm in Section III-C to estimate the local AoAs corresponding to the TX. Finally, the coordinates of the remaining TX antennas are acquired using the estimated positions of the first and last TX antennas and the known TX geometry. The LoS channel is constructed from the estimated coordinates using (1). Note that the pilot transmissions can be performed from different antennas at the TX using spread sequences.
IV Simulation Results
We consider antennas per subarray, and subarrays for the hybrid beamforming architecture at the RX. The TX is equipped with a fully digital architecture with antennas and RF chains. The carrier frequency in the system is set to , which corresponds to . The length of the arrays at the TX and the RX are and . It may be possible to mount the RX array on a augmented reality headset, and the TX array on a wearable like smart watch. The spacing between the midpoint of successive RX subarrays is , and the spacing between neighbouring TX antennas is . The TX and RX arrays are placed on a horizontal plane at a height of , in a room of dimensions . The properties of the ceiling, side-walls, and the floor were set according to the parameters in [4]. The channel in our simulations has components corresponding to reflections from the walls in addition to the LoS component. Our algorithm, however, can only estimate the dominant signal path corresponding to the LoS component. The distance between the transceivers, i.e., , is chosen uniformly at random from for and . The orientations of the TX and RX arrays are chosen at random to cover a wide range of possible configurations. The RX SNR for pilot measurements, , is chosen to be fairly small to demonstrate the resilience of the algorithm against estimation inaccuracies.

We evaluate three different algorithms in a short range setting. The first method uses the ZC-based compression matrix defined in Sec. III-A, and independently maximizes the likelihood functions in (5). The second approach estimates local AoAs using the proposed geometry-aided message passing algorithm. Our algorithm uses information from the likelihoods, and the geometry factors that are based on the RX array configuration. For a benchmark, we define the third approach based on exhaustive beam search, i.e., the DFT dictionary is used for , and channel measurements are acquired in each subarray. The resolution in the angle () space is set to , using . An example of the likelihood functions is shown in Fig. 3. It can be observed that maximizing the likelihood functions independently for can result in local AoAs that are significantly different from each other. The proposed method, however, is able to “repair” the mismatched AoAs using geometry factors and message passing. In Fig. 4, we plot the empirical cumulative distribution function (ECDF) of the errors in the local AoA estimates with the maximum likelihood approach and the proposed method. It can be observed from Fig. 4 that the local AoAs recovered by our algorithm are significantly closer to those obtained with the benchmark, when compared to the maximum likelihood approach.


The accuracy of the reconstructed MIMO channel is studied using the achievable rate obtained with the estimated channel as a figure of merit. The rate analysis with practical hybrid beamforming structures is left for future study. Let denote the singular value decomposition (SVD) of the estimated channel. Singular matrices and are considered to configure the precoders and the combiners at the TX and the RX. The achievable rate of the system (assuming ideal all-digital RX) is determined from the capacity of MIMO channel . The simulation results are then averaged over random orientations. It can be observed from Fig. 5 that channel estimation using the proposed algorithm results in a reasonable achievable rate that is comparable to the one achieved with exhaustive scan. While the exhaustive scan requires channel measurements per subarray, our algorithm achieves good performance with just measurements per subarray. The performance gap to the perfect CSI case is due to fairly low dB SNR available for estimation of both local AoA and in (4), as well as, unaccounted non-LoS components in the actual channel model.
V Conclusions
Short range channels exhibit different structure when compared to the commonly studied far field channels. For example, the angles-of-arrival in a short range setting can vary across multiple sections of the receive antenna array. In this paper, we have shown that the angles can be statistically dependent, with the dependencies determined by the receive array geometry. We have developed a message passing algorithm that exploits the statistical dependencies between the angles, for channel estimation with sub-Nyquist measurements in the angular domain. Our results indicate that geometry information can serve as strong regularizer for the channel estimation problem.
Acknowledgements
This research is supported by the Academy of Finland under grant numbers 311741 and 318927 (6Genesis Flagship), and by the U.S. National Science Foundation under grant numbers CNS-1702800 and ECCS-1711702.
References
- [1] R. W. Heath, N. Gonzalez-Prelcic, S. Rangan, W. Roh, and A. M. Sayeed, “An overview of signal processing techniques for millimeter wave MIMO systems,” IEEE J. Sel. Topics Signal Process., vol. 10, no. 3, pp. 436–453, 2016.
- [2] J. Rodríguez-Fernández, N. González-Prelcic, K. Venugopal, and R. W. Heath, “Frequency-domain compressive channel estimation for frequency-selective hybrid millimeter wave MIMO systems,” IEEE Trans. on Wireless Commun., vol. 17, no. 5, pp. 2946–2960, 2018.
- [3] D. Tse and P. Viswanath, Fundamentals of wireless communication. Cambridge university press, 2005.
- [4] E. Torkildson, U. Madhow, and M. Rodwell, “Indoor millimeter wave MIMO: Feasibility and performance,” IEEE Trans. on Wireless Commun., vol. 10, no. 12, pp. 4150–4160, 2011.
- [5] E. J. Candès and M. B. Wakin, “An introduction to compressive sampling,” IEEE Signal Process. Mag., vol. 25, no. 2, pp. 21–30, 2008.
- [6] Y. Ding, S.-E. Chiu, and B. D. Rao, “Bayesian channel estimation algorithms for massive MIMO systems with hybrid analog-digital processing and low-resolution ADCs,” IEEE J. of Sel. Topics in Signal Process., vol. 12, no. 3, pp. 499–513, 2018.
- [7] F. R. Kschischang, B. J. Frey, H.-A. Loeliger et al., “Factor graphs and the sum-product algorithm,” IEEE Trans. on Inform. theory, vol. 47, no. 2, pp. 498–519, 2001.
- [8] M. Karimi and A. H. Banihashemi, “Message-passing algorithms for counting short cycles in a graph,” IEEE Trans. on Commun., vol. 61, no. 2, pp. 485–495, 2013.