ST-CGMDiff: Sparse-Conditioned Cross-Frequency Spatio-Temporal Latent Diffusion for Dynamic 3D Channel Gain Map Reconstruction

Abstract

Accurate and efficient reconstruction of channel gain maps (CGMs) is essential for environment-aware wireless communications. Unlike prior works that focus on static CGMs at a single frequency band, this paper studies the construction of dynamic three-dimensional (3D) CGMs across multiple frequency bands over consecutive time slots, using limited channel measurements. Specifically, for a given transmitter (Tx) location, we represent the CGM at any given receiver (Rx) height as a two-dimensional (2D) grayscale image, where each pixel indicates the average channel gain over the corresponding Rx spatial grid cell. Accordingly, we propose ST-CGMDiff, a spatio-temporal latent diffusion framework that integrates sparse-observation conditioning and cross-frequency modeling to reconstruct dynamic 3D CGMs across multiple carrier frequencies and varying Rx heights within a unified model. In this framework, we jointly encode the input parameters, including the current sparse observations, observation masks, physical Tx and Rx z-coordinates, and carrier frequencies as spatial conditions. A frequency-conditioned convolutional gated recurrent unit (ConvGRU) then extracts causal information from previously constructed CGMs, while a spatially adaptive fusion mechanism balances historical information and current observations. Furthermore, bidirectional convolutional long short-term memory (BiConvLSTM) modules are introduced to capture the ordered dependencies among the selected Rx z-layers. Finally, we perform few-step latent diffusion followed by high-resolution residual refinement to generate the final CGM with improved spatial fidelity.