人工智能大語(yǔ)言模型的運(yùn)行原理

作者：韓松

單位：中國(guó)移動(dòng)智慧家庭運(yùn)營(yíng)中心

隨著互聯(lián)網(wǎng)技術(shù)的發(fā)展，網(wǎng)絡(luò)應(yīng)用類(lèi)型日益復(fù)雜，從Web、郵件等傳統(tǒng)應(yīng)用，發(fā)展到各類(lèi)P2P應(yīng)用、自定義業(yè)務(wù)協(xié)議等。識(shí)別網(wǎng)絡(luò)流量中的應(yīng)用協(xié)議類(lèi)型成為網(wǎng)絡(luò)管理中的一個(gè)重要問(wèn)題。傳統(tǒng)的模式匹配和特征提取方法在面對(duì)復(fù)雜應(yīng)用協(xié)議時(shí)存在局限性。近年來(lái)，人工智能領(lǐng)域出現(xiàn)的大語(yǔ)言模型為應(yīng)用協(xié)議識(shí)別帶來(lái)新的契機(jī)。大語(yǔ)言模型通過(guò)預(yù)訓(xùn)練學(xué)習(xí)獲得對(duì)語(yǔ)言的深刻理解，可以運(yùn)用到對(duì)網(wǎng)絡(luò)流量的語(yǔ)義分析中。

具體來(lái)說(shuō)，大語(yǔ)言模型可以“讀懂”網(wǎng)絡(luò)流量中應(yīng)用層的協(xié)議語(yǔ)義，判斷HTTP、DNS等常見(jiàn)協(xié)議的格式，并可以對(duì)不常見(jiàn)協(xié)議進(jìn)行anomaly detection。另外，大語(yǔ)言模型學(xué)習(xí)的知識(shí)可以幫助進(jìn)行零樣本和少樣本學(xué)習(xí)，識(shí)別全新或樣本稀少的應(yīng)用協(xié)議。業(yè)界一些最新研究已經(jīng)證明了大語(yǔ)言模型在應(yīng)用協(xié)議識(shí)別中的效果。例如華為提出的ProtocolMate系統(tǒng)使用RoBERTa模型，實(shí)現(xiàn)了高達(dá)99.7%的準(zhǔn)確率。另外基于BERT等模型的語(yǔ)義模糊方法，也表現(xiàn)出色。這些成果都表明了大語(yǔ)言模型的潛力。

Part 01 ●??應(yīng)用協(xié)議識(shí)別是什么?●?

應(yīng)用協(xié)議識(shí)別(Application Protocol Identification)指的是識(shí)別網(wǎng)絡(luò)流量所使用的應(yīng)用層協(xié)議的方法。互聯(lián)網(wǎng)上的應(yīng)用通信需要遵循某種應(yīng)用層協(xié)議，比如HTTP協(xié)議用于網(wǎng)頁(yè)瀏覽，DNS協(xié)議用于域名解析等。為了管理網(wǎng)絡(luò)流量，需要能自動(dòng)識(shí)別流量所使用的應(yīng)用協(xié)議。

應(yīng)用協(xié)議識(shí)別的主要方法包括:

- 基于端口的識(shí)別：通過(guò)判斷特定端口號(hào)來(lái)識(shí)別協(xié)議，如80端口通常為HTTP。但有局限性。

- 模式匹配：通過(guò)查找協(xié)議特有的字節(jié)模式來(lái)匹配協(xié)議。

- 統(tǒng)計(jì)分析：提取流量統(tǒng)計(jì)特征，使用機(jī)器學(xué)習(xí)方法識(shí)別。

- 深度學(xué)習(xí)：使用LSTM、CNN等對(duì)流量進(jìn)行端到端的深度學(xué)習(xí),自動(dòng)提取特征。

- 語(yǔ)法分析：解析應(yīng)用層數(shù)據(jù)，判斷遵循的協(xié)議語(yǔ)法。

- 語(yǔ)義分析：利用大語(yǔ)言模型分析應(yīng)用層語(yǔ)義，識(shí)別協(xié)議含義。

