用少量样本实现高精度恶意软件分类,提升新变种识别效率。
MalMixer: Few-Shot Malware Classification with Retrieval-Augmented Semi-Supervised Learning
- 结合检索增强与半监督学习,减少对大量标注数据依赖。
- 在少样本场景下达到当前最优分类性能,有效应对新型恶意软件。
- 适合安全研究人员快速部署,应对海量新出现的恶意样本。
恶意软件的持续增长给及时分类新样本带来了挑战。传统基于逆向工程的方法耗时费力,而现有深度学习分类器通常需要大量人工标注样本进行训练。当出现训练集外的新样本时,还需额外逆向工作更新数据集。本文提出 MalMixer,一种基于半监督学习的恶意软件家族分类器,可在极少训练样本下实现高准确率。通过引入领域知识感知的数据增强方法,提升恶意软件特征表示的泛化能力。实验表明,MalMixer 在少样本分类设置中达到当前最优性能,验证了轻量级、领域知识驱动的数据增强在恶意软件分类中的有效性,展示了此类半监督模型在实际威胁响应中的潜力。
原文摘要 · Abstract (English)
Recent growth and proliferation of malware have tested practitioners ability to promptly classify new samples according to malware families. In contrast to labor-intensive reverse engineering efforts, machine learning approaches have demonstrated increased speed and accuracy. However, most existing deep-learning malware family classifiers must be calibrated using a large number of samples that are painstakingly manually analyzed before training. Furthermore, as novel malware samples arise that are beyond the scope of the training set, additional reverse engineering effort must be employed to update the training set. The sheer volume of new samples found in the wild creates substantial pressure on practitioners ability to reverse engineer enough malware to adequately train modern classifiers. In this paper, we present MalMixer, a malware family classifier using semi-supervised learning that achieves high accuracy with sparse training data. We present a domain-knowledge-aware data augmentation technique for malware feature representations, enhancing few-shot performance of semi-supervised malware family classification. We show that MalMixer achieves state-of-the-art performance in few-shot malware family classification settings. Our research confirms the feasibility and effectiveness of lightweight, domain-knowledge-aware data augmentation methods for malware features and shows the capabilities of similar semi-supervised classifiers in addressing malware classification issues.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。