Radio map construction via generative diffusion models — UNIC Lab, Xidian University

RadioDiff — The foundational diffusion model for radio map construction.   📄 Paper  |  💻 Code  | 

RadioDiff-k² — PINN-enhanced diffusion guided by the Helmholtz equation.   📄 Paper  |  💻 Code  | 

iRadioDiff — Indoor radio map construction with physical information integration.   📄 Paper  |  💻 Code  |   

RadioDiff-Turbo — Efficiency-enhanced RadioDiff for accelerated inference.   📄 Paper  | 

RadioDiff-Flux — Adaptive reconstruction under dynamic environments and base station location changes.   📄 Paper  | 

RadioDiff-3D — 3D radio map construction with the UrbanRadio3D dataset.   📄 Paper  |  💻 Code  | 

RadioDiff-FS — Few-shot learning for radio map construction with limited measurements.   📄 Paper  |  💻 Code  | 

RadioDiff-Inverse — Sparse measurement-based radio map recovery for ISAC applications.   📄 Paper  |  💻 Code  | 

RadioDiff-Loc — Sparse measurement-based NLoS localization using diffusion models.   📄 Paper  | 

📚 For a comprehensive categorized overview of radio map research, visit Awesome-Radio-Map-Categorized.

1. install torch

conda create -n radiodiff python=3.9
conda avtivate radiodiff
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

1. install other packages.

pip install -r requirement.txt

1. prepare accelerate config.

accelerate config # HOW MANY GPUs YOU WANG TO USE.

• The data structure should look like:

|-- $RadioMapSeer
|   |-- gain
|   |-- |-- carsDPM
|   |-- |-- |-- XXX_XX.PNG
|   |-- |-- |-- XXX_XX.PNG
|   ...
|   |-- png
|   |-- |-- buildings_complete
|   |-- |-- |-- XXX_XX.PNG
|   |-- |-- |-- XXX_XX.PNG
|	...

1. train the first stage model (AutoEncoder):

accelerate launch train_vae.py --cfg ./configs/first_radio.yaml

1. train latent diffusion-edge model:

accelerate launch train_cond_ldm.py --cfg ./configs/radio_train.yaml

1. fine-tune the pretrained model on IRT4 few-shot data:

accelerate launch train_cond_ldm_finetune.py --cfg ./configs/radio_train_irt4_finetune.yaml

python sample_cond_ldm.py --cfg ./configs/radio_test_finetune.yaml