计算机工程与科学

作者：王栋, 杨珂, 玄佳兴, 韩雨桐, 赵丽花, 王旭仁

单位：1.国网电子商务有限公司国网雄安金融科技集团有限公司,北京 100053;2.国家电网有限公司区块链技术实验室,北京 100053;3.中国科学院信息工程研究所,北京 100093;4.首都师范大学信息工程学院,北京 100048

关键词：深度学习,一维卷积神经网络,半监督学习,生成对抗网络,恶意代码分类,

基金：国家自然科学基金(61872252);国家重点研发计划项目（2018YFB0805005）;国网电商公司科技项目(2500/2020-72001B)

随着互联网的发展，恶意代码呈现海量化与多态化的趋势，恶意代码家族分类是网络空间安全面临的挑战之一。将半监督生成对抗网络与深度卷积学习网络相结合，构建半监督深度卷积生成对抗网络，提出了一种恶意代码家族分类模型，通过恶意代码家族特征分析，对恶意代码进行特征提取，转化为一维灰度图像；然后基于一维卷积神经网络1D-CNN，构建半监督生成对抗网络SGAN，形成恶意代码家族分类模型SGAN-CNN。从特征提取优化、半监督生成对抗训练算法优化等方面进行恶意代码家族分类能力提升。为了验证SGAN-CNN模型的分类效果，在Microsoft Malware Classification Challenge数据集上进行实验。5折交叉验证测试显示，本文提出的模型在样本标注标签占80%的情况下，分类的平均准确率达到98.81%；在样本标注标签仅有20%的情况下，分类的平均准确率达到98.01%，取得了较好的分类效果。在小样本数量情况下，也能取得不错的分类准确率。

来源：2022年第5期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2022年第5期