RFWeave: Learning Wireless Propagation Across Scenes and Transceiver Configurations
Organizations: Southeast University · The Hong Kong University of Science and Technology
Abstract
Wireless propagation modeling is foundational to the evaluation and design of modern communication systems. To achieve accurate wireless propagation prediction, recent learning-based approaches leverage scene geometry and materials, demonstrating promising generalization across transceiver locations and unseen scenes. However, these approaches remain largely confined to a fixed configuration of array size, receiver orientation, and transmit beam. In this work, we introduce RFWeave, a learned wireless propagation model designed to generalize across scenes, transceiver locations, and transceiver configurations. To accommodate varying array structures, RFWeave incorporates variable-length transceiver tokens that represent antenna elements individually. Furthermore, to capture the geometry between transmit and receive antenna elements, RFWeave aggregates their pairwise spatial relations into propagation tokens. A configuration-conditioned decoder predicts the spatial spectrum and received power. RFWeave is trained on a large-scale simulated dataset spanning 10,000 indoor scenes and more than 5 million channel records under diverse transceiver configurations. Extensive evaluations show that RFWeave achieves state-of-the-art performance across all evaluated generalization settings, improving spatial spectrum structural similarity (SSIM) by up to 15.7% and reducing received power mean absolute error (MAE) by up to 35.4% over the strongest baselines on unseen scenes. Additional experiments on measured DICHASUS data and in beam management further demonstrate RFWeave's broader applicability to real-world channels and downstream wireless tasks.
Figures & tables
| Method | Per-scene Opt. | WiPTR | Wi3R | ||||||||||
| Held-out ID Scenes | OOD Scenes | Held-out ID Scenes | OOD Scenes | ||||||||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | ||
| + Proposed Decoder | |||||||||||||
| PointNet++ | None | 0.7988 | 3.16 | 4.75 | 0.7907 | 3.26 | 4.51 | 0.7414 | 4.00 | 8.61 | 0.7073 | 4.43 | 10.91 |
| PointNeXt | None | 0.7993 | 3.16 | 4.34 | 0.7924 | 3.25 | 4.37 | 0.7421 | 3.96 | 8.60 | 0.7092 | 4.38 | 10.87 |
| Wi-GATr | None | 0.7996 | 3.13 | 3.92 | 0.7963 | 3.14 | 4.02 | 0.7371 | 4.03 | 8.62 | 0.5821 | 5.77 | 11.46 |
| Method | Array Size | Receiver Orientation | Transmit Beam | ||||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | |
| Seen configurations, unseen scenes | |||||||||
| PointNet++ | 0.7583 | 3.63 | 4.84 | 0.7848 | 3.35 | 4.35 | 0.7521 | 3.71 | 4.36 |
| PointNeXt | 0.7563 | 3.65 | 4.52 | 0.7909 | 3.28 | 4.09 | 0.7520 | 3.69 | 4.34 |
| Wi-GATr | 0.7633 | 3.56 | 3.93 | 0.7843 | 3.34 | 3.58 | 0.7526 | 3.68 | 3.60 |
| Transformer | 0.7411 | 3.84 | 4.71 | 0.7690 | 3.51 | 4.34 | 0.7297 | 3.93 | 4.48 |
| Method | Seen Configurations | Angular Interpolation | Array Size Extrapolation | ||||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | |
| Unseen scenes | |||||||||
| PointNet++ | 0.6952 | 4.41 | 4.74 | 0.6540 | 4.89 | 6.24 | 0.5244 | 6.84 | 5.33 |
| PointNeXt | 0.6968 | 4.40 | 4.55 | 0.6585 | 4.87 | 6.22 | 0.5267 | 6.87 | 5.27 |
| Wi-GATr | 0.6866 | 4.48 | 3.53 | 0.6519 | 4.93 | 5.75 | 0.5396 | 6.75 | 4.93 |
| Transformer | 0.6712 | 4.69 | 4.58 | 0.6444 | 5.01 | 6.01 | 0.5250 | 6.98 | 5.48 |
| Variant | SSIM | MAE Spec | MAE Pwr | |||
| Value | (%) | Value | (%) | Value | (%) | |
| RFWeave-B (full) | 0.8060 | – | 3.12 | – | 2.28 | – |
| w/o material properties | 0.7932 | 3.31 | 2.43 | |||
| w/o pairwise geometry | 0.7756 | 3.48 | 2.67 | |||
| w/o configuration conditioning | 0.7618 | 3.69 | 2.86 | |||
| Direct covariance supervision | 0.7827 | 3.41 | 2.59 | |||
| Training samples | RFWeave-B (FT) | NeRF 2 | WRF-GS | ||||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | |
| 100 | 0.7769 | 2.43 | 1.46 | 0.6689 | 2.91 | 2.64 | 0.6428 | 3.76 | 3.03 |
| 500 | 0.7820 | 2.28 | 1.38 | 0.7407 | 2.34 | 1.54 | 0.6433 | 3.53 | 2.76 |
| 1000 | 0.7836 | 2.28 | 1.35 | 0.7443 | 2.34 | 1.47 | 0.6662 | 3.41 | 2.53 |
| 5000 | 0.7921 | 2.27 | 1.31 | 0.7471 | 2.31 | 1.26 | 0.6710 | 3.20 | 2.46 |
| Strategy | Top-1 | Top-2 | Top-3 |
| Exhaustive sweep (GT) | 14.64 / 14.86 | – | – |
| Location-based steering | 7.24 / 7.31 | – | – |
| RFWeave-B (FT) | 9.74 / 10.30 | 10.33 / 11.26 | 10.78 / 11.61 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Data type | Input parameterization | Tokenization | Type vector |
| Scene | , , | 1 token per final sampled center | |
| TX/RX elements | , | 1 token per antenna element | |
| Array centers | , | 1 token per array | |
| TX–RX link | 1 token per link |
| Model | Hidden dim. | Heads | Prop. stages | Backbone (M) | Decoder (M) | Total (M) |
| RFWeave-T | 128 | 8 | 6 | 3.62 | 1.27 | 4.89 |
| RFWeave-S | 192 | 8 | 6 | 7.38 | 2.70 | 10.08 |
| RFWeave-B | 256 | 8 | 8 | 16.63 | 5.04 | 21.67 |
| Regime | Array size | Receiver orientation | Transmit beam |
| Training | 6 seen sizes ( – ) | 8 seen orientations | 16 seen directions |
| Single-factor: array size | 4 unseen sizes ( – ) | fixed | fixed |
| Single-factor: receiver orientation | fixed | 16 unseen orientations | fixed |
| Single-factor: transmit beam | fixed | fixed | 9 unseen directions |
| Joint: seen | 6 seen sizes ( – ) | 8 seen orientations | 16 seen directions |
| Joint: angular interpolation | 6 seen sizes ( – ) | 8 seen orientations | 9 unseen directions |
| Steps | WiPTR Held-out ID Scenes | Wi3R OOD Scenes | ||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | |
| 500 | 0.9398 | 1.498 | 0.962 | 0.8858 | 2.069 | 2.782 |
| 1000 | 0.9372 | 1.576 | 0.970 | 0.8828 | 2.113 | 2.115 |
| 2000 | 0.9407 | 1.556 | 1.045 | 0.8603 | 2.438 | 3.010 |
| Method | Params. (M) | GFLOPs | GFLOPs |
| PointNet++ | 5.19 | 21.05 | 67.49 |
| PointNeXt | 8.52 | 28.53 | 74.97 |
| Wi-GATr | 6.47 | 37.73 | 84.16 |
| Transformer | 8.22 | 21.50 | 67.93 |
| RFWeave-T | 4.89 | 10.36 | 28.99 |
| RFWeave-S | 10.08 | 16.80 | 47.90 |
| Unseen RX Orientation | Unseen TX Beam | ||||||
| SSIM | MAE Spec | MAE Pwr | SSIM | MAE Spec | MAE Pwr | ||
| 0 | 0 | 0.7128 | 3.809 | 2.692 | 0.7205 | 4.051 | 6.789 |
| 1 | 0 | 0.7485 | 3.525 | 2.771 | 0.7260 | 3.965 | 6.622 |
| 2 | 0 | 0.7471 | 3.513 | 2.785 | 0.7144 | 4.094 | 6.911 |
| 0 | 1 | 0.7138 | 3.876 | 3.052 | 0.7472 | 4.045 | 7.197 |
| 0 | 2 | 0.7171 | 3.812 | 2.879 | 0.7349 | 3.955 | 6.845 |
Radio Propagation Modeling
Radio Propagation Modeling predicts how electromagnetic signals travel through environments for wireless planning, channel estimation, and sensing.
See all 61 papers