應(yīng)用協(xié)議識(shí)別對(duì)于網(wǎng)絡(luò)流量監(jiān)控、安全防護(hù)等都有重要作用。隨著網(wǎng)絡(luò)應(yīng)用日益復(fù)雜，更智能高效的應(yīng)用協(xié)議識(shí)別技術(shù)變得尤為重要。

Part 02 ●?大語(yǔ)言模型的特點(diǎn)?●?

大語(yǔ)言模型具有以下幾個(gè)主要特點(diǎn)：（1）卓越的語(yǔ)義理解能力；（2）強(qiáng)大的遷移學(xué)習(xí)能力；（3）多樣化的應(yīng)用形式。

（1）卓越的語(yǔ)義理解能力

大語(yǔ)言模型通過(guò)預(yù)訓(xùn)練，可以深度理解語(yǔ)言的上下文和語(yǔ)義關(guān)系，對(duì)詞匯、語(yǔ)法、常識(shí)都有很強(qiáng)的理解能力。這使其可以進(jìn)行復(fù)雜的語(yǔ)義分析、文本生成等高難度語(yǔ)言處理任務(wù)。

（2）強(qiáng)大的遷移學(xué)習(xí)能力

大語(yǔ)言模型學(xué)到的語(yǔ)言知識(shí)具有很好的普適性，可以遷移至下游的不同任務(wù)中。即使下游任務(wù)數(shù)據(jù)不足，也可以取得不錯(cuò)的效果。這使大語(yǔ)言模型可以擴(kuò)展到更多不同的應(yīng)用領(lǐng)域。

（3）多樣化的應(yīng)用形式

大語(yǔ)言模型可以以不同的形式集成到實(shí)際應(yīng)用中，如通過(guò)微調(diào)進(jìn)行文本分類(lèi)、句子匹配；用作Encoder來(lái)提取語(yǔ)義特征；生成回復(fù)文本等。這使其能夠靈活地服務(wù)于不同的NLP應(yīng)用。

Part 03

●??大語(yǔ)言模型的數(shù)據(jù)結(jié)構(gòu)?●?

大語(yǔ)言模型通過(guò)組合使用各種特殊的數(shù)據(jù)結(jié)構(gòu)，模擬并實(shí)現(xiàn)了人類(lèi)語(yǔ)言的關(guān)鍵能力。它使用詞向量把詞轉(zhuǎn)換成數(shù)字編碼，類(lèi)似詞典定義。注意力機(jī)制讓詞語(yǔ)之間進(jìn)行交流，理解上下文，就像人類(lèi)的語(yǔ)言交流。深層網(wǎng)絡(luò)提取語(yǔ)義特征，殘差連接傳遞信息，位置編碼理解順序，都增強(qiáng)了模型的語(yǔ)言理解能力。遮蔽語(yǔ)言模型進(jìn)行自主學(xué)習(xí)，像人類(lèi)通過(guò)閱讀學(xué)習(xí)語(yǔ)言。數(shù)以?xún)|計(jì)的參數(shù)幫助記憶知識(shí)。通過(guò)集成這些數(shù)據(jù)結(jié)構(gòu)，大語(yǔ)言模型建立了一個(gè)像人腦一樣的語(yǔ)言理解和生成系統(tǒng)。它不僅能學(xué)習(xí)語(yǔ)言知識(shí)，理解語(yǔ)義，還能應(yīng)用語(yǔ)言進(jìn)行創(chuàng)造性的生成。大語(yǔ)言模型的數(shù)據(jù)結(jié)構(gòu)實(shí)現(xiàn)了它對(duì)語(yǔ)言的深刻理解，使其達(dá)到了接近人類(lèi)的語(yǔ)言處理能力。

　　數(shù)據(jù)結(jié)構(gòu)作用

　　嵌入層（Embedding Layer）將詞符號(hào)表示轉(zhuǎn)換為稠密向量表示

　　多頭注意力機(jī)制（Multi-Head Attention）允許模型同時(shí)關(guān)注文本不同位置信息

　　前饋全連接網(wǎng)絡(luò)（Feed Forward Fully Connected Network）在多頭注意力后進(jìn)行語(yǔ)義特征提取

