---
title: 'EB-P2WNet: Env-Aided CSI Prediction'
url: https://www.emergentmind.com/topics/eb-based-partial-to-whole-csi-prediction-network-eb-p2wnet
type: topic
---

# EB-P2WNet: Env-Aided CSI Prediction

Searching arXiv for the specified papers to ground the article in the current literature.
arXiv search query: 2508.05142
EB-Based Partial-to-Whole CSI Prediction Network (EB-P2WNet) is a neural architecture for digital-twin-channel-aided channel state information (CSI) prediction that fuses an environment-specific channel subspace basis with real-time partial CSI to reconstruct the entire spatial-frequency domain channel at the present time and to predict CSI for the next channel coherent time. It is introduced within an environment-specific channel subspace basis (EB)-aided partial-to-whole channel state information prediction method (EB-P2WCP) for low-overhead and robust channel prediction in complex sixth-generation mobile communication scenarios, where the static electromagnetic environment is encoded through a digital twin and incorporated as prior information for CSI inference [2508.05142].

## 1. Conceptual scope and problem formulation

EB-P2WNet addresses the overhead of CSI acquisition in massive MIMO-OFDM settings by combining two information sources: a sparse pilot-based estimate of the instantaneous channel and an environment-based prior extracted from a digital twin map. In the formulation reported for the method, the true full OFDM MIMO channel is denoted by $\mathbf H\in\mathbb C^{M_t\times N_{sc}}$, while only $N_p\ll N_{sc}$ subcarriers per antenna are sounded in order to save pilots. The partial CSI estimate is represented as
$$
\mathbf H^0=\mathbf B\odot \mathbf H,
$$
where $\mathbf B\in\{0,1\}^{M_t\times N_{sc}}$ is a binary mask with each row containing exactly $N_p$ ones, and the vectorized form is
$$
\mathbf h^0_{\mathrm{flat}}=\mathrm{vec}(\mathbf H^0)\in\mathbb C^{N_H},
$$
with $N_H=M_t\times N_{sc}$ [2508.05142].

The corresponding prediction tasks are defined as a present-CSI mapping
$$
M:\;(\mathbf H^0_t,\mathbf K_t)\longmapsto \widehat{\mathbf H}_t
$$
and a future-CSI mapping
$$
G:\;(\mathbf H^0_{t-T+1},\dots,\mathbf H^0_t,\mathbf K_t)\longmapsto \widehat{\mathbf H}_{t+1}.
$$
Here $\mathbf K_t$ denotes the environment-specific basis associated with the user’s location in the digital twin. The method therefore belongs to a class of partial-to-whole CSI predictors in which subspace priors and sparse pilot observations are jointly exploited rather than treating CSI recovery as an exclusively data-driven interpolation problem [2508.05142].

A central design premise is that the static properties of the electromagnetic environment can be represented compactly and reused across prediction instances. This differs from approaches that rely only on instantaneous second-order statistics or only on partial CSI. A plausible implication is that EB-P2WNet is intended to shift a portion of CSI acquisition burden from online pilot transmission to offline environmental modeling.

## 2. Environment-based subspace basis extraction

The environment-specific basis, abbreviated EB, is extracted from a high-precision 3D reconstruction of the static environment in the digital twin, including buildings and roads. The coverage area is partitioned into uniform $5\,\mathrm m\times 5\,\mathrm m$ grids. In each grid, four corner locations $i=1,2,3,4$ are selected, and at each vertex $t_m=10$ Doppler-shifted channel snapshots $\mathbf h_{\mathrm{flat}}^{\,i,j}\in\mathbb C^{N_H}$ are simulated. From these samples, the covariance matrix is formed as
$$
\mathbf R=\sum_{i=1}^4\sum_{j=1}^{t_m}\mathbf h_{\mathrm{flat}}^{\,i,j}\bigl(\mathbf h_{\mathrm{flat}}^{\,i,j}\bigr)^H\in\mathbb C^{N_H\times N_H}.
$$
Singular-value decomposition is then applied:
$$
\mathbf R=\mathbf U\,\mathbf S\,\mathbf V^H,\qquad
\mathbf U=[\mathbf u_1,\mathbf u_2,\dots,\mathbf u_{N_H}]\in\mathbb C^{N_H\times N_H},
$$
where the singular values satisfy $\lambda_1\ge\lambda_2\ge\cdots\ge\lambda_{N_H}$ [2508.05142].

