用GAN生成旧恶意软件样本,缓解分类模型遗忘问题。
MalCL: Leveraging GAN-Based Generative Replay to Combat Catastrophic Forgetting in Malware Classification
- 用GAN加特征匹配损失生成高质量恶意软件样本。
- 在Windows数据集上平均准确率达55%,比之前方法高28%。
- 适合关注持续学习与恶意软件检测的工程师和研究者。
针对恶意软件不断演化的威胁,持续学习(CL)在恶意软件分类中面临新类型不断出现带来的挑战。基于生成回放(GR)的CL系统通过生成模型合成旧数据,与新数据结合以重训练主模型。传统机器学习方法常出现灾难性遗忘,导致旧类别性能下降。本文提出一种基于GAN的GR系统,采用特征匹配损失生成高质量恶意软件样本,并设计基于模型隐藏层表示的新回放样本选择策略。在类增量学习场景下,对Windows和Android恶意软件数据集进行评估,结果表明该方法显著优于现有方案:例如,在Windows数据集上平均准确率达到55%,较其他GR模型提升28%。本研究为改进基于生成回放的恶意软件分类系统提供了实用洞见。代码将公开于https://github.com/MalwareReplayGAN/MalCL。
原文摘要 · Abstract (English)
Continual Learning (CL) for malware classification tackles the rapidly evolving nature of malware threats and the frequent emergence of new types. Generative Replay (GR)-based CL systems utilize a generative model to produce synthetic versions of past data, which are then combined with new data to retrain the primary model. Traditional machine learning techniques in this domain often struggle with catastrophic forgetting, where a model's performance on old data degrades over time. In this paper, we introduce a GR-based CL system that employs Generative Adversarial Networks (GANs) with feature matching loss to generate high-quality malware samples. Additionally, we implement innovative selection schemes for replay samples based on the model's hidden representations. Our comprehensive evaluation across Windows and Android malware datasets in a class-incremental learning scenario -- where new classes are introduced continuously over multiple tasks -- demonstrates substantial performance improvements over previous methods. For example, our system achieves an average accuracy of 55% on Windows malware samples, significantly outperforming other GR-based models by 28%. This study provides practical insights for advancing GR-based malware classification systems. The implementation is available at \url {https://github.com/MalwareReplayGAN/MalCL}\footnote{The code will be made public upon the presentation of the paper}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。