Large Wireless Localization Model (LWLM): A Foundation Model for Positioning in 6G Networks

Guangjin Pan, Kaixuan Huang, Hui Chen, Shunqing Zhang, Christian Häger +1 more
5/15/2025
eess.SPcs.AIcs.LG

No official code found. Best implementation: guangjinpan/LWLM (20 stars, last commit 2026-03-16).

Abstract

Accurate and robust localization is a critical enabler for emerging 5G and 6G applications, including autonomous driving, extended reality (XR), and smart manufacturing. While data-driven approaches have shown promise, most existing models require large amounts of labeled data and struggle to generalize across deployment scenarios and wireless configurations. To address these limitations, we propose a foundation-model-based solution tailored for wireless localization. We first analyze how different self-supervised learning (SSL) tasks acquire general-purpose and task-specific semantic features based on information bottleneck (IB) theory. Building on this foundation, we design a pretraining methodology for the proposed Large Wireless Localization Model (LWLM). Specifically, we propose an SSL framework that jointly optimizes three complementary objectives: (i) spatial-frequency masked channel modeling (SF-MCM), (ii) domain-transformation invariance (DTI), and (iii) position-invariant contrastive learning (PICL). These objectives jointly capture the underlying semantics of wireless channel from multiple perspectives. We further design lightweight decoders for key downstream tasks, including time-of-arrival (ToA) estimation, angle-of-arrival (AoA) estimation, single base station (BS) localization, and multiple BS localization. Comprehensive experimental results confirm that LWLM consistently surpasses both model-based and supervised learning baselines across all localization tasks. In particular, LWLM achieves 26.0%--87.5% improvement over transformer models without pretraining, and exhibits strong generalization under label-limited fine-tuning and unseen BS configurations, confirming its potential as a foundation model for wireless localization.

View on arXivView PDF

Code Implementations(1)

guangjinpan/LWLMCommunity100%

Large AI model for wireless localization

202PythonMay 13, 20257 months ago

Similar Papers

Cite this paper

@article{pan2025large,
  title  = {Large Wireless Localization Model (LWLM): A Foundation Model for Positioning in 6G Networks},
  author = {Guangjin Pan and Kaixuan Huang and Hui Chen and Shunqing Zhang and Christian Häger and Henk Wymeersch},
  year   = {2025},
  eprint = {2505.10134},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.10134v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.10134.svg)](https://codeofpaper.com/paper/2505.10134)

Discussion