The basis dimension $N_b$ is chosen as the smallest value such that the cumulative energy ratio satisfies
$$
\eta_{N_b}=
\frac{\sum_{i=1}^{N_b}\lambda_i}{\sum_{i=1}^{N_H}\lambda_i}\ge 0.95.
$$
In the reported simulations, $N_b=15$ captures over 95% of the energy, and the environment-specific basis for a grid is
$$
\mathbf K=[\mathbf u_1,\dots,\mathbf u_{N_b}]\in\mathbb C^{N_H\times N_b}.
$$
One such $\mathbf K$ is stored per grid in the digital twin; at runtime, the user’s GPS locates the corresponding grid and the matching basis is fetched [2508.05142].

This construction makes EB a location-indexed subspace prior rather than a generic statistical prior. Because the basis is extracted from the static environment, it is designed to characterize environmental structure that persists across users and time instants within the same grid. A common misunderstanding is to treat EB as a direct estimate of the instantaneous channel; in the reported method it instead serves as prior information that is fused with partial CSI during reconstruction.

## 3. Network architecture and fusion pathway

EB-P2WNet is composed of three main blocks: an EB Feature Extraction Subnetwork, an Initial Reconstruction Subnetwork, and a Dual-Input CSI Reconstruction Subnetwork; for future CSI prediction, an LSTM is appended on top of the Initial Reconstruction block [2508.05142].

The EB Feature Extraction Subnetwork takes $\mathbf K\in\mathbb C^{N_H\times N_b}$, reshapes it to $(M_t,N_{sc},N_b)$, and processes it through several 2D convolutions, batch normalization, LeakyReLU, and max-pooling operations to produce smaller feature maps. It then applies 1–2 Transformer encoder layers to capture long-range dependencies among the $N_b$ channels and upsamples through transposed-convolution layers with batch normalization and LeakyReLU back to $(M_t,N_{sc},1)$. The output is the learned feature map $\mathbf F_b\in\mathbb C^{M_t\times N_{sc}}$ [2508.05142].

The Initial Reconstruction Subnetwork receives the sparse CSI $\mathbf H^0$ and unrolls two iterations of a learned proximal-gradient method:
$$
\mathbf H_{\mathrm{ini}}^{(l+1)}
=\mathcal P_\theta^{(l)}\Bigl(
\mathbf H_{\mathrm{ini}}^{(l)}
-\alpha_\theta^{(l)}\bigl(\mathbf B\odot\mathbf H_{\mathrm{ini}}^{(l)}-\mathbf H^0\bigr)
\Bigr),
$$
where each $\mathcal P_\theta^{(l)}$ is a two-layer CNN with LeakyReLU, and $\alpha_\theta^{(l)}$ is a one-layer CNN producing a scalar step-size map. The initialization is $\mathbf H_{\mathrm{ini}}^{(1)}=\mathbf H^0$, and after two steps the network obtains $\mathbf H_{\mathrm{ini}}\in\mathbb C^{M_t\times N_{sc}}$ [2508.05142].

The Dual-Input CSI Reconstruction Subnetwork takes the concatenation
$$
[\mathbf H_{\mathrm{ini}},\mathbf F_b]\in\mathbb C^{M_t\times N_{sc}\times 2}
$$
and processes it with residual CNN blocks of the form conv $\rightarrow$ BN $\rightarrow$ LeakyReLU $\rightarrow$ conv $\rightarrow$ BN with a skip connection. The output is $\widehat{\mathbf H}_t\in\mathbb C^{M_t\times N_{sc}}$. For future CSI prediction, the network uses the past $T$ initial reconstructions $\{\mathbf H_{\mathrm{ini},t-T+1},\dots,\mathbf H_{\mathrm{ini},t}\}$ as input to an LSTM, which predicts $\mathbf H_{\mathrm{ini},t+1}$; this is fused with the fixed $\mathbf F_b$ in the same dual-input reconstruction block to produce $\widehat{\mathbf H}_{t+1}$ [2508.05142].

The nonlinear activation throughout the architecture is LeakyReLU with negative slope $0.0005$. This layered design separates environmental feature extraction, data-consistent reconstruction from sparse pilots, and final feature fusion. This suggests that the model is explicitly structured to prevent the environmental prior from overwhelming the measurement-driven component.

## 4. Objectives, training protocol, and simulation setup

Training is organized in two stages. Stage 1, for present CSI only, uses the mean-squared error loss
$$
\mathcal L_{\mathrm{MSE}}
=\frac{1}{M_t\,N_{sc}}\bigl\|\mathbf H-\widehat{\mathbf H}\bigr\|_F^2.
$$
Stage 2, for joint present and future CSI fine-tuning, uses
$$
\mathcal L
=\lambda_1\,\mathrm{MSE}(\mathbf H_t,\widehat{\mathbf H}_t)
+\lambda_2\,\mathrm{MSE}(\mathbf H_{t+1},\widehat{\mathbf H}_{t+1}),
$$
with $\lambda_1=0.7$ and $\lambda_2=0.3$ [2508.05142].

