融合SENet和Transformer的应用层协议识别方法.pdfVIP

计算机科学与探索1673-9418/2024/18(03)-0805-13

JournalofFrontiersofComputerScienceandTechnologydoi:10.3778/j.issn.1673-9418.2304045

融合SENet和Transformer的应用层协议识别方法

+

陈乾，洪征，司健鹏

中国人民解放军陆军工程大学指挥控制工程学院，南京210000

+通信作者E-mail:hz5215@163.com

摘要：协议识别技术在网络通信和信息安全领域具有至关重要的地位和作用。针对现有基于时空特征的协

议识别方法提取协议特征不充分、不全面的问题，提出了一种基于SENet和Transformer的应用层协议识别方

法。该方法关注协议数据的时空特征，由加入SENet注意力的残差网络构成的空间特征提取模块和Trans-

former网络编码器构成的时间提取模块组成。空间特征提取阶段，在残差网络结构中加入SE块获取多个卷积

通道间的联系，自适应地为通道分配权重，提取不同通道中更加活跃的协议空间特征；时间特征提取阶段，由

基于多头注意力机制的Transformer编码器通过堆叠的方式构建时间特征提取模块，利用输入数据的位置信息

全面地获取协议数据的时间特征。通过对更加充足的空间特征和更加全面的时间特征的提取和学习，可以获

得更有效的协议识别信息，进而提高协议识别性能。在ISCX2012和CSE_CIC_IDS2018混合数据集上的实验

结果表明，所提模型的总体识别准确率达到99.20%，值达到98.99%，高于对比模型。

F1

关键词：SENet；残差网络；自注意力；Transformer；协议识别；网络安全

文献标志码：A中图分类号：TP398.08

ApplicationLayerProtocolRecognitionIncorporatingSENetandTransformer

+

CHENQian,HONGZheng,SIJianpeng

CommandandControlEngineeringCollege,ArmyEngineeringUniversityofPLA,Nanjing210000,China

Abstract:Protocolrecognitiontechnologyassumesacrucialpositionandexertssignificantinfluenceinthedo-

mainsofnetworkcommunicationandinformationsecurity.Existingprotocolrecognitionmethodsbasedonspatio-

temporalfeaturescannotadequatelyandcomprehensivelyextractprotocolfeatures.Anapplicationlayerprotocol

recognitionmethodincorporatingSENetchannelattentionandTransformerisproposed.Themodelfocusesonspatio-

temporalfeatureextractionofprotocoldata,andthemodelconsistsofaspatialfeatureextractionmoduleandatime

extractionmodule.SEblocksareaddedtotheresidualnetworktocapturetheassociationsbetweenmultiplechan-

nelsandadaptivelyassignweights,soastoextra