【儀表網(wǎng) 研發(fā)快訊】近日，北大電子學(xué)院程翔教授團隊提出首個面向信道預(yù)測的無線基座模型WiFo，該研究成果以“WiFo: Wireless Foundation Model for Channel Prediction”為題發(fā)表于期刊SCIENCE CHINA Information Sciences(IF=7.3，JCR Q1，CCF-A類期刊)。
 

　　精確的信道狀態(tài)信息(CSI)是B5G/6G MIMO-OFDM無線通信系統(tǒng)設(shè)計的基石，在預(yù)編碼和資源分配等關(guān)鍵任務(wù)中發(fā)揮著決定性作用。然而，傳統(tǒng)的基于信道估計的CSI獲取方式在高速移動等復(fù)雜場景中易受到“信道老化”等問題影響，嚴(yán)重制約系統(tǒng)性能。相比之下，信道預(yù)測技術(shù)通過預(yù)測獲取CSI，大大降低了信道估計和反饋開銷，顯著提升了系統(tǒng)頻譜效率，因而在過去幾十年中得到了廣泛研究，如圖1所示。早期工作探索了基于參數(shù)化模型的信道預(yù)測方案，但其性能嚴(yán)重依賴于信道模型的準(zhǔn)確度。而基于深度學(xué)習(xí)小模型的方案在高動態(tài)等復(fù)雜環(huán)境下性能受限，且泛化能力較差，難以滿足未來通信系統(tǒng)高速率和低開銷的要求。
 

　　近年來，基座模型的興起推動了深度學(xué)習(xí)范式的變革，即通過在大規(guī)模數(shù)據(jù)集上進行自監(jiān)督預(yù)訓(xùn)練，基座模型可以在多種下游任務(wù)中展現(xiàn)出卓越的泛化能力，并大幅超越任務(wù)專用模型。基座模型在自然語言處理等領(lǐng)域取得了突破性進展，催生了以ChatGPT和DeepSeek為代表的大語言模型(LLM)。前期LLM賦能通信物理層的相關(guān)工作局限于語言形式和語義提取等相關(guān)任務(wù)，難以直接處理非語言形式的信道預(yù)測任務(wù)。為突破語言局限，團隊前期工作(LLM4CP)首次將LLM應(yīng)用于信道預(yù)測任務(wù)，通過微調(diào)技術(shù)實現(xiàn)預(yù)訓(xùn)練LLM的通用知識遷移，提升了預(yù)測精度和泛化能力。然而，由于語言域和信道域的天然差異，LLM4CP在面臨超高速移動場景時仍然性能不足。當(dāng)遷移到新場景時，LLM4CP只能降低而無法完全避免額外的數(shù)據(jù)收集和訓(xùn)練開銷。此外，現(xiàn)有基于深度學(xué)習(xí)小模型和LLM的方案均針對時域和頻域兩種信道預(yù)測任務(wù)單獨設(shè)計網(wǎng)絡(luò)，大大增加了基站側(cè)模型部署開銷。
 

　　綜上所述，由于并非針對信道域進行構(gòu)建，基于通用基座模型(LLM)的信道預(yù)測方案在預(yù)測精度和泛化能力方面存在局限，無法從根本上突破傳統(tǒng)深度學(xué)習(xí)設(shè)計范式，無法打破新場景需重新訓(xùn)練和模型分立設(shè)計的瓶頸。那么，能否針對信道域構(gòu)建專用的無線基座模型，在大幅提升預(yù)測精度的同時實現(xiàn)一模多用，并能夠直接應(yīng)用于新場景而不加任何微調(diào)？ 程翔團隊對該問題首次進行了初步探索和系統(tǒng)回答，提出了首個面向信道預(yù)測的無線基座模型WiFo(Wireless Foundation Model)，開辟了全新的信道預(yù)測設(shè)計范式，從根本上克服了現(xiàn)有基于深度學(xué)習(xí)小模型和LLM信道預(yù)測方案的局限性，開創(chuàng)了信道預(yù)測技術(shù)的基座模型時代。
 

圖1. 信道預(yù)測技術(shù)發(fā)展歷史
 

