UniRM: A Universal Large Model for Multiband 3D Radio Map Construction

Radio maps play a crucial role in optimizing wireless network performance and configuration, providing insights into the spatial distribution of radio frequency signal power. Existing solutions often face challenges in generalizing and adapting across various environments, frequency bands, and vertical dimensions. To overcome these limitations, we propose UniRM, a universal large model designed for constructing multiband 3D radio maps. UniRM leverages large-scale pre-training and prompt learning techniques to accurately generate radio maps across diverse environments, altitudes, and frequency bands. Specifically, UniRM employs a UNet-based encoder-decoder architecture during pre-training to extract universal latent representations that capture shared features across different environmental conditions. A prompt learning module further enhances this by transforming auxiliary inputs, such as environmental descriptions, frequency bands, and altitudes, into discriminative embeddings, thereby enabling effective cross-domain generalization and ensuring robustness in unseen scenarios. Extensive experiments using a large, diverse dataset covering numerous scenarios demonstrate that UniRM outperforms state-of-the-art baselines by over 10% in key metrics, including mean squared error, normalized mean squared error, root mean squared error, and peak signal-to-noise ratio. Notably, zero-shot evaluations highlight UniRM’s strong ability to generalize to new environments without retraining. The code for UniRM is available at: https://github.com/Shirleyue/UniRM.