Stochastic Neural Ray Tracing for Radio Frequency Channel Modeling
Yinyan Bu, Jiajie Yu, Xingyu Chen, Bo Wen, Xinyu Zhang, Piya Pal
Abstract
Wireless channel modeling is essential for the design, analysis, and optimization of modern wireless sensing and communication systems. However, accurately modeling wireless channels in electrically large and complex environments remains a long-standing challenge, owing to the intricate interactions between radio-frequency (RF) signals and surrounding objects (e.g., reflection, diffraction, and scattering). Unlike conventional ray-tracing pipelines that rely on hand-engineer interaction rules, or black-box neural surrogates that do not explicitly model physical structure, we introduce SNRFT, a novel framework that integrates neural representations with physics-based RF propagation modeling. Our key idea is to view RF transport as a stochastic propagation process, from which a material-dependent attenuation coefficient emerges naturally as the rate parameter governing transport dynamics. This formulation inherently satisfies key physical constraints such as reciprocity and reversibility. Building on this foundation, we employ implicit neural representations to capture complex RF-object interactions while preserving the composability of traditional ray tracing. Extensive evaluations on real-world wireless communication and sensing testbeds demonstrate that SNRFT consistently outperforms existing methods, while requiring significantly fewer training samples.
BibTeX
@inproceedings{
bu2026stochastic,
title={Stochastic Neural Ray Tracing for Radio Frequency Channel Modeling},
author={Yinyan Bu and Jiajie Yu and Xingyu Chen and Bo Wen and Xinyu Zhang and Piya Pal},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=fIFBaE8fzO}
}