PRI-Net: A Lightweight Multimodal Framework for 3D UAV Localization
Organizations: College of Information Science and Electronic Engineering, Zhejiang University, Hangzhou, China
Abstract
Accurate 3D localization of unmanned aerial vehicles (UAVs) remains challenging for existing multimodal approaches due to sparse LiDAR geometry, modality-imbalanced fusion, and redundant feature transmission over constrained edge-to-server links. To address these limitations, we propose PRI-Net, an efficient and lightweight multimodal fusion framework for UAV localization that integrates point cloud splatting, residual attention fusion, and an information bottleneck. Specifically, a 3D point cloud splatting (3DPCS) strategy is introduced to transform sparse LiDAR observations into geometrically consistent dense depth maps. A residual attention fusion (RAF) module is then designed to alleviate modal bias by using an image branch for coarse estimation and a gated fusion branch for refinement. In addition, a multimodal information bottleneck (MIB) module compacts features by filtering task-irrelevant redundancy. Experiments show that PRI-Net achieves high localization accuracy with lightweight architectures, while reducing feature dimensionality and improving edge-to-server UAV sensing efficiency and robustness.
Figures & tables
| Module | Architectural Construction and Mathematical Operation |
|---|---|
| Stereo | |
| Encoder | |
| Depth Stem | |
| Depth Encoder | |
| Cross Fusion | |
| Pre-branch |
| Noise Level | Model | Latent Dimension ( ) | |||
|---|---|---|---|---|---|
| (m / ) | Type | 8 | 16 | 32 | 64 |
| None | IB | 0.1589 | 0.1542 | 0.1580 | 0.1595 |
| (Original) | MLP | 0.1629 | 0.1563 | 0.1596 | 0.1570 |
| IB | 0.2041 | 0.1820 | 0.1966 | 0.1908 | |
| MLP | 0.2133 | 0.1908 | 0.2014 | 0.1876 | |
| IB | 0.2033 | 0.1867 | 0.2022 | 0.1923 | |
Robot Localization
Robot localization estimates a robot’s position and orientation from onboard sensors, maps, and observations of its surroundings.
See all 177 papers