空军工程大学学报

北大核心,JST,Pж(AJ),CSCD扩展版,

国内刊号:61-1525/N

国际刊号:2097-1915

空军工程大学学报杂志2025年第1期:基于扩散模型的恶意代码数据集扩充方法

发布日期:

作者:李思聪1,王坚1,宋亚飞1,王硕2,冯存前1

单位:1. 空军工程大学防空反导学院,西安,710051;2. 95285部队,广西桂林,541000\

关键词:恶意代码检测;扩散模型;恶意代码可视化;数据增强技术;U-Net

基金:国家自然科学基金(61806219,61703426,61876189);陕西省自然科学基础研究计划(2021JM-226);陕西省高校科协青年人才托举计划(20190108,20220106);陕西省创新能力支撑计划(2020KJXX-065)

随着大数据的支撑,深度学习模型在计算机视觉和自然语言处理等领域展现出卓越的能力。然而,在恶意代码图像领域应用中,可能会出现训练数据不足的情况。由于部分恶意家族训练样本数量有限,无法充分描述整个数据集的分布特征,深度学习模型可能会过度拟合于这些稀缺数据,导致模型的性能不佳。针对以上问题,提出一种基于扩散模型生成新样本的数据集扩充方法,通过学习从原始数据到噪声的转换过程,并利用反向过程还原噪声样本为新的相似样本,实现数据集的扩充,生成与原始数据集相似但不同的新样本,以缓解部分家族数据不平衡对分类检测任务的影响,提高模型的泛化能力。

来源:2025年第1期

《空军工程大学学报》期刊编辑部

查看空军工程大学学报杂志2025年第1期

声明

严正声明:本站非期刊官网,非中介代理。

本站仅提供学术规范服务:快速预审、润色编辑服务、中英文查重、降重、去重服务、推荐合适的期刊投稿等学术规范服务。 如需提供学术规范服务请联系在线编辑。

联系我们

  • 地址:陕西省西安市灞桥区长乐东路甲字1号
  • 电话:029-84786242
  • E-mail:kgdbjb@163.com

咨询工作人员