[發(fā)明專利]網(wǎng)絡(luò)流量分類方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202210039374.9
• 申請(qǐng)日: 2022-01-13
• 公開(kāi)（公告）號(hào): CN114338437A
• 公開(kāi)（公告）日: 2022-04-12
• 發(fā)明（設(shè)計(jì)）人: 楊楊;高志鵬;嚴(yán)雨;呂睿;高博文;趙斌男;李昱廷;郭義豪;龔興樂(lè);胡皓;劉澳倫;龍雨寒
• 申請(qǐng)（專利權(quán)）人: 北京郵電大學(xué)
• 主分類號(hào): H04L43/026
• 分類號(hào): H04L43/026;H04L43/12;H04L41/14;G06K9/62
• 代理公司: 北京路浩知識(shí)產(chǎn)權(quán)代理有限公司 11002
• 代理人: 喬慧

1.一種網(wǎng)絡(luò)流量分類方法，其特征在于，包括：

將捕獲的pcap文件切分為流序列，所述流序列由多個(gè)流量數(shù)據(jù)包組成；

從所述流序列中提取各個(gè)流量數(shù)據(jù)包的字節(jié)特征，得到以流為單位的字節(jié)序列；

對(duì)所述字節(jié)序列中的各個(gè)字節(jié)進(jìn)行位置編碼，并將編碼后的所述字節(jié)序列輸入至流量分類網(wǎng)絡(luò)模型中，得到所述流量分類網(wǎng)絡(luò)模型輸出的流量分類結(jié)果；其中，所述流量分類網(wǎng)絡(luò)模型是基于以流為單位的樣本和樣本對(duì)應(yīng)的流量分類結(jié)果訓(xùn)練后得到的。

2.根據(jù)權(quán)利要求1所述的網(wǎng)絡(luò)流量分類方法，其特征在于，所述將捕獲的pcap文件切分為流序列，包括：

基于五元組對(duì)所述pcap文件中的數(shù)據(jù)包流進(jìn)行切分，得到流序列；所述五元組包括：源IP地址、源端口、目的IP地址、目的端口和協(xié)議號(hào)。

3.根據(jù)權(quán)利要求1所述的網(wǎng)絡(luò)流量分類方法，其特征在于，從所述流序列中提取各個(gè)流量數(shù)據(jù)包的字節(jié)特征，得到以流為單位的字節(jié)序列，包括：

基于預(yù)設(shè)規(guī)則從所述流序列中提取各個(gè)流量數(shù)據(jù)包預(yù)設(shè)數(shù)量的字節(jié)特征，得到以流為單位的字節(jié)序列。

4.根據(jù)權(quán)利要求1所述的網(wǎng)絡(luò)流量分類方法，其特征在于，所述對(duì)所述字節(jié)序列中的各個(gè)字節(jié)進(jìn)行位置編碼，包括：

基于下述公式對(duì)所述字節(jié)序列中的各個(gè)字節(jié)進(jìn)行位置編碼，將每個(gè)字節(jié)在數(shù)據(jù)包中的位置轉(zhuǎn)化為d維的特征向量P_pos，所述公式為：

P(pos,2i)＝sin(pos/m^2i/d)

P(pos,2i+1)＝cos(pos/m²ⁱ)

其中，2i,2i+1∈[0,d-1]，表示生成的位置編碼的每個(gè)通道，m為常數(shù)，用于使每個(gè)字節(jié)的位置對(duì)應(yīng)唯一的位置編碼。

5.根據(jù)權(quán)利要求1所述的網(wǎng)絡(luò)流量分類方法，其特征在于，所述流量分類網(wǎng)絡(luò)模型由N個(gè)自動(dòng)編碼器層構(gòu)成，N≥2，所述流量分類網(wǎng)絡(luò)模型的損失函數(shù)為：

其中，h_i-1為第i個(gè)自動(dòng)編碼器的輸入層，N為以流為單位的樣本數(shù)量。

6.一種網(wǎng)絡(luò)流量分類裝置，其特征在于，包括：

第一處理模塊，用于將捕獲的pcap文件切分為流序列，所述流序列由多個(gè)流量數(shù)據(jù)包組成；

第二處理模塊，用于從所述流序列中提取各個(gè)流量數(shù)據(jù)包的字節(jié)特征，得到以流為單位的字節(jié)序列；

第三處理模塊，用于對(duì)所述字節(jié)序列中的各個(gè)字節(jié)進(jìn)行位置編碼，并將編碼后的所述字節(jié)序列輸入至流量分類網(wǎng)絡(luò)模型中，得到所述流量分類網(wǎng)絡(luò)模型輸出的流量分類結(jié)果；其中，所述流量分類網(wǎng)絡(luò)模型是基于以流為單位的樣本和樣本對(duì)應(yīng)的流量分類結(jié)果訓(xùn)練后得到的。

7.根據(jù)權(quán)利要求6所述的網(wǎng)絡(luò)流量分類裝置，其特征在于，所述第三處理模塊，具體用于：

基于下述公式對(duì)所述字節(jié)序列中的各個(gè)字節(jié)進(jìn)行位置編碼，將每個(gè)字節(jié)在數(shù)據(jù)包中的位置轉(zhuǎn)化為d維的特征向量P_pos，所述公式為：

P(pos,2i)＝sin(pos/m^2i/d)

P(pos,2i+1)＝cos(pos/m²ⁱ)

其中，2i,2i+1∈[0,d-1]，表示生成的位置編碼的每個(gè)通道，m為常數(shù)，用于使每個(gè)字節(jié)的位置對(duì)應(yīng)唯一的位置編碼。

8.一種電子設(shè)備，包括存儲(chǔ)器、處理器及存儲(chǔ)在所述存儲(chǔ)器上并可在所述處理器上運(yùn)行的計(jì)算機(jī)程序，其特征在于，所述處理器執(zhí)行所述程序時(shí)實(shí)現(xiàn)如權(quán)利要求1至5任一項(xiàng)所述網(wǎng)絡(luò)流量分類方法的步驟。

9.一種非暫態(tài)計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其上存儲(chǔ)有計(jì)算機(jī)程序，其特征在于，所述計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)如權(quán)利要求1至5任一項(xiàng)所述網(wǎng)絡(luò)流量分類方法的步驟。

10.一種計(jì)算機(jī)程序產(chǎn)品，包括計(jì)算機(jī)程序，其特征在于，所述計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)如權(quán)利要求1至5任一項(xiàng)所述網(wǎng)絡(luò)流量分類方法的步驟。