基于隐式随机梯度下降的恶意代码分类算法  

作　　者：张青[1] 曹金璇[1] 芦天亮[1] 

机构地区：[1]中国人民公安大学,北京100038

出　　处：《网络安全技术与应用》2023年第1期27-32,共6页Network Security Technology & Application

基　　金：国家社科基金重大项目资助(21&ZD193);中国人民公安大学2020年基本科研业务费重大项目(2020JKF101)。

摘　　要：针对目前基于深度学习的恶意代码分类算法易出现灾难性遗忘导致分类准确率不高、收敛过慢的问题,提出基于隐式随机梯度下降的恶意代码分类算法。与现有算法不同,该算法构造内外循环网络结构来协同学习最优网络模型以提高恶意代码分类准确率。在内循环优化阶段,通过优化带有偏好正则项的损失函数迫使内循环网络沿外循环网络方向更新权重从而避免内循环网络遗忘过去学到的知识。在外循环优化阶段,通过求解近似外循环网络梯度并利用隐式随机梯度下降优化外循环网络权重,使得外循环网络能够更快更稳定地收敛。在三个恶意代码数据集上的实验结果表明,该算法有效避免了灾难性遗忘,使用较少训练轮数取得了最高的分类准确率,显著提升了恶意代码分类的稳定性和鲁棒性。

关 键 词：恶意代码 深度学习 内外循环 偏好正则项 隐式随机梯度下降 

分 类 号：TP393.08[自动化与计算机技术—计算机应用技术]