用一维信号替代图像,提升恶意软件分类准确率
Signal-Based Malware Classification Using 1D CNNs
- 将二进制文件转为一维信号,避免图像化带来的信息损失
- 在MalNet数据集上实现0.874、0.503、0.507的高精度分类
- 专用于一维信号的残差网络,适合处理原始二进制特征
恶意软件分类是网络安全中的持续挑战:现代混淆技术可逃避传统静态分析,而动态分析又因资源消耗过大难以大规模部署。现有方法通过启发式将二进制文件重构成二维图像,再利用计算机视觉模型进行纹理识别,但该过程引入量化噪声和虚假二维依赖,造成显著信息丢失,限制分类性能。本文提出直接将文件重采样为一维信号,避免了启发式重构,并以浮点格式存储,消除量化误差。现有二维卷积网络可轻松适配一维信号分类。此外,基于ResNet与挤压-激励模块设计专用一维CNN,在MalNet数据集上实现二分类、类型级和家族级分类的F1分数分别为0.874、0.503和0.507,达到当前最优水平,为后续研究提供新信号模态。
原文摘要 · Abstract (English)
Malware classification is a contemporary and ongoing challenge in cyber-security: modern obfuscation techniques are able to evade traditional static analysis, while dynamic analysis is too resource intensive to be deployed at a large scale. One prominent line of research addresses these limitations by converting malware binaries into 2D images by heuristically reshaping them into a 2D grid before resizing using Lanczos resampling. These images can then be classified based on their textural information using computer vision approaches. While this approach can detect obfuscated malware more effectively than static analysis, the process of converting files into 2D images results in significant information loss due to both quantisation noise, caused by rounding to integer pixel values, and the introduction of 2D dependencies which do not exist in the original data. This loss of signal limits the classification performance of the downstream model. This work addresses these weaknesses by instead resizing the files into 1D signals which avoids the need for heuristic reshaping, and additionally these signals do not suffer from quantisation noise due to being stored in a floating-point format. It is shown that existing 2D CNN architectures can be readily adapted to classify these 1D signals for improved performance. Furthermore, a bespoke 1D convolutional neural network, based on the ResNet architecture and squeeze-and-excitation layers, was developed to classify these signals and evaluated on the MalNet dataset. It was found to achieve state-of-the-art performance on binary, type, and family level classification with F1 scores of 0.874, 0.503, and 0.507, respectively, paving the way for future models to operate on the proposed signal modality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。