　　殘差連接（Residual Connection）將前一層輸出同當(dāng)前層輸出相加，緩解梯度消失

　　Layer Normalization加速深層網(wǎng)絡(luò)訓(xùn)練，穩(wěn)定訓(xùn)練過(guò)程

　　位置編碼（Positional Encoding）為模型提供單詞順序信息

　　Mask機(jī)制在預(yù)訓(xùn)練中屏蔽部分輸入，實(shí)現(xiàn)自監(jiān)督

　　權(quán)重參數(shù)矩陣（Weight Matrices）計(jì)算注意力分?jǐn)?shù)、變換等，占用大量存儲(chǔ)和計(jì)算

大語(yǔ)言模型的運(yùn)行原理簡(jiǎn)要概括如下:

首先，大語(yǔ)言模型會(huì)將輸入的文本序列轉(zhuǎn)化為數(shù)字表示的詞向量，就像一個(gè)詞典將詞語(yǔ)映射為向量。然后這些詞向量被輸入到由多層Transformer模塊組成的編碼器結(jié)構(gòu)中。在每個(gè)Transformer模塊內(nèi)，通過(guò)多頭注意力機(jī)制使詞向量之間進(jìn)行交互，計(jì)算出詞與詞之間的相關(guān)性，以此來(lái)學(xué)習(xí)文本的上下文語(yǔ)義信息。

接著，經(jīng)過(guò)一個(gè)前饋全連接網(wǎng)絡(luò)進(jìn)行特征提取和表示變換。為了訓(xùn)練更深層的模型，還使用了殘差連接、層規(guī)范化等技術(shù)。位置編碼為模型加入了順序信息。大量的參數(shù)支持復(fù)雜的語(yǔ)義計(jì)算。在預(yù)訓(xùn)練階段,大語(yǔ)言模型通過(guò)自監(jiān)督任務(wù)如遮蔽語(yǔ)言模型學(xué)到通用的語(yǔ)言知識(shí)。在微調(diào)階段，加入監(jiān)督學(xué)習(xí)特定下游任務(wù)。這樣，大語(yǔ)言模型通過(guò)先獲得泛化語(yǔ)言表示能力，再遷移到具體任務(wù)中，實(shí)現(xiàn)了對(duì)人類(lèi)語(yǔ)言的建模和強(qiáng)大的語(yǔ)義理解與生成。它模擬人類(lèi)獲取語(yǔ)言知識(shí)然后應(yīng)用的過(guò)程。

圖2 大語(yǔ)言模型核心模塊運(yùn)行機(jī)制

各種數(shù)據(jù)源通過(guò)Kafka接入到數(shù)據(jù)平臺(tái)層，數(shù)據(jù)平臺(tái)講明細(xì)數(shù)據(jù)存入數(shù)據(jù)存儲(chǔ)層的ClickHouse中，明細(xì)數(shù)據(jù)的存活時(shí)間可以根據(jù)業(yè)務(wù)需求設(shè)置。同時(shí)可以根據(jù)業(yè)務(wù)報(bào)表查詢(xún)的不同維度，利用ClickHouse的物化視圖形成預(yù)聚合數(shù)據(jù)，提高數(shù)據(jù)查詢(xún)效率。由數(shù)據(jù)服務(wù)層的定時(shí)任務(wù)周期性地從ClickHouse的預(yù)聚合數(shù)據(jù)中查詢(xún)業(yè)務(wù)所需的展示數(shù)據(jù)，把展示數(shù)據(jù)存入MySQL。由數(shù)據(jù)服務(wù)層的報(bào)表服務(wù)向數(shù)據(jù)展示層提供查詢(xún)服務(wù)，報(bào)表服務(wù)直接查詢(xún)MySQL中的結(jié)果數(shù)據(jù)，保證了查詢(xún)效率和并發(fā)性。

Part 04

●??大語(yǔ)言模型在應(yīng)用協(xié)議識(shí)別中的應(yīng)用?●?

