Diffusion-Based Generative Priors for Efficient Beam Alignment in Directional Networks

arXiv:2604.09653v1 Announce Type: cross Abstract: Beam alignment is a key challenge in directional mmWave and THz systems, where narrow beams require accurate yet low-overhead training. Existing learning-based approaches typically predict a single beam and do not quantify uncertainty, limiting adaptive beam sweeping. We recast beam alignment as a generative task and propose a conditional diffusion model that learns a probabilistic beam prior from compact geometric and multipath features. The learned priors guide top-k sweeps and capture the SNR loss induced by limited probing. Using a ray-traced DeepMIMO scenario with an 8-beam DFT codebook, our best conditional diffusion model achieves strong ranking performance (Hit@1 approx 0.61, Hit@3 approx 0.90, Hit@5 approx 0.97) while preserving SNR at small sweep budgets. Compared with a deterministic classifier baseline, diffusion improves Hit@1 by about 180%. Results further highlight the importance of informative conditioning and the ability of diffusion sampling to flexibly trade accuracy for computational efficiency. The proposed diffusion framework achieves substantial improvements in small-k Hit rates, translating into reduced beam training overhead and enabling low-latency, energy-efficient beam alignment for mmWave and THz systems while preserving received SNR.

Related

• SOAR: Self-Correction for Optimal Alignment and Refinement in Diffusion Models
• Re-Mask and Redirect: Exploiting Denoising Irreversibility in Diffusion Language Models
• Closed-Form Concept Erasure via Double Projections
• ART-VITON: Measurement-Guided Latent Diffusion for Artifact-Free Virtual Try-On
• StableSketcher: Enhancing Diffusion Model for Pixel-based Sketch Generation via Visual Question Answering Feedback
• ECHO: Efficient Chest X-ray Report Generation with One-step Block Diffusion

Source: arXiv cs.AI | 2026-04-14