Heteroscedastic Neural Networks for Path Loss Prediction with Link-Specific Uncertainty
Jonathan Ethier
TL;DR
The paper tackles the problem of path loss prediction with reliable uncertainty estimates by employing heteroscedastic neural networks trained with Gaussian negative log-likelihood to jointly predict mean path loss and link-specific variance. It systematically compares three parameter-sharing architectures and validates them on large blind drive-test datasets, finding that the shared-parameter model offers the best training stability and predictive performance, including well-calibrated 95% prediction intervals. The approach yields RMSE around 7.4 dB and calibrated intervals (PICP ~95%) with informative width (MPIW ~29.6 dB), outperforming a homoscedastic baseline and providing spatially varying uncertainty to support RF planning and diagnostics. Spatial uncertainty maps and diagnostic insights reveal where training data are deficient, enabling targeted data collection and feature enhancement for improved reliability in real-world deployments.
Abstract
Traditional and modern machine learning-based path loss models typically assume a constant prediction variance. We propose a neural network that jointly predicts the mean and link-specific variance by minimizing a Gaussian negative log-likelihood, enabling heteroscedastic uncertainty estimates. We compare shared, partially shared, and independent-parameter architectures using accuracy, calibration, and sharpness metrics on blind test sets from large public RF drive-test datasets. The shared-parameter architecture performs best, achieving an RMSE of 7.4 dB, 95.1 percent coverage for 95 percent prediction intervals, and a mean interval width of 29.6 dB. These uncertainty estimates further support link-specific coverage margins, improve RF planning and interference analyses, and provide effective self-diagnostics of model weaknesses.