大語(yǔ)言模型具有強(qiáng)大的語(yǔ)言理解和建模能力，在應(yīng)用協(xié)議識(shí)別等領(lǐng)域中展現(xiàn)出巨大的應(yīng)用潛力。

下面將具體闡述大語(yǔ)言模型在應(yīng)用協(xié)議識(shí)別任務(wù)中的運(yùn)作機(jī)制和應(yīng)用流程：

應(yīng)用協(xié)議識(shí)別針對(duì)網(wǎng)絡(luò)流量中的應(yīng)用層協(xié)議報(bào)文，判斷其所屬的協(xié)議類(lèi)別，是網(wǎng)絡(luò)流量分析的關(guān)鍵環(huán)節(jié)。針對(duì)不同協(xié)議類(lèi)型，可以進(jìn)行定制化分析。相比基于規(guī)則的方法，基于大語(yǔ)言模型的協(xié)議識(shí)別具有更強(qiáng)的適應(yīng)性和拓展性。

大語(yǔ)言模型在協(xié)議識(shí)別任務(wù)中的應(yīng)用流程可以分成以下幾個(gè)步驟：

（1）數(shù)據(jù)預(yù)處理，需要收集大規(guī)模的應(yīng)用協(xié)議報(bào)文數(shù)據(jù)，如HTTP、DNS等，對(duì)報(bào)文數(shù)據(jù)進(jìn)行清洗,提取純凈的協(xié)議語(yǔ)料。

（2）構(gòu)建協(xié)議詞表，將報(bào)文通過(guò)詞表轉(zhuǎn)換為數(shù)字id序列，方便模型處理。在獲取處理后的數(shù)據(jù)集后,需要預(yù)訓(xùn)練語(yǔ)言模型以學(xué)習(xí)通用的協(xié)議語(yǔ)義特征。這里常用的預(yù)訓(xùn)練模型是BERT等變種。通過(guò)使用大量協(xié)議報(bào)文數(shù)據(jù)Fine-tune預(yù)訓(xùn)練模型，使其適應(yīng)協(xié)議語(yǔ)言的模式，獲得協(xié)議方面的先驗(yàn)知識(shí)。預(yù)訓(xùn)練時(shí)也會(huì)使用Mask等技巧增強(qiáng)模型對(duì)協(xié)議語(yǔ)義的建模能力。

（3）微調(diào)模型以適應(yīng)具體的協(xié)議識(shí)別任務(wù)。這里將建立一個(gè)協(xié)議類(lèi)別分類(lèi)模型，使用協(xié)議報(bào)文及其類(lèi)別標(biāo)簽進(jìn)行監(jiān)督訓(xùn)練。通過(guò)反向傳播等技術(shù)迭代優(yōu)化模型參數(shù)，使其逐步適應(yīng)協(xié)議識(shí)別任務(wù)，輸出精確的類(lèi)別判斷。

（4）部署微調(diào)模型。經(jīng)過(guò)預(yù)訓(xùn)練和微調(diào)后，可以導(dǎo)出獲得的協(xié)議識(shí)別模型，并集成到在線(xiàn)網(wǎng)絡(luò)流量分析系統(tǒng)中。在線(xiàn)部署后，對(duì)實(shí)時(shí)網(wǎng)絡(luò)流量進(jìn)行抓取，提取協(xié)議報(bào)文，輸入到導(dǎo)出的協(xié)議識(shí)別模型中，完成在線(xiàn)流量的智能協(xié)議分析。

整個(gè)流程充分利用了大語(yǔ)言模型支持的遷移學(xué)習(xí)范式。先通過(guò)自監(jiān)督在大規(guī)模語(yǔ)料上學(xué)習(xí)通用語(yǔ)言表示,然后快速適應(yīng)下游任務(wù)。相比獨(dú)立訓(xùn)練，這種方法顯著減少了人工特征工程，降低了對(duì)任務(wù)特定樣本量的依賴(lài)。同時(shí)也增強(qiáng)了模型的泛化能力。

