arXiv:2606.12949cs.CRcs.CV2026-06

让病毒检测模型学会识别打包文件,提升鲁棒性。

ViPER: Vision-based Packing-Aware Encoder for Robust Malware Detection

论文配图:ViPER: Vision-based Packing-Aware Encoder for Robust Malware Detection
图 1 · 摘自论文原文
  • 用双头视觉模型同时判断恶意与否和是否打包
  • 在20万张图像上准确率达85.21%,打包识别AUC达99.49%
  • 适合需要抗逃避的恶意软件检测场景

基于可视化的恶意软件检测将原始二进制字节映射为灰度图像,并应用学习到的视觉分类器,提供一种抗逃避且无需反汇编的替代分析流程。然而,可执行文件打包仍是关键失效点:打包后的二进制生成高熵图像,掩盖了模型依赖的结构模式。由于打包也广泛存在于良性软件中(如压缩或版权保护),仅凭打包状态无法可靠判断恶意性,现有方法未在统一监督框架内解决此问题。我们提出ViPER,一种基于视觉的打包感知编码器,用于鲁棒恶意软件检测。ViPER基于LoRA微调的ViT-B/14主干网络,采用双头架构,联合学习恶意软件分类与打包检测。一个打包感知门控机制根据推断的打包状态条件化恶意预测,实现对打包与非打包输入的不同决策边界。为缓解训练中的打包标签偏移,采用频率加权损失并分层采样联合类别-打包层级。在20万张Windows PE字节图上评估,ViPER达到平衡准确率0.8521,ROC-AUC 0.9260,AUPR 0.9279,优于代表性最先进基线,在所有主要指标上均表现更优,同时打包检测的AUC达到0.9949。

原文摘要 · Abstract (English)

Visualization-based malware detection maps raw binary bytes to grayscale images and applies learned visual classifiers, providing an evasion-resistant and disassembly-free alternative to conventional analysis pipelines. However, executable packing remains a critical failure mode: packed binaries produce high-entropy images that obscure the structural patterns these models rely on. Because packing is also prevalent in benign software (e.g., for compression or copy protection), packing state alone is not a reliable indicator of maliciousness, and existing approaches do not address this challenge within a unified supervised framework. We present ViPER, a Vision-based Packing-Aware Encoder for Robust malware detection. ViPER builds on a LoRA-adapted ViT-B/14 backbone with a dual-head architecture that jointly learns malware classification and packing detection. A packing-aware gating mechanism conditions malware predictions on the inferred packing state, enabling distinct decision boundaries for packed and unpacked inputs. To address packing label skew during training, we employ frequency-weighted losses with stratified sampling over joint class-packing strata. Evaluated on 200,000 Windows PE byteplot images, ViPER achieves a balanced accuracy of 0.8521, ROC-AUC of 0.9260, and AUPR of 0.9279, outperforming representative state-of-the-art baselines across all primary metrics, while attaining a packing detection AUC of 0.9949.

恶意软件检测视觉模型打包识别

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。