# BeamRMX: Radiation-Pattern-Driven Learning for Generalizable Beam Radio Map Prediction and Beam Management

arXiv: https://arxiv.org/abs/2609.00615 · HTML: https://mlsanity.com/paper/2609.00615/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 28 of its 34 references that are in the map.
- First posted on arXiv 1 Sep 2026.

## Builds on (most cited first, 28 of 28)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2006.11239/) (2020)
- [Scalable Diffusion Models with Transformers](https://mlsanity.com/paper/2212.09748/) (2022)
- [Feature Pyramid Networks for Object Detection](https://mlsanity.com/paper/1612.03144/) (2016)
- [ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders](https://mlsanity.com/paper/2301.00808/) (2023)
- [Sionna: An Open-Source Library for Next-Generation Physical Layer Research](https://mlsanity.com/paper/2203.11854/) (2022)
- [RadioUNet: Fast Radio Map Estimation with Convolutional Neural Networks](https://mlsanity.com/paper/1911.09002/) (2019)
- [DeepMIMO: A Generic Deep Learning Dataset for Millimeter Wave and Massive MIMO Applications](https://mlsanity.com/paper/1902.06435/) (2019)
- [A Tutorial on Environment-Aware Communications via Channel Knowledge Map for 6G](https://mlsanity.com/paper/2309.07460/) (2023)
- [RadioDiff: An Effective Generative Diffusion Model for Sampling-Free Dynamic Radio Map Construction](https://mlsanity.com/paper/2408.08593/) (2024)
- [RME-GAN: A Learning Framework for Radio Map Estimation based on Conditional Generative Adversarial Network](https://mlsanity.com/paper/2212.12817/) (2022)
- [Deep Completion Autoencoders for Radio Map Estimation](https://mlsanity.com/paper/2005.05964/) (2020)
- [A Tutorial on Beam Management for 3GPP NR at mmWave Frequencies](https://mlsanity.com/paper/1804.01908/) (2018)
- [A Tutorial on Extremely Large-Scale MIMO for 6G: Fundamentals, Signal Processing, and Applications](https://mlsanity.com/paper/2307.07340/) (2023)
- [RadioDiff-k2: Helmholtz Equation Informed Generative Diffusion Model for Multi-Path Aware Radio Map Construction](https://mlsanity.com/paper/2504.15623/) (2025)
- [RadioGAT: A Joint Model-based and Data-driven Framework for Multi-band Radiomap Reconstruction via Graph Attention Networks](https://mlsanity.com/paper/2403.16397/) (2024)
- [RadioDiff-3D: A 3D×3D Radio Map Dataset and Generative Diffusion Based Benchmark for 6G Environment-Aware Communication](https://mlsanity.com/paper/2507.12166/) (2025)
- [The First Pathloss Radio Map Prediction Challenge](https://mlsanity.com/paper/2310.07658/) (2023)
- [UAV-aided Radio Map Construction Exploiting Environment Semantics](https://mlsanity.com/paper/2107.10574/) (2021)
- [RadioDiff-Inverse: Diffusion Enhanced Bayesian Inverse Estimation for ISAC Radio Map Construction](https://mlsanity.com/paper/2504.14298/) (2025)
- [Radio Map Prediction from Aerial Images and Application to Coverage Optimization](https://mlsanity.com/paper/2410.17264/) (2024)
- [Diffusion Model-based Incentive Mechanism with Prospect Theory for Edge AIGC Services in 6G IoT](https://mlsanity.com/paper/2407.10979/) (2024)
- [BeamCKM: A Framework of Channel Knowledge Map Construction for Multi-Antenna Systems](https://mlsanity.com/paper/2511.18376/) (2025)
- [Energy-modified Leverage Sampling for Radio Map Construction via Matrix Completion](https://mlsanity.com/paper/2404.08286/) (2024)
- [U6G XL-MIMO Radiomap Prediction: Multi-Config Dataset and Beam Map Approach](https://mlsanity.com/paper/2603.06401/) (2026)
- [BeamCKMDiff: Beam-Aware Channel Knowledge Map Construction via Diffusion Transformer](https://mlsanity.com/paper/2601.10207/) (2026)
- [RadioDiff-FS: Physics-Informed Manifold Alignment in Few-Shot Diffusion Models for High-Fidelity Radio Map Construction](https://mlsanity.com/paper/2603.18865/) (2026)
- [RadioPiT: Radio Map Generation with Pixel Transformer Driven by Ultra-Sparse Real-World Data](https://mlsanity.com/paper/2512.01451/) (2025)

## Abstract (arXiv)

The evolution toward sixth-generation (6G) wireless networks is driving larger antenna arrays and highly directional multi-beam transmission, making accurate knowledge of beam-dependent spatial coverage important for beam management and environment-aware network operation. Radio maps (RMs) provide such a representation, yet conventional RM prediction assumes omnidirectional or transmitter-level radiation. In beamformed multiple-input multiple-output (MIMO) systems, one propagation scene instead gives rise to many configuration-dependent beam radio maps (BeamRMs), creating challenges in beam representation and generalization. Existing methods either condition prediction on beam descriptors or use beam maps as auxiliary inputs to generic architectures. We propose BeamRMX, which, to the best of our knowledge, is the first dedicated framework to treat the spatial radiation pattern as the primary BeamRM query and learn how scene geometry transforms it into the received power field. XBase learns multiscale interactions between the radiation query and scene geometry, while an optional Evidence Adapter uses a few cross-configuration BeamRMs from the same scene. Matched-domain and zero-shot experiments show consistent gains over deterministic and diffusion baselines, including mean absolute error reductions of 26.1% on unseen scenes and 47.8% on an unseen configuration. Cross-configuration evidence further improves reconstruction and intra-sector beam refinement.