綜上所述，依托預(yù)訓(xùn)練-微調(diào)框架，大語(yǔ)言模型可以高效適應(yīng)協(xié)議識(shí)別等專(zhuān)業(yè)領(lǐng)域任務(wù)。它模擬人類(lèi)語(yǔ)言學(xué)習(xí)過(guò)程，在大規(guī)模非標(biāo)注語(yǔ)料上學(xué)習(xí)語(yǔ)義知識(shí),然后遷移應(yīng)用。這為解決更多特定領(lǐng)域的語(yǔ)言理解問(wèn)題，提供了一個(gè)可靠的通用框架和技術(shù)路線(xiàn)。下圖給出了采用AI大語(yǔ)言模型的協(xié)議識(shí)別方法相對(duì)于傳統(tǒng)的基于域名規(guī)則的協(xié)議識(shí)別方法在準(zhǔn)確率，識(shí)別精度和召回率上的效果提升。

各種數(shù)據(jù)源通過(guò)Kafka接入到數(shù)據(jù)平臺(tái)層，數(shù)據(jù)平臺(tái)講明細(xì)數(shù)據(jù)存入數(shù)據(jù)存儲(chǔ)層的ClickHouse中，明細(xì)數(shù)據(jù)的存活時(shí)間可以根據(jù)業(yè)務(wù)需求設(shè)置。同時(shí)可以根據(jù)業(yè)務(wù)報(bào)表查詢(xún)的不同維度，利用ClickHouse的物化視圖形成預(yù)聚合數(shù)據(jù)，提高數(shù)據(jù)查詢(xún)效率。由數(shù)據(jù)服務(wù)層的定時(shí)任務(wù)周期性地從ClickHouse的預(yù)聚合數(shù)據(jù)中查詢(xún)業(yè)務(wù)所需的展示數(shù)據(jù)，把展示數(shù)據(jù)存入MySQL。由數(shù)據(jù)服務(wù)層的報(bào)表服務(wù)向數(shù)據(jù)展示層提供查詢(xún)服務(wù)，報(bào)表服務(wù)直接查詢(xún)MySQL中的結(jié)果數(shù)據(jù)，保證了查詢(xún)效率和并發(fā)性。

Part 05

●??總結(jié)展望?●?

總結(jié)來(lái)說(shuō)，大語(yǔ)言模型通過(guò)預(yù)訓(xùn)練-微調(diào)的框架，先在大規(guī)模協(xié)議語(yǔ)料上學(xué)習(xí)通用語(yǔ)義表示，然后遷移應(yīng)用到具體的協(xié)議識(shí)別任務(wù)，實(shí)現(xiàn)了對(duì)網(wǎng)絡(luò)流量的智能解析。相比規(guī)則方法，這種方式顯著提高了模型的適應(yīng)性和拓展性。

展望未來(lái)，大語(yǔ)言模型在應(yīng)用協(xié)議識(shí)別領(lǐng)域還有多個(gè)方向的探索價(jià)值：(1) 構(gòu)建更大規(guī)模的跨協(xié)議預(yù)訓(xùn)練語(yǔ)料庫(kù)，增強(qiáng)模型對(duì)協(xié)議語(yǔ)言的理解能力。(2)嘗試不同模型架構(gòu)，如基于編碼器-解碼器的BART等，進(jìn)一步提升建模效果。(3) 多任務(wù)學(xué)習(xí)框架，同時(shí)適配相關(guān)任務(wù)如協(xié)議語(yǔ)義解析，共享語(yǔ)義知識(shí)。(4) 在線(xiàn)增量學(xué)習(xí)機(jī)制，使deployed模型能隨新協(xié)議更新迭代。(5) 模型壓縮技術(shù)，部署輕量高效的協(xié)議識(shí)別引擎。(6) 可解釋性和安全性等考量，增加模型判斷的透明度和可控性。隨著模型和數(shù)據(jù)規(guī)模的提升，大語(yǔ)言模型必將持續(xù)改進(jìn)應(yīng)用協(xié)議理解，在更廣泛的網(wǎng)絡(luò)分析任務(wù)中發(fā)揮關(guān)鍵作用。

參考：

【1】自然語(yǔ)言大模型介紹，2023年04月01日，

編輯：黃飛

?