The data are generated by ray-tracing in Wireless InSite over a $240\,\mathrm m\times 240\,\mathrm m$ urban macro scenario containing 22 buildings, with 6th-order reflections and 1st-order diffractions, at center frequency $6.5\,\mathrm{GHz}$, bandwidth $25\,\mathrm{MHz}$, $N_{sc}=208$, and $M_t=16\times 8$. The coverage is divided into 1,000 grids of $5\,\mathrm m$ side, with 25 random user points per grid and four vertices times ten Doppler instants per grid used for basis extraction. Noise is added at SNR values $\{10\,\mathrm{dB},0\,\mathrm{dB}\}$. For the future prediction task, the model uses $T=4$ past instants with $3\,\mathrm{km/h}$ mobility and a $10\,\mathrm{ms}$ interval [2508.05142].

The reported hyperparameters are batch size 32 and Adam optimization. Learning rates are $6\mathrm e{-4}$ for the EB feature extraction block and $4\mathrm e{-4}$ for the initial and dual-input reconstruction blocks over 200 epochs. For the future LSTM stage, stage 1 uses learning rate $3\mathrm e{-4}$ for 100 epochs, followed by stage 2 fine-tuning with learning rate $5\mathrm e{-5}$ for 50 epochs. These settings define the experimental regime under which the quantitative results are reported [2508.05142].

## 5. Quantitative performance and robustness characteristics

Performance is evaluated under SNR values $\{10\,\mathrm{dB},0\,\mathrm{dB}\}$ and pilot ratios $N_p/N_{sc}=\{1/4,1/8,1/16,1/32\}$. The baselines are LMMSE, which is assumed to know the full second-order statistics; IEB-PR, an ideal noise-free EB projection upper bound; EB-PR, a practical noisy EB projection using
$$
\hat{\mathbf c}=\mathbf U^H\mathbf h^0_{\mathrm{flat}},\qquad
\hat{\mathbf h}=\mathbf U\hat{\mathbf c};
$$
and P2WCP, which uses the same network as EB-P2WNet but without any $\mathbf K$-branch. The reported metrics are NMSE,
$$
\mathrm{NMSE}=\frac{\|\mathbf H-\widehat{\mathbf H}\|_F^2}{\|\mathbf H\|_F^2},
$$
cosine similarity,
$$
\mathrm{CS}=\frac{|\mathbf h^H\widehat{\mathbf h}|}{\|\mathbf h\|\,\|\widehat{\mathbf h}\|},
$$
and average achievable rate under beamforming [2508.05142].

For present CSI at $0\,\mathrm{dB}$ and pilot ratio $1/32$, EB-P2WNet yields $\mathrm{NMSE}\approx 0.0823$ versus $\mathrm{NMSE}=0.1834$ for P2WCP, corresponding to an approximately $4.99\,\mathrm{dB}$ gain. Under the same conditions, the method saves 40–50% of pilots to match NMSE. The abstract states that incorporating EB provides significant benefits under low signal-to-noise ratio and pilot ratio conditions, achieving a reduction of up to 50% in pilot overhead [2508.05142].

In the multi-user interference experiment, two users’ CSI are mixed as
$$
\mathbf H_{x\text{–int}}=\mathbf H_{x\text{–noise}}+a\mathbf H_{y\text{–noise}},\qquad a\in[0.1,0.9].
$$
EB-P2WNet remains within $0.5\,\mathrm{dB}$ NMSE of interference-free performance for $a\le 0.5$ and outperforms all baselines by $1.9$–$2.7\,\mathrm{dB}$ at $a=0.9$. In the localization-error study, with random position errors from 1 to $10\,\mathrm m$, a $3\,\mathrm m$ error at $0\,\mathrm{dB}$ and pilot ratio $1/8$ causes only a $+0.56\,\mathrm{dB}$ NMSE increase for EB-P2WNet, while P2WCP is unaffected because it does not use EB. EB-P2WNet tolerates approximately $5\,\mathrm m$ error before the EB priors become neutral [2508.05142].

The method is also tested under coarser EB granularity by merging $5\,\mathrm m$ grids into $10\,\mathrm m$ grids. In that setting, the accuracy drop is less than $0.1\,\mathrm{dB}$, and the localization tolerance extends to $3.5\,\mathrm m$ for a $0.5\,\mathrm{dB}$ NMSE penalty. For future CSI prediction under pilot ratio $1/8$ at $10\,\mathrm{dB}$, EB-P2WNet achieves $\mathrm{NMSE}=0.0232$ and $\mathrm{CS}=0.9895$, compared with $\mathrm{NMSE}=0.0341$ and $\mathrm{CS}=0.9842$ for P2WCP, and completes prediction in approximately $1.3\,\mathrm{ms}$ [2508.05142].

