基于知識蒸餾的深度神經(jīng)網(wǎng)絡模型的流量分類方法及裝置

基本信息

申請?zhí)?/td> CN202011186016.8 申請日 -
公開(公告)號 CN112367273A 公開(公告)日 2021-02-12
申請公布號 CN112367273A 申請公布日 2021-02-12
分類號 H04L12/851(2013.01)I; 分類 電通信技術;
發(fā)明人 趙宇;魯敏;周斌;卜智勇 申請(專利權)人 上海瀚訊信息技術股份有限公司
代理機構 上海智信專利代理有限公司 代理人 鄧琪
地址 200335上海市長寧區(qū)金鐘路999號4幢601室
法律狀態(tài) -

摘要

摘要 本發(fā)明提供一種基于知識蒸餾的深度神經(jīng)網(wǎng)絡模型的流量分類方法,包括:構造一個深度神經(jīng)網(wǎng)絡作為用于流量分類的教師網(wǎng)絡模型;獲取并預處理真實網(wǎng)絡環(huán)境中的流量數(shù)據(jù),得到網(wǎng)絡流量數(shù)據(jù)庫,作為訓練集、驗證集和測試集;使用網(wǎng)絡流量數(shù)據(jù)庫和硬標簽來訓練教師網(wǎng)絡模型,訓練結(jié)束后保存經(jīng)過訓練的教師網(wǎng)絡模型;構建復雜度低于教師網(wǎng)絡模型的學生網(wǎng)絡模型;基于知識蒸餾方法輔助訓練學生網(wǎng)絡模型;采用學生網(wǎng)絡模型對流量數(shù)據(jù)進行分類。本發(fā)明還提供了相應的流量分類裝置。本發(fā)明的流量分類方法可以進一步減少流量分類模型的存儲和計算成本、對促進入侵檢測系統(tǒng)的邊緣部署和網(wǎng)絡智能化管理具有十分重要的意義。??