RadioDiff-FS: Physics-Informed Manifold Alignment in Few-Shot Diffusion Models for High-Fidelity Radio Map Construction
Organizations: State Key Laboratory of ISN and School of Telecommunications Engineering, Xidian University, Xi’an 710071, China · School of Physics, Xidian University, Xi’an 710071, China
Abstract
Radio maps (RMs) provide spatially continuous propagation characterizations essential for 6G network planning, but high-fidelity RM construction remains challenging. Rigorous electromagnetic solvers incur prohibitive computational latency, while data-driven models demand massive labeled datasets and generalize poorly from simplified simulations to complex multipath environments. This paper proposes RadioDiff-FS, a few-shot diffusion framework that adapts a pretrained main-path generator to multipath-rich target domains with only a small number of high-fidelity samples. The adaptation is grounded in a theoretical decomposition of the multipath RM into a dominant main-path component and a directionally sparse residual. This decomposition shows that the cross-domain shift corresponds to a bounded and geometrically structured feature translation rather than an arbitrary distribution change. A direction-consistency loss (DCL) is then introduced to constrain diffusion score updates along physically plausible propagation directions, thereby suppressing phase-inconsistent artifacts that arise in the low-data regime. Experiments show that RadioDiff-FS reduces NMSE by 59.5% on static RMs and by 74.0% on dynamic RMs relative to the vanilla diffusion baseline, achieving an SSIM of 0.9752 and a PSNR of 36.37 dB under severely limited supervision. Even in a one-shot setting with a single target-domain sample per scene, RadioDiff-FS outperforms all fully supervised baselines, confirming that the directional constraint provides an effective inductive bias under extreme data scarcity. Code is available at https://github.com/UNIC-Lab/RadioDiff-FS.
Figures & tables
| Methods | RadioDiff-FS | RadioDiff-FS (one-shot) | PhyRMDM | RadioDiff | RME-GAN | RadioUNet | Rate vs. w/o Fine-Tuning (%) |
| NMSE | 0.0049 | 0.0204 | 0.0100 | 0.0121 | 0.0155 | 0.0159 | 59.50% |
| RMSE | 0.0196 | 0.0278 | 0.0278 | 0.0309 | 0.0340 | 0.0344 | 36.57% |
| SSIM | 0.9752 | 0.9285 | 0.9180 | 0.9268 | 0.9123 | 0.9102 | 5.22% |
| PSNR | 36.37 | 0.9285 | 31.40 | 30.44 | 29.74 | 29.64 | 19.48% |
| Methods | RadioDiff-FS | RadioDiff-FS (one-shot) | PhyRMDM | RadioDiff | RME-GAN | RadioUNet | Rate vs. w/o Fine-Tuning (%) |
| NMSE | 0.0121 | 0.0214 | 0.1113 | 0.0465 | 0.0183 | 0.0381 | 73.98% |
| RMSE | 0.0214 | 0.0296 | 0.2178 | 0.1319 | 0.0358 | 0.0307 | 83.78% |
| SSIM | 0.9510 | 0.8969 | 0.7182 | 0.7968 | 0.8823 | 0.8487 | 19.35% |
| PSNR | 32.68 | 30.61 | 25.23 | 27.86 | 30.74 | 29.57 | 17.30% |
| Configuration | NMSE | RMSE | PSNR (dB) | SSIM | |
| Full FT | w/o DCL | 0.0053 | 0.0204 | 35.79 | 0.9715 |
| w/ DCL | 0.0049 | 0.0196 | 36.37 | 0.9752 | |
| LoRA | w/o DCL | 0.0069 | 0.0233 | 31.88 | 0.9550 |
| w/ DCL | 0.0064 | 0.0225 | 32.65 | 0.9582 | |
| Configuration | NMSE | RMSE | PSNR (dB) | SSIM | |
| Full FT | w/o DCL | 0.0163 | 0.0264 | 31.79 | 0.9105 |
| w/ DCL | 0.0121 | 0.0214 | 32.68 | 0.9510 | |
| LoRA | w/o DCL | 0.0259 | 0.0302 | 28.88 | 0.8750 |
| w/ DCL | 0.0195 | 0.0270 | 30.65 | 0.8982 | |