LWM-CDE: A Representation Space for Wireless Data Reasoning and Transferability

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Alikhani, Sadjad, Malhotra, Akshay, Hamidi-Rad, Shahab, Alkhateeb, Ahmed
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916040783429632
author Alikhani, Sadjad
Malhotra, Akshay
Hamidi-Rad, Shahab
Alkhateeb, Ahmed
author_facet Alikhani, Sadjad
Malhotra, Akshay
Hamidi-Rad, Shahab
Alkhateeb, Ahmed
contents Machine learning deployments in real-world wireless communication tasks face significant generalization challenges due to location and environment-specific signal structure, high diversity in data across different deployments, and limited availability of real-world data. Current approaches for assessing data similarity between training and inference (deployment) distributions, as well as evaluating model transferability, suffer from high computational costs and inconsistent performance, leaving critical model deployment and model life cycle management decisions without a principled foundation. To address this, we introduce a dataset similarity framework built upon the feature space of a pretrained wireless foundation model. Our method, LWM-CDE (Contrastive learning of Dataset Embedding), fine-tunes the dataset embeddings of the foundation model using a combination of contrastive and geometry-shaping losses, creating a structured manifold where distance reliably indicates transferability. Extensive experiments on wireless benchmarks show that LWM-CDE achieves stronger correlation with empirical transfer performance than existing metrics while being more computationally efficient. The learned representation space supports more effective and data-efficient decision-making for tasks like source dataset selection, label-aware augmentation, and budgeted pretraining, demonstrating its broader utility across different wireless communication applications.
format Preprint
id arxiv_https___arxiv_org_abs_2605_24077
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle LWM-CDE: A Representation Space for Wireless Data Reasoning and Transferability
Alikhani, Sadjad
Malhotra, Akshay
Hamidi-Rad, Shahab
Alkhateeb, Ahmed
Signal Processing
Machine Learning
Machine learning deployments in real-world wireless communication tasks face significant generalization challenges due to location and environment-specific signal structure, high diversity in data across different deployments, and limited availability of real-world data. Current approaches for assessing data similarity between training and inference (deployment) distributions, as well as evaluating model transferability, suffer from high computational costs and inconsistent performance, leaving critical model deployment and model life cycle management decisions without a principled foundation. To address this, we introduce a dataset similarity framework built upon the feature space of a pretrained wireless foundation model. Our method, LWM-CDE (Contrastive learning of Dataset Embedding), fine-tunes the dataset embeddings of the foundation model using a combination of contrastive and geometry-shaping losses, creating a structured manifold where distance reliably indicates transferability. Extensive experiments on wireless benchmarks show that LWM-CDE achieves stronger correlation with empirical transfer performance than existing metrics while being more computationally efficient. The learned representation space supports more effective and data-efficient decision-making for tasks like source dataset selection, label-aware augmentation, and budgeted pretraining, demonstrating its broader utility across different wireless communication applications.
title LWM-CDE: A Representation Space for Wireless Data Reasoning and Transferability
topic Signal Processing
Machine Learning
url https://arxiv.org/abs/2605.24077