　　打破分立設(shè)計的瓶頸，WiFo首次將時域和頻域的MIMO-OFDM信道預(yù)測統(tǒng)一建模為信道重建任務(wù)。為解決信道預(yù)測任務(wù)和CSI數(shù)據(jù)的異構(gòu)性，團隊提出了一種基于掩碼自編碼器(MAE)的網(wǎng)絡(luò)架構(gòu)，從而可以處理任意尺寸的三維CSI數(shù)據(jù)并應(yīng)用于異構(gòu)的信道預(yù)測任務(wù)。同時，他們設(shè)計了一種新的空時頻位置編碼結(jié)構(gòu)，以學(xué)習(xí)CSI樣本的三維位置信息。為充分捕獲通用的信道空時頻特征，團隊設(shè)計了三種自監(jiān)督預(yù)訓(xùn)練任務(wù)，即隨機掩碼重建，時域掩碼重建和頻域掩碼重建。預(yù)訓(xùn)練后的WiFo可直接零樣本用于信道預(yù)測任務(wù)，而無需額外訓(xùn)練。
 

圖2. WiFo網(wǎng)絡(luò)架構(gòu)圖
 

　　團隊基于QuaDRiGa仿真器構(gòu)建了大規(guī)模異構(gòu)CSI數(shù)據(jù)集，包括16個預(yù)訓(xùn)練數(shù)據(jù)集(包括16萬條CSI訓(xùn)練樣本)和3個泛化性測試數(shù)據(jù)集，覆蓋多種場景、用戶速度、頻點、天線和時間等配置，并考慮了5種參數(shù)量大小(從0.3M到86.1M)的模型，如表1和表2所示。WiFo在16個預(yù)訓(xùn)練數(shù)據(jù)集的時域和頻域預(yù)測結(jié)果分別如表3和表4所示，WiFo的平均NMSE(歸一化均方根誤差)性能達到SOTA，并且在大多數(shù)數(shù)據(jù)集上實現(xiàn)最優(yōu)或次優(yōu)，說明WiFo能夠同時在多個異構(gòu)數(shù)據(jù)集上訓(xùn)練，并同時掌握時域和頻域預(yù)測，實現(xiàn)“一模多用”。值得注意的是，為完成上述16個數(shù)據(jù)集上的時域和頻域信道預(yù)測任務(wù)，每種對比方案需要訓(xùn)練32個單獨的模型，而WiFo僅需訓(xùn)練1個模型。以D17數(shù)據(jù)集為例，WiFo的零樣本泛化性能如圖3所示。對比方案的零樣本泛化能力較差，難以有效預(yù)測，而WiFo的零樣本泛化性能超越了所有對比方案的全樣本訓(xùn)練性能，有望不加微調(diào)直接部署于新場景。此外，WiFo-Base的參數(shù)量僅為LLM4CP的1/4，而推理時間與普通小模型相當(dāng)，有望實際部署。
 

　　圖3. D17數(shù)據(jù)集：WiFo零樣本預(yù)測NMSE和對比方案的全樣本/零樣本預(yù)測NMSE

 

　　該工作的主要亮點包括：首個面向信道預(yù)測的無線基座模型和首個零樣本信道預(yù)測方案，首次實現(xiàn)一個模型同時解決異構(gòu)的信道預(yù)測任務(wù)和多樣的CSI數(shù)據(jù)；設(shè)計了基于MAE的無線基座模型架構(gòu)以解決任務(wù)與數(shù)據(jù)異構(gòu)性，并設(shè)計多種掩碼重建自監(jiān)督預(yù)訓(xùn)練任務(wù)以充分捕獲空時頻信道特征；構(gòu)建了大規(guī)模異構(gòu)數(shù)據(jù)集用于預(yù)訓(xùn)練(訓(xùn)練樣本數(shù)16萬)，仿真結(jié)果說明WiFo具備優(yōu)越的多數(shù)據(jù)集聯(lián)合學(xué)習(xí)能力，在新場景上的零樣本預(yù)測性能超越對比方案的全樣本性能，并具備與普通深度學(xué)習(xí)小模型相當(dāng)?shù)耐评黹_銷。
 

　　北京大學(xué)為該研究工作的第一完成單位，合作單位包括香港科技大學(xué)(廣州)。程翔為該論文的通訊作者，北京大學(xué)博士生劉伯珣為第一作者。這項工作得到了國家自然科學(xué)基金委杰出青年科學(xué)基金、國家重點研發(fā)計劃、新基石科學(xué)基金會科學(xué)探索獎等項目的支持。