构建分层无线基础模型,实现多任务优化与跨场景泛化。
Hierarchical Wireless Foundation Model for Multi-Task Optimization

- 通过自监督掩码重建提取通用信道表征,结合可微输出头生成多任务决策。
- 在多种环境和配置下表现稳定,推理延迟显著低于传统数值方法。
- 模块化设计支持快速适配新任务,适合复杂无线系统研发场景。
下一代无线网络的日益复杂推动了人工智能在无线通信中的应用。然而,现有研究多聚焦于单一场景的任务特定深度学习技术,难以在不同任务、信道条件和系统配置间泛化。为突破这一瓶颈,我们提出一种分层无线基础模型(WFM),通过几何感知交叉注意力将上游基础信道编码器(FCE)与下游基础优化解码器(FOD)耦合。FCE利用自监督掩码重建提取任务无关的信道表征,FOD则通过可微输出头生成多任务优化决策。采用监督到无监督的混合训练策略,克服纯监督学习的性能瓶颈。该模型架构模块化,支持对未见通信任务的高效适配,参数开销极低。仿真结果表明,所提WFM能学习高保真信道表征,在多任务优化中表现优异,同时显著降低优化推理延迟。此外,其对未见传播环境、变化约束参数及异构系统配置均表现出强鲁棒性。
原文摘要 · Abstract (English)
The increasing complexity of next-generation wireless networks has driven the integration of artificial intelligence (AI) into wireless communications. However, most existing studies focus on developing task-specific deep learning techniques for single scenarios, which limits their ability to generalize across diverse tasks, channel conditions, and system configurations. To address this generalization bottleneck, we propose a hierarchical wireless foundation model (WFM) for multi-task optimization. The proposed WFM couples an upstream foundation channel encoder (FCE) with a downstream foundation optimization decoder (FOD) via geometry-aware cross-attention. Specifically, the FCE extracts task-agnostic channel representations via self-supervised masked reconstruction while the FOD generates multi-task optimization decisions through differentiable output heads. Moreover, a hybrid supervised-to-unsupervised training strategy is employed to overcome the performance ceiling of purely supervised learning, and the modular architecture of the WFM enables efficient adaptation to unseen communication tasks with minimal parameter overhead. Simulation results show that the proposed WFM learns high-fidelity channel representations and achieves competitive multi-task optimization performance while substantially reducing optimization inference latency relative to numerical baselines. Furthermore, it exhibits robust generalization to unseen propagation environments, varying constraint parameters, and heterogeneous system configurations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。