Link Both Views: Cross-Image Routing Attention for Dual-Endpoint V2I Beam-Pair Power Prediction
Organizations: Joint SDU-NTU Centre for Artificial Intelligence Research (C-FAIR), Shandong University, Jinan 250101, China · School of Software, Shandong University · State Key Laboratory of Photonics and Communications, School of Electronics, Peking University, Beijing 100871, China
Abstract
For sensing-aided communications, single-endpoint sensing offers only a partial view of propagation, motivating link-centric sensing across both endpoints. We propose cross-image routing attention (CIRA) for sparse association and mutual enhancement of propagation-relevant features from dual-endpoint RGB images. Building on CIRA, CiFormer predicts millimeter-wave (mmWave) beam-pair power matrices for vehicle-to-infrastructure (V2I) communications. Experiments show that CiFormer reduces mean absolute error and improves top-1 beam-pair selection accuracy by 3.15 percentage points over the strongest baseline. Visualizations and ablation studies support dual-endpoint sensing and selective cross-view interaction in capturing propagation-relevant dependencies, suggesting broader applications in wireless tasks beyond beam prediction.
Figures & tables
| Model | Params. (M) | RMSE (dB) | MAE (dB) | Acc@1 (%) | Acc@3 (%) | Mean BF loss (dB) |
|---|---|---|---|---|---|---|
| ViT | 10.995 | 6.318 | 4.143 | 68.25 | 91.81 | 0.943 |
| MobileNetV4 | 14.950 | 6.553 | 4.156 | 68.39 | 91.16 | 0.924 |
| MambaOut | 11.831 | 6.014 | 3.864 | 69.26 | 93.03 | 0.658 |
| CiFormer | 10.986 | 5.810 | 3.638 | 72.41 | 94.68 | 0.641 |
| Variant | RMSE (dB) | MAE (dB) | Acc@1 (%) | Acc@3 (%) | Mean BF loss (dB) |
|---|---|---|---|---|---|
| w/o Tx | 10.097 | 7.047 | 40.95 | 65.30 | 6.590 |
| w/o Rx | 12.344 | 6.867 | 54.96 | 73.56 | 4.414 |
| w/o Interaction | 6.001 | 3.740 | 70.98 | 93.89 | 0.723 |
| w/o top- routing | 5.922 | 3.672 | 71.19 | 93.97 | 0.703 |
| w/o Gating | 5.827 | 3.678 | 71.19 | 93.53 | 0.756 |
| w/o SAB | 5.893 | 3.767 | 71.05 | 93.53 | 0.708 |