A Geometric Algebra-informed NeRF Framework for Generalizable Wireless Channel Prediction
Organizations: Knight Foundation School of Computing and Information Sciences, Florida International University, Miami, FL 33199, US · Department of Electrical and Computer Engineering, Auburn University, Auburn, AL 36849, US
Abstract
In this paper, we propose the geometric algebra-informed neural radiance fields (GAI-NeRF), a novel framework for wireless channel prediction that leverages geometric algebra attention mechanisms to capture ray-object interactions in complex propagation environments. Our approach incorporates global token representations, drawing inspiration from transformer architectures in language and vision domains, to aggregate learned spatial-electromagnetic features and enhance scene understanding. We identify limitations in conventional static ray tracing modules that hinder model generalization and address this challenge through a new ray tracing architecture. This design enables effective generalization across diverse wireless scenarios while maintaining computational efficiency. Experimental results demonstrate that GAI-NeRF achieves superior performance in channel prediction tasks by combining geometric algebra principles with neural scene representations, offering a promising direction for next-generation wireless communication systems. Moreover, GAI-NeRF greatly outperforms existing methods across multiple wireless scenarios. To ensure comprehensive assessment, we further evaluate our approach against multiple benchmarks using newly collected real-world indoor datasets tailored for single-scene downstream tasks and generalization testing, confirming its robust performance in unseen environments and establishing its high efficacy for wireless channel prediction.
Figures & tables
| Method | Room 1 (Our RSSI Dataset) | Room 2 (Our RSSI Dataset) | Other CSI Datasets | ||||
| MAE (dB) | SNR (dB) | ||||||
| 2.4 GHz | 5.0 GHz | 2.4 GHz | 5.0 GHz | MIMO | Bedroom | Conference | |
| MLP | 7.31 | 9.28 | 8.15 | 9.87 | 19.86 | 17.67 | 16.12 |
| VAE | 5.75 | 5.49 | 4.53 | 2.73 | 21.47 | 15.11 | 18.90 |
| DCGAN | 4.01 | 3.39 | 4.22 | 2.95 | 13.12 | 14.10 | 13.45 |
| NeRF 2 | 3.58 | 2.87 | 3.92 | 2.07 | 23.85 | 21.65 | 20.88 |
| Model | Scene 1 | Scene 2 | Scene 3 | |||
| 2.4 GHz | 5 GHz | 2.4 GHz | 5 GHz | 2.4 GHz | 5 GHz | |
| MLP | 11.92 | 9.65 | 12.52 | 10.08 | 13.51 | 10.21 |
| VAE | 6.63 | 5.51 | 5.91 | 6.13 | 6.14 | 5.50 |
| DCGAN | 4.61 | 4.48 | 4.91 | 4.33 | 4.23 | 4.17 |
| NeRF 2 | 4.34 | 3.79 | 4.14 | 2.93 | 3.67 | 2.97 |
| GAI-NeRF (Ours) | 3.25 | 3.37 | 3.48 | 2.33 | 3.45 | 2.27 |