| Setting | EB-P2WNet result | Comparator |
|---|---:|---:|
| Present CSI, $0\,\mathrm{dB}$, pilot $1/32$ | NMSE $\approx 0.0823$ | P2WCP: $0.1834$ |
| Future CSI, $1/8$ pilot, $10\,\mathrm{dB}$ | NMSE $=0.0232$, CS $=0.9895$ | P2WCP: NMSE $=0.0341$, CS $=0.9842$ |
| Localization error, $3\,\mathrm m$ | $+0.56\,\mathrm{dB}$ NMSE increase | P2WCP unaffected |

These results are reported as evidence that the environmental subspace prior remains useful under low SNR, colored interference, and location uncertainty, rather than only in idealized inference conditions.

## 6. Relation to earlier environment-aided CSI prediction

A directly relevant precursor is the environment-aware CSI prediction framework introduced in “Can Wireless Environmental Information Decrease Pilot Overhead: A CSI Prediction Example” [2408.06558]. That work proposes WEI-CSIP, where scatterers around the mobile station are abstracted from environmental information using multiview images, an environmental feature map is extracted by a CNN, a deep probabilistic subsampling network acquires an optimal fixed pilot pattern, and a CNN-based channel prediction network predicts complete CSI from the environmental feature map and partial CSI [2408.06558].

In that earlier formulation, the environmental feature map $\Phi_{\mathrm{env}}$ is extracted from a panoramic image built from $N_A$ RGB camera views,
$$
I=[I_1,I_2,\dots,I_{N_A}]\in\mathbb R^{(w\cdot N_A)\times h\times 3},
$$
using three convolutional layers and one pooling layer. Pilot pattern optimization is performed with a binary mask $A\in\{0,1\}^{N_T\times N_c}$ having exactly $N_p$ ones, learned through a Gumbel-Top-k or Gumbel-Softmax relaxation. The reconstruction stage combines a small proximal iterative block with a fusion CNN so that
$$
\widehat H=f(H_{\mathrm{partial}},\Phi_{\mathrm{env}};\Theta).
$$
The reported test-set result for EB-P2WNet in that framework at pilot ratio $1/8$ is $\mathrm{NMSE}=0.0113$ and cosine similarity $0.9945$, compared with $\mathrm{NMSE}=0.0893$ and cosine similarity $0.9545$ for the random-sampling, no-environment baseline; the reported pilot overhead reduction is from $1/5$ to $1/8$ [2408.06558].

The later digital-twin-channel formulation differs in the form of the environmental prior. Instead of multi-view imagery and a learned environmental feature map tied to scatterer abstraction, it employs a grid-indexed environment-specific channel subspace basis extracted from a digital twin map. This suggests a methodological transition from image-conditioned environmental side information toward explicit subspace priors derived from simulated electromagnetic structure. A common misconception is that these are interchangeable representations; the available descriptions indicate that they use different sensing modalities, different prior parameterizations, and different fusion pathways.

## 7. Significance, interpretation, and limitations of the reported evidence

Within the reported framework, EB-P2WNet is presented as a low-overhead and robust channel prediction scheme for various complex scenarios, combining a small number of environment-extracted subspace bases ($N_b=15$) with sparse pilot measurements through a CNN+Transformer+LSTM architecture [2508.05142]. The core empirical claims are threefold: reduction of required pilot ratio by up to 50%, robustness under low SNR and multi-user interference, and prediction of the next coherence-time CSI within approximately $1.3\,\mathrm{ms}$ [2508.05142].

The evidence also delineates the operating assumptions under which these conclusions hold. The environmental prior is tied to a preconstructed digital twin and to grid-based localization; the method stores one basis per grid and fetches it using GPS at runtime. Consequently, robustness to localization errors is not absolute but quantified: approximately $5\,\mathrm m$ error before EB priors become neutral under the reported settings, with different behavior under coarser EB granularity [2508.05142]. This clarifies that the method does not eliminate dependence on environmental registration; rather, it moderates its sensitivity.

A further interpretive point concerns the role of EB relative to purely learned CSI completion. Because P2WCP is defined as the same network without the $\mathbf K$-branch, the comparison isolates the contribution of the environment-specific basis within the reported architecture. The measured gap between EB-P2WNet and P2WCP at low pilot ratios and low SNR suggests that the principal gain arises from introducing structured environmental priors into the reconstruction process rather than from architectural scaling alone [2508.05142]. A plausible implication is that the framework is most advantageous when pilot scarcity and channel uncertainty are sufficiently severe that the static environmental subspace becomes a useful regularizer.

Source: https://www.emergentmind.com/topics/eb-based-partial-to-whole-csi-prediction-network-eb-p2wnet