arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.02490cs.CRcs.AI
收藏

基于预训练大语言模型加权层次集成的自动化恶意软件家族分类

Automated Malware Family Classification using Weighted Hierarchical Ensembles of Large Language Models

Canadian Institute for Cybersecurity, Faculty of Computer Science, University of New Brunswick(加拿大网络安全研究所,新不伦瑞克大学计算机科学学院)

Samita Bai, Hamed Jelodar, Tochukwu Emmanuel Nwankwo, Parisa Hamedi, Mohammad Meymani, Roozbeh Razavi-Far, Ali A. Ghorbani

AI总结:

本文提出基于预训练大语言模型加权层次集成的零标签恶意软件家族分类框架,通过聚合多个互补推理能力的语言模型决策预测,提升分类鲁棒性和分析员推理一致性。

摘要:

恶意软件家族分类仍然是自动化恶意软件分析中的挑战性任务,特别是在具有混淆、打包和快速演变威胁的现实环境中。现有的机器学习和深度学习方法通常依赖于标记数据集、手工特征、监督训练或动态分析,这限制了它们在开放世界场景中的可扩展性和有效性。本文提出了一种基于预训练大语言模型加权层次集成的零标签恶意软件家族分类框架。与依赖于特征级学习或模型再训练的方法不同,所提出的方法通过聚合多个互补推理能力的语言模型的决策级预测进行整合。模型输出使用经验确定的宏F1分数进行加权,并按层次结构组织,首先解决粗粒度的恶意行为,然后分配细粒度的恶意软件家族。这种结构增强了鲁棒性,减少了单个模型的不稳定性,并与分析员式的推理相一致。

英文摘要:

Malware family classification remains a challenging task in automated malware analysis, particularly in real-world settings characterized by obfuscation, packing, and rapidly evolving threats. Existing machine learning and deep learning approaches typically depend on labeled datasets, handcrafted features, supervised training, or dynamic analysis, which limits their scalability and effectiveness in open-world scenarios. This paper presents a zero-label malware family classification framework based on a weighted hierarchical ensemble of pretrained large language models (LLMs). Rather than relying on feature-level learning or model retraining, the proposed approach aggregates decision-level predictions from multiple LLMs with complementary reasoning strengths. Model outputs are weighted using empirically derived macro-F1 scores and organized hierarchically, first resolving coarse-grained malicious behavior before assigning fine-grained malware families. This structure enhances robustness, reduces individual model instability, and aligns with analyst-